Generalize the URL branch

This commit is contained in:
Lonami 2018-10-04 15:50:13 +02:00 committed by GitHub
parent 2c4a8b0354
commit 6ad2f920de
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -161,13 +161,17 @@ def unparse(text, entities, delimiters=None, url_fmt=None):
delimiter = delimiters.get(type(entity), None)
if delimiter:
text = text[:s] + delimiter + text[s:e] + delimiter + text[e:]
elif isinstance(entity, MessageEntityTextUrl) and url_fmt:
text = (
text[:s] +
add_surrogate(url_fmt.format(text[s:e], entity.url)) +
text[e:]
)
elif isinstance(entity, MessageEntityMentionName):
text = text[:s] + "[{}](tg://user?id={})".format(
text[s:e], entity.user_id) + text[e:]
elif url_fmt:
url = None
if isinstance(entity, MessageEntityTextUrl):
url = entity.url
elif isinstance(entity, MessageEntityMentionName):
url = 'tg://user?id={}'.format(entity.user_id)
if url:
text = (
text[:s] +
add_surrogate(url_fmt.format(text[s:e], url)) +
text[e:]
)
return del_surrogate(text)