hoelee/Telethon
1
0
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-08-05 03:22:29 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Allow auto-casting custom.Dialog into input_entity

Browse Source
This commit is contained in:
Lonami Exo
2018-04-06 19:11:31 +02:00
parent baa6976a0b
commit 0cd44b245c
2 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
telethon/utils.py
View File

@@ -91,7 +91,11 @@ def get_input_peer(entity, allow_self=True):
if entity.SUBCLASS_OF_ID == 0xc91c90b6: # crc32(b'InputPeer')
return entity
except AttributeError:
_raise_cast_fail(entity, 'InputPeer')
if hasattr(entity, 'input_entity'):
# e.g. custom.Dialog (can't cyclic import)
return entity.input_entity
else:
_raise_cast_fail(entity, 'InputPeer')
if isinstance(entity, User):
if entity.is_self and allow_self:
@@ -105,7 +109,6 @@ def get_input_peer(entity, allow_self=True):
if isinstance(entity, (Channel, ChannelForbidden)):
return InputPeerChannel(entity.id, entity.access_hash or 0)
# Less common cases
if isinstance(entity, InputUser):
return InputPeerUser(entity.user_id, entity.access_hash)