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

Handle text_mentions

Browse Source
This commit is contained in:
Dan 2018-04-05 20:59:41 +02:00
parent f3fcfb17df
commit 055d1c80ea
1 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
pyrogram/client/utils.py
View File

@ -15,11 +15,11 @@ ENTITIES = {
types.MessageEntityCode.ID: "code", types.MessageEntityCode.ID: "code",
types.MessageEntityPre.ID: "pre", types.MessageEntityPre.ID: "pre",
types.MessageEntityTextUrl.ID: "text_link", types.MessageEntityTextUrl.ID: "text_link",
# TODO: text_mention types.MessageEntityMentionName.ID: "text_mention"
} }
def parse_entities(entities: list) -> list: def parse_entities(entities: list, users: dict) -> list:
output_entities = [] output_entities = []
for entity in entities: for entity in entities:
@ -30,7 +30,13 @@ def parse_entities(entities: list) -> list:
type=entity_type, type=entity_type,
offset=entity.offset, offset=entity.offset,
length=entity.length, length=entity.length,
url=getattr(entity, "url", None) url=getattr(entity, "url", None),
user=parse_user(
users.get(
getattr(entity, "user_id", None),
None
)
)
)) ))
return output_entities return output_entities
@ -121,7 +127,7 @@ def parse_message(
chats: dict, chats: dict,
replies: int = 1 replies: int = 1
) -> pyrogram.Message: ) -> pyrogram.Message:
entities = parse_entities(message.entities) entities = parse_entities(message.entities, users)
forward_from = None forward_from = None
forward_from_chat = None forward_from_chat = None