python/aiogram
1
0
Fork 0
mirror of https://github.com/aiogram/aiogram.git synced 2026-04-08 16:37:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Configure Deserializable object. (Set bot for types) and stop using __slots__

Browse source
This commit is contained in:
Alex Root Junior 2017-05-26 05:17:26 +03:00
parent a703fc9587
commit 716865695c
21 changed files with 24 additions and 69 deletions
Download patch file Download diff file Expand all files Collapse all files

10
aiogram/bot.py
View file

@ -34,6 +34,10 @@ class AIOGramBot:
def _on_exit(self):
self.session.close()
def prepare_object(self, obj):
obj.bot = self
return obj
@property
async def me(self) -> User:
if not hasattr(self, '_me'):
@ -45,12 +49,12 @@ class AIOGramBot:
async def get_me(self) -> User:
raw = await self.request(ApiMethods.GET_ME)
return User.de_json(raw)
return self.prepare_object(User.de_json(raw))
async def get_chat(self, chat_id) -> Chat:
payload = generate_payload(**locals())
raw = await self.request(ApiMethods.GET_CHAT, payload)
return Chat.de_json(raw)
return self.prepare_object(Chat.de_json(raw))
async def get_updates(self, offset=None, limit=None, timeout=None, allowed_updates=None):
"""
@ -64,4 +68,4 @@ class AIOGramBot:
"""
payload = generate_payload(**locals())
raw = await self.request(ApiMethods.GET_UPDATES, payload)
return [Update.de_json(raw_update) for raw_update in raw]
return [self.prepare_object(Update.de_json(raw_update)) for raw_update in raw]