mir/pyrogram
2
0
Fork 0
mirror of https://github.com/pyrogram/pyrogram synced 2025-08-27 20:37:54 +00:00
Code Issues Releases Wiki Activity

Add ChatPhoto to User objects

Browse Source
This commit is contained in:
Dan 2018-04-15 22:28:08 +02:00
parent d948831b64
commit d9fdd89c49
1 changed files with 22 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
pyrogram/client/message_parser.py
View File

@ -62,27 +62,6 @@ def parse_entities(entities: list, users: dict) -> list:
return output_entities return output_entities
def parse_user(user: types.User) -> pyrogram.User or None:
return pyrogram.User(
id=user.id,
is_bot=user.bot,
first_name=user.first_name,
last_name=user.last_name,
username=user.username,
language_code=user.lang_code,
phone_number=user.phone
) if user else None
def parse_chat(message: types.Message, users: dict, chats: dict) -> pyrogram.Chat:
if isinstance(message.to_id, types.PeerUser):
return parse_user_chat(users[message.to_id.user_id if message.out else message.from_id])
elif isinstance(message.to_id, types.PeerChat):
return parse_chat_chat(chats[message.to_id.chat_id])
else:
return parse_channel_chat(chats[message.to_id.channel_id])
def parse_chat_photo(photo): def parse_chat_photo(photo):
if not isinstance(photo, (types.UserProfilePhoto, types.ChatPhoto)): if not isinstance(photo, (types.UserProfilePhoto, types.ChatPhoto)):
return None return None
@ -112,6 +91,28 @@ def parse_chat_photo(photo):
) )
def parse_user(user: types.User) -> pyrogram.User or None:
return pyrogram.User(
id=user.id,
is_bot=user.bot,
first_name=user.first_name,
last_name=user.last_name,
username=user.username,
language_code=user.lang_code,
phone_number=user.phone,
photo=parse_chat_photo(user.photo)
) if user else None
def parse_chat(message: types.Message, users: dict, chats: dict) -> pyrogram.Chat:
if isinstance(message.to_id, types.PeerUser):
return parse_user_chat(users[message.to_id.user_id if message.out else message.from_id])
elif isinstance(message.to_id, types.PeerChat):
return parse_chat_chat(chats[message.to_id.chat_id])
else:
return parse_channel_chat(chats[message.to_id.channel_id])
def parse_user_chat(user: types.User) -> pyrogram.Chat: def parse_user_chat(user: types.User) -> pyrogram.Chat:
return pyrogram.Chat( return pyrogram.Chat(
id=user.id, id=user.id,