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

Fix update dispatch errors

Browse Source
This commit is contained in:
Lonami Exo 2022-03-02 13:10:33 +01:00
parent 2a750fa696
commit 4d6e75a54e
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
telethon/_client/updates.py
View File

@ -18,6 +18,7 @@ from .._events.base import StopPropagation, EventBuilder, EventHandler
from .._events.filters import make_filter, NotResolved from .._events.filters import make_filter, NotResolved
from .._misc import utils from .._misc import utils
from .. import _tl from .. import _tl
from ..types._custom import User, Chat
if typing.TYPE_CHECKING: if typing.TYPE_CHECKING:
from .telegramclient import TelegramClient from .telegramclient import TelegramClient
@ -220,12 +221,13 @@ def _preprocess_updates(self, updates, users, chats):
class Entities: class Entities:
def __init__(self, client, users, chats): def __init__(self, client, users, chats):
self.self_id = client._session_state.user_id self.self_id = client._session_state.user_id
self._client = client
self._entities = {e.id: e for e in itertools.chain( self._entities = {e.id: e for e in itertools.chain(
(User(client, u) for u in users), (User._new(client, u) for u in users),
(Chat(client, c) for u in chats), (Chat._new(client, c) for u in chats),
)} )}
def get(self, client, peer): def get(self, peer):
if not peer: if not peer:
return None return None
@ -233,11 +235,11 @@ class Entities:
try: try:
return self._entities[id] return self._entities[id]
except KeyError: except KeyError:
entity = client._entity_cache.get(query.user_id) entity = self._client._entity_cache.get(query.user_id)
if not entity: if not entity:
raise RuntimeError('Update is missing a hash but did not trigger a gap') raise RuntimeError('Update is missing a hash but did not trigger a gap')
self._entities[entity.id] = User(client, entity) if entity.is_user else Chat(client, entity) self._entities[entity.id] = User(self._client, entity) if entity.is_user else Chat(self._client, entity)
return self._entities[entity.id] return self._entities[entity.id]
@ -246,7 +248,7 @@ async def _dispatch(self, update, entities):
try: try:
event_cache = {} event_cache = {}
for handler in self._update_handlers: for handler in self._update_handlers:
event, entities = event_cache.get(handler._event) event = event_cache.get(handler._event)
if not event: if not event:
# build can fail if we're missing an access hash; we want this to crash # build can fail if we're missing an access hash; we want this to crash
event_cache[handler._event] = event = handler._event._build(self, update, entities) event_cache[handler._event] = event = handler._event._build(self, update, entities)