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

Some more changed in order to enhance UserStatus API design

Browse Source
This commit is contained in:
Dan 2018-10-15 10:07:46 +02:00
parent b2b599e211
commit 79a9ddfab5
1 changed files with 3 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
pyrogram/client/ext/utils.py
View File

@ -129,15 +129,11 @@ def parse_chat_photo(photo):
) )
def parse_update_user_status(update: types.UpdateUserStatus) -> pyrogram_types.User: def parse_user_status(user_status, user_id: int = None, is_bot: bool = False) -> pyrogram_types.UserStatus or None:
return pyrogram_types.User(id=update.user_id, status=parse_user_status(update.status))
def parse_user_status(user_status, is_bot: bool = False) -> pyrogram_types.UserStatus or None:
if is_bot: if is_bot:
return None return None
status = pyrogram_types.UserStatus() status = pyrogram_types.UserStatus(user_id)
if isinstance(user_status, types.UserStatusOnline): if isinstance(user_status, types.UserStatusOnline):
status.online = True status.online = True
@ -171,7 +167,7 @@ def parse_user(user: types.User) -> pyrogram_types.User or None:
language_code=user.lang_code, language_code=user.lang_code,
phone_number=user.phone, phone_number=user.phone,
photo=parse_chat_photo(user.photo), photo=parse_chat_photo(user.photo),
status=parse_user_status(user.status, user.bot), status=parse_user_status(user.status, is_bot=user.bot),
) if user else None ) if user else None