diff --git a/telethon/client/updates.py b/telethon/client/updates.py index d6400d7d..8bac5c9c 100644 --- a/telethon/client/updates.py +++ b/telethon/client/updates.py @@ -154,7 +154,7 @@ class UpdateMethods(UserMethods): else: state = d.intermediate_state - self._handle_update(types.Updates( + await self._handle_update(types.Updates( users=d.users, chats=d.chats, date=state.date, @@ -186,9 +186,9 @@ class UpdateMethods(UserMethods): itertools.chain(update.users, update.chats)} for u in update.updates: u._entities = entities - self._handle_update(u) + await self._handle_update(u) elif isinstance(update, types.UpdateShort): - self._handle_update(update.update) + await self._handle_update(update.update) else: update._entities = getattr(update, '_entities', {}) if self._updates_queue is None: diff --git a/telethon/network/mtprotosender.py b/telethon/network/mtprotosender.py index eb868fcd..86f5b739 100644 --- a/telethon/network/mtprotosender.py +++ b/telethon/network/mtprotosender.py @@ -504,7 +504,7 @@ class MTProtoSender: __log__.debug('Handling update {}' .format(message.obj.__class__.__name__)) if self._update_callback: - self._update_callback(message.obj) + await self._update_callback(message.obj) async def _handle_pong(self, message): """