LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-02-10 08:30:52 +03:00
Code Issues Packages Projects Releases Wiki Activity

Allow using a callable key on EntityDatabase

Browse Source
This commit is contained in:
Lonami Exo 2017-10-12 19:54:23 +02:00
parent 16a0cecf46
commit f2338e49ae
1 changed files with 13 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
telethon/tl/entity_database.py
View File

@ -125,8 +125,13 @@ class EntityDatabase:
def _parse_key(self, key): def _parse_key(self, key):
"""Parses the given string, integer or TLObject key into a """Parses the given string, integer or TLObject key into a
marked user ID ready for use on self._entities. Raises marked user ID ready for use on self._entities.
ValueError if it cannot be parsed.
If a callable key is given, the entity will be passed to the
function, and if it returns a true-like value, the marked ID
for such entity will be returned.
Raises ValueError if it cannot be parsed.
""" """
if isinstance(key, str): if isinstance(key, str):
phone = EntityDatabase.parse_phone(key) phone = EntityDatabase.parse_phone(key)
@ -144,20 +149,15 @@ class EntityDatabase:
if isinstance(key, TLObject): if isinstance(key, TLObject):
return utils.get_peer_id(key, add_mark=True) return utils.get_peer_id(key, add_mark=True)
if callable(key):
for k, v in self._entities.items():
if key(v):
return k
raise ValueError() raise ValueError()
def __getitem__(self, key): def __getitem__(self, key):
"""Accepts a digit only string as phone number, """See the ._parse_key() docstring for possible values of the key"""
otherwise it's treated as an username.
If an integer is given, it's treated as the ID of the desired User.
The ID given won't try to be guessed as the ID of a chat or channel,
as there may be an user with that ID, and it would be unreliable.
If a Peer is given (PeerUser, PeerChat, PeerChannel),
its specific entity is retrieved as User, Chat or Channel.
Note that megagroups are channels with .megagroup = True.
"""
try: try:
return self._entities[self._parse_key(key)] return self._entities[self._parse_key(key)]
except (ValueError, KeyError) as e: except (ValueError, KeyError) as e: