LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2024-11-22 17:36:34 +03:00
Code Issues Packages Projects Releases Wiki Activity

Handle more cases on .get_display_name

Browse Source
This commit is contained in:
feodoran 2017-06-16 09:11:49 +02:00 committed by Lonami
parent 86d45cc276
commit 13199f737e
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
telethon/utils.py
View File

@ -15,9 +15,14 @@ def get_display_name(entity):
"""Gets the input peer for the given "entity" (user, chat or channel)
Returns None if it was not found"""
if isinstance(entity, User):
if entity.last_name is not None:
if entity.last_name and entity.first_name:
return '{} {}'.format(entity.first_name, entity.last_name)
return entity.first_name
elif entity.first_name:
return entity.first_name
elif entity.last_name:
return entity.last_name
else:
return '(No name)'
if isinstance(entity, Chat) or isinstance(entity, Channel):
return entity.title