LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2024-11-22 17:36:34 +03:00
Code Issues Packages Projects Releases Wiki Activity

Make EntityDatabase.get a no-op if key is an entity already

Browse Source
This commit is contained in:
Lonami Exo 2017-10-05 12:28:04 +02:00
parent e5c4df98df
commit d2e244817a
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
telethon/tl/entity_database.py
View File

@ -117,8 +117,14 @@ class EntityDatabase:
if isinstance(key, int):
return self._entities[key] # normal IDs are assumed users
if isinstance(key, TLObject) and type(key).SUBCLASS_OF_ID == 0x2d45687:
return self._entities[utils.get_peer_id(key, add_mark=True)]
if isinstance(key, TLObject):
sc = type(key).SUBCLASS_OF_ID
if sc == 0x2d45687:
# Subclass of "Peer"
return self._entities[utils.get_peer_id(key, add_mark=True)]
elif sc in {0x2da17977, 0xc5af5d94, 0x6d44b7db}:
# Subclass of "User", "Chat" or "Channel"
return key
raise KeyError(key)