LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-08-07 13:34:59 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add name mention formating to HTML and Markdown

Browse Source
This commit is contained in:
Painor 2018-10-03 15:08:23 +01:00
parent c2966297f1
commit 0dc890e371
2 changed files with 17 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
telethon/extensions/html.py
View File

@ -9,8 +9,8 @@ from html.parser import HTMLParser
from ..tl.types import (
MessageEntityBold, MessageEntityItalic, MessageEntityCode,
MessageEntityPre, MessageEntityEmail, MessageEntityUrl,
MessageEntityTextUrl
)
MessageEntityTextUrl,
MessageEntityMentionName)
# Helpers from markdown.py
@ -178,6 +178,9 @@ def unparse(text, entities):
elif entity_type == MessageEntityTextUrl:
html.append('<a href="{}">{}</a>'
.format(escape(entity.url), entity_text))
elif entity_type == MessageEntityMentionName:
html.append('<a href="{}">{}</a>'
.format(entity.user_id, entity_text))
else:
skip_entity = True
last_offset = entity.offset + (0 if skip_entity else entity.length)

9
telethon/extensions/markdown.py
View File

@ -9,8 +9,8 @@ from ..helpers import add_surrogate, del_surrogate
from ..tl import TLObject
from ..tl.types import (
MessageEntityBold, MessageEntityItalic, MessageEntityCode,
MessageEntityPre, MessageEntityTextUrl
)
MessageEntityPre, MessageEntityTextUrl,
MessageEntityMentionName)
DEFAULT_DELIMITERS = {
'**': MessageEntityBold,
@ -167,5 +167,8 @@ def unparse(text, entities, delimiters=None, url_fmt=None):
add_surrogate(url_fmt.format(text[s:e], entity.url)) +
text[e:]
)
elif isinstance(entity, MessageEntityMentionName):
text = text[:entity.offset] + "[{}](tg://user?id={})".format(
text[entity.offset:entity.offset + entity.length], entity.user_id) + text[
entity.offset + entity.length:]
return del_surrogate(text)