mir/pyrogram
2
0
Fork 0
mirror of https://github.com/pyrogram/pyrogram synced 2025-08-28 12:57:52 +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.MessageEntityPre.ID: "pre",
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 = []
for entity in entities:
@ -30,7 +30,13 @@ def parse_entities(entities: list) -> list:
type=entity_type,
offset=entity.offset,
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
@ -121,7 +127,7 @@ def parse_message(
chats: dict,
replies: int = 1
) -> pyrogram.Message:
entities = parse_entities(message.entities)
entities = parse_entities(message.entities, users)
forward_from = None
forward_from_chat = None