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

Improve parsing of Chat

Browse Source
This commit is contained in:
Dan 2021-12-08 07:46:41 +01:00
parent 8a08dcb92a
commit a73f990667
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
pyrogram/types/user_and_chats/chat.py
View File

@ -245,16 +245,17 @@ class Chat(Object):
chats: dict, chats: dict,
is_chat: bool is_chat: bool
) -> "Chat": ) -> "Chat":
from_id = utils.get_raw_peer_id(message.from_id)
peer_id = utils.get_raw_peer_id(message.peer_id)
chat_id = (peer_id or from_id) if is_chat else (from_id or peer_id)
if isinstance(message.peer_id, raw.types.PeerUser): if isinstance(message.peer_id, raw.types.PeerUser):
user_id = message.peer_id.user_id if is_chat else message.from_id.user_id return Chat._parse_user_chat(client, users[chat_id])
return Chat._parse_user_chat(client, users[user_id])
if isinstance(message.peer_id, raw.types.PeerChat): if isinstance(message.peer_id, raw.types.PeerChat):
chat_id = message.peer_id.chat_id if is_chat else message.from_id.chat_id
return Chat._parse_chat_chat(client, chats[chat_id]) return Chat._parse_chat_chat(client, chats[chat_id])
channel_id = message.peer_id.channel_id if is_chat else message.from_id.channel_id return Chat._parse_channel_chat(client, chats[chat_id])
return Chat._parse_channel_chat(client, chats[channel_id])
@staticmethod @staticmethod
def _parse_dialog(client, peer, users: dict, chats: dict): def _parse_dialog(client, peer, users: dict, chats: dict):