Xtao-Labs/MTPyroger
3
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge branch 'master' into docs

Browse Source
This commit is contained in:
Dan 2018-01-26 15:24:45 +01:00
commit faec832f2b
1 changed files with 15 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
pyrogram/client/client.py
View File

@ -524,6 +524,7 @@ class Client:
peer_id,
peer_access_hash
)
peer_id = int("-100" + str(peer_id))
else:
continue
@ -547,35 +548,38 @@ class Client:
user_id=resolved_peer.users[0].id,
access_hash=resolved_peer.users[0].access_hash
)
chat_id = input_peer.user_id
peer_id = input_peer.user_id
elif type(resolved_peer.peer) is PeerChannel:
input_peer = InputPeerChannel(
channel_id=resolved_peer.chats[0].id,
access_hash=resolved_peer.chats[0].access_hash
)
chat_id = input_peer.channel_id
peer_id = int("-100" + str(input_peer.channel_id))
else:
raise PeerIdInvalid
self.peers_by_username[username] = input_peer
self.peers_by_id[chat_id] = input_peer
self.peers_by_id[peer_id] = input_peer
return input_peer
def resolve_peer(self, chat_id: int or str):
if chat_id in ("self", "me"):
def resolve_peer(self, peer_id: int or str):
if peer_id in ("self", "me"):
return InputPeerSelf()
else:
if type(chat_id) is str:
chat_id = chat_id.lower().strip("@")
if type(peer_id) is str:
peer_id = peer_id.lower().strip("@")
try:
return self.peers_by_username[chat_id]
return self.peers_by_username[peer_id]
except KeyError:
return self.resolve_username(chat_id)
return self.resolve_username(peer_id)
else:
try:
return self.peers_by_id[chat_id]
return self.peers_by_id[peer_id]
except KeyError:
try:
return self.peers_by_id[int("-100" + str(peer_id))]
except KeyError:
raise PeerIdInvalid