diff --git a/animu/client.py b/animu/client.py index 38ade06..46f790e 100644 --- a/animu/client.py +++ b/animu/client.py @@ -2,6 +2,7 @@ from animu import model class Client: + def __init__(self, token: str): self._header = {"Auth": token} self._baseurl = "https://animu.ml/api/" diff --git a/animu/model.py b/animu/model.py index 1cebcb4..44375b2 100644 --- a/animu/model.py +++ b/animu/model.py @@ -1,15 +1,12 @@ -class Base: +class _Base: def __init__(self, data): self._data = data - + def __getitem__(self, key): return self._data[key] - def __str__(self): - return str(self._data) - -class Fact(Base): +class Fact(_Base): def __init__(self, data): super().__init__(data) @@ -17,32 +14,44 @@ def __init__(self, data): self.fact = data.get('fact') self.tags = data.get('tags') + def __str__(self): + return self.fact -class Password(Base): - def __init__(self, data): + +class Password(_Base): + def __init__(self, data): super().__init__(data) self.password = data.get('pass') + def __str__(self): + return self.password + -class Quote(Base): +class Quote(_Base): def __init__(self, data): super().__init__(data) - + self.id = data.get('_id') self.quote = data.get('quote') self.anime = data.get('anime') self.said = data.get('said') + def __str__(self): + return self.quote + -class Roleplay(Base): +class Roleplay(_Base): def __init__(self, data): super().__init__(data) self.url = data.get('url') + def __str__(self): + return self.url + -class Waifu(Base): +class Waifu(_Base): def __init__(self, data): super().__init__(data) @@ -63,4 +72,7 @@ def __init__(self, data): self.love = self.statistics.get('love') self.hate = self.statistics.get('hate') self.upvote = self.statistics.get('upvote') - self.downvote = self.statistics.get('downvote') \ No newline at end of file + self.downvote = self.statistics.get('downvote') + + def __str__(self): + return self.en_name or self.jp_name or self.alt_name \ No newline at end of file