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:
thec0sm0s
2020-05-12 12:20:32 +05:30
parent bc22196695
commit 6365ab461a
4 changed files with 33 additions and 7 deletions

View File

@@ -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.