mirror of
https://github.com/yawaflua/Flask-Discord.git
synced 2025-12-10 12:29:30 +02:00
Use DiscordModelsMeta class. Set ROUTE as required class attribute for bases. Add fetch_from_api classmethod which is shorthand to return the current model
This commit is contained in:
@@ -1,8 +1,22 @@
|
||||
from flask import current_app
|
||||
from abc import ABC
|
||||
from abc import ABCMeta, abstractmethod
|
||||
|
||||
|
||||
class DiscordModelsBase(ABC):
|
||||
class DiscordModelsMeta(ABCMeta):
|
||||
|
||||
ROUTE = str()
|
||||
|
||||
def __init__(cls, name, *args, **kwargs):
|
||||
if not cls.ROUTE and name != "DiscordModelsBase":
|
||||
raise NotImplementedError(f"ROUTE must be specified in a Discord model: {name}.")
|
||||
super().__init__(name, *args, **kwargs)
|
||||
|
||||
|
||||
class DiscordModelsBase(metaclass=DiscordModelsMeta):
|
||||
|
||||
@abstractmethod
|
||||
def __init__(self, payload):
|
||||
self._payload = payload
|
||||
|
||||
@staticmethod
|
||||
def _request(*args, **kwargs):
|
||||
@@ -12,6 +26,11 @@ class DiscordModelsBase(ABC):
|
||||
"""
|
||||
return current_app.discord.request(*args, **kwargs)
|
||||
|
||||
@classmethod
|
||||
def fetch_from_api(cls):
|
||||
"""A class method which returns instance of this model by implicitly making an API call to Discord."""
|
||||
return cls(cls._request(cls.ROUTE))
|
||||
|
||||
def to_json(self):
|
||||
"""A utility method which returns raw payload object as it was received from discord.
|
||||
|
||||
|
||||
Reference in New Issue
Block a user