LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-03-25 04:14:29 +03:00
Code Issues Packages Projects Releases Wiki Activity

Don't crash when receiving updates prior to login

Browse Source 
Fixes #1467, and enables #1471.
This commit is contained in:
Lonami Exo 2020-06-05 21:17:09 +02:00
parent 493f69f195
commit bfa995d52b
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
telethon/client/updates.py
View File

@ -397,6 +397,9 @@ class UpdateMethods:
# Some updates require our own ID, so we must make sure # Some updates require our own ID, so we must make sure
# that the event builder has offline access to it. Calling # that the event builder has offline access to it. Calling
# `get_me()` will cache it under `self._self_input_peer`. # `get_me()` will cache it under `self._self_input_peer`.
#
# It will return `None` if we haven't logged in yet which is
# fine, we will just retry next time anyway.
await self.get_me(input_peer=True) await self.get_me(input_peer=True)
built = EventBuilderDict(self, update, others) built = EventBuilderDict(self, update, others)
@ -566,8 +569,15 @@ class EventBuilderDict:
try: try:
return self.__dict__[builder] return self.__dict__[builder]
except KeyError: except KeyError:
# Updates may arrive before login (like updateLoginToken) and we
# won't have our self ID yet (anyway only new messages need it).
self_id = (
self.client._self_input_peer.user_id
if self.client._self_input_peer
else None
)
event = self.__dict__[builder] = builder.build( event = self.__dict__[builder] = builder.build(
self.update, self.others, self.client._self_input_peer.user_id) self.update, self.others, self_id)
if isinstance(event, EventCommon): if isinstance(event, EventCommon):
event.original_update = self.update event.original_update = self.update