From 7c40b974bb7ce933674ba8dfa2162ed827205423 Mon Sep 17 00:00:00 2001 From: Alex Root Junior Date: Sat, 27 Jan 2018 05:26:51 +0200 Subject: [PATCH] Parse text mention from entities. --- aiogram/types/message_entity.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/aiogram/types/message_entity.py b/aiogram/types/message_entity.py index 76f51327..24e6da5f 100644 --- a/aiogram/types/message_entity.py +++ b/aiogram/types/message_entity.py @@ -40,6 +40,8 @@ class MessageEntity(base.TelegramObject): return self._apply(text, lambda url: markdown.link(url, url)) elif self.type == MessageEntityType.TEXT_LINK: return self._apply(text, lambda url: markdown.link(url, self.url)) + if self.type == MessageEntityType.TEXT_MENTION and self.user: + return self._apply(text, lambda name: self.user.get_mention(name, as_html=False)) return text def apply_html(self, text): @@ -61,6 +63,8 @@ class MessageEntity(base.TelegramObject): return self._apply(text, lambda url: markdown.hlink(url, url)) elif self.type == MessageEntityType.TEXT_LINK: return self._apply(text, lambda url: markdown.hlink(url, self.url)) + if self.type == MessageEntityType.TEXT_MENTION and self.user: + return self._apply(text, lambda name: self.user.get_mention(name, as_html=True)) return text