mir/pyrogram
2
0
Fork 0
mirror of https://github.com/pyrogram/pyrogram synced 2025-08-28 21:07:59 +00:00
Code Issues Releases Wiki Activity

Add resolve_username method

Browse Source
This commit is contained in:
Dan 2018-01-25 16:41:59 +01:00
parent 894a0d0369
commit 16c5e63c8b
1 changed files with 41 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
pyrogram/client/client.py
View File

@ -513,16 +513,49 @@ class Client:
peer_username = peer_username.lower()
self.peers_by_username[peer_username] = input_peer
def resolve_username(self, username: str):
username = username.lower().strip("@")
resolved_peer = self.send(
functions.contacts.ResolveUsername(
username=username
)
) # type: types.contacts.ResolvedPeer
if type(resolved_peer.peer) is PeerUser:
input_peer = InputPeerUser(
user_id=resolved_peer.users[0].id,
access_hash=resolved_peer.users[0].access_hash
)
chat_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
else:
raise PeerIdInvalid
self.peers_by_username[username] = input_peer
self.peers_by_id[chat_id] = input_peer
return input_peer
def resolve_peer(self, chat_id: int or str):
if chat_id in ("self", "me"):
return InputPeerSelf()
else:
if type(chat_id) is str:
chat_id = chat_id.lower().strip("@")
try:
return (
self.peers_by_username[chat_id.lower().strip("@")]
if isinstance(chat_id, str)
else self.peers_by_id[chat_id]
)
return self.peers_by_username[chat_id]
except KeyError:
return self.resolve_username(chat_id)
else:
try:
return self.peers_by_id[chat_id]
except KeyError:
raise PeerIdInvalid