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

Create utils.get_peer

Browse Source
This commit is contained in:
Lonami Exo 2018-08-02 12:56:40 +02:00
parent 80a5e709cb
commit 10da14418c
1 changed files with 35 additions and 34 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

67
telethon/utils.py
View File

@ -660,6 +660,28 @@ def get_inner_text(text, entities):
return result return result
def get_peer(peer):
try:
if peer.SUBCLASS_OF_ID == 0x2d45687:
return peer
elif isinstance(peer, (
types.contacts.ResolvedPeer, types.InputNotifyPeer,
types.TopPeer)):
return peer.peer
elif isinstance(peer, types.ChannelFull):
return types.PeerChannel(peer.id)
peer = get_input_peer(peer, allow_self=False)
if isinstance(peer, types.InputPeerUser):
return types.PeerUser(peer.user_id)
elif isinstance(peer, types.InputPeerChat):
return types.PeerChat(peer.chat_id)
elif isinstance(peer, types.InputPeerChannel):
return types.PeerChannel(peer.channel_id)
except (AttributeError, TypeError):
_raise_cast_fail(peer, 'Peer')
def get_peer_id(peer, add_mark=True): def get_peer_id(peer, add_mark=True):
""" """
Finds the ID of the given peer, and converts it to the "bot api" format Finds the ID of the given peer, and converts it to the "bot api" format
@ -674,51 +696,30 @@ def get_peer_id(peer, add_mark=True):
return peer if add_mark else resolve_id(peer)[0] return peer if add_mark else resolve_id(peer)[0]
try: try:
if peer.SUBCLASS_OF_ID not in (0x2d45687, 0xc91c90b6): peer = get_peer(peer)
if isinstance(peer, ( except TypeError:
types.contacts.ResolvedPeer, types.InputNotifyPeer,
types.TopPeer)):
peer = peer.peer
else:
# Not a Peer or an InputPeer, so first get its Input version
peer = get_input_peer(peer, allow_self=False)
except AttributeError:
_raise_cast_fail(peer, 'int') _raise_cast_fail(peer, 'int')
# Set the right ID/kind, or raise if the TLObject is not recognised if isinstance(peer, types.PeerUser):
if isinstance(peer, (types.PeerUser, types.InputPeerUser)):
return peer.user_id return peer.user_id
elif isinstance(peer, (types.PeerChat, types.InputPeerChat)): elif isinstance(peer, types.PeerChat):
# Check in case the user mixed things up to avoid blowing up # Check in case the user mixed things up to avoid blowing up
if not (0 < peer.chat_id <= 0x7fffffff): if not (0 < peer.chat_id <= 0x7fffffff):
peer.chat_id = resolve_id(peer.chat_id)[0] peer.chat_id = resolve_id(peer.chat_id)[0]
return -peer.chat_id if add_mark else peer.chat_id return -peer.chat_id if add_mark else peer.chat_id
elif isinstance(peer, ( else: # if isinstance(peer, types.PeerChannel):
types.PeerChannel, types.InputPeerChannel, types.ChannelFull)):
if isinstance(peer, types.ChannelFull):
# Special case: .get_input_peer can't return InputChannel from
# ChannelFull since it doesn't have an .access_hash attribute.
i = peer.id
else:
i = peer.channel_id
# Check in case the user mixed things up to avoid blowing up # Check in case the user mixed things up to avoid blowing up
if not (0 < i <= 0x7fffffff): if not (0 < peer.channel_id <= 0x7fffffff):
i = resolve_id(i)[0] peer.channel_id = resolve_id(peer.channel_id)[0]
if isinstance(peer, types.ChannelFull):
peer.id = i if not add_mark:
else: return peer.channel_id
peer.channel_id = i
if add_mark:
# Concat -100 through math tricks, .to_supergroup() on # Concat -100 through math tricks, .to_supergroup() on
# Madeline IDs will be strictly positive -> log works. # Madeline IDs will be strictly positive -> log works.
return -(i + pow(10, math.floor(math.log10(i) + 3))) return -(peer.channel_id + pow(
else: 10, math.floor(math.log10(peer.channel_id) + 3)))
return i
_raise_cast_fail(peer, 'int')
def resolve_id(marked_id): def resolve_id(marked_id):