From dfc334ef2016b06974ad39b9fcd11b14b438fcfe Mon Sep 17 00:00:00 2001 From: Suren Khorenyan Date: Sun, 11 Aug 2019 23:45:31 +0300 Subject: [PATCH] Minor refactor MessageEntity#parse --- aiogram/types/message_entity.py | 30 ++++++++++++------------------ 1 file changed, 12 insertions(+), 18 deletions(-) diff --git a/aiogram/types/message_entity.py b/aiogram/types/message_entity.py index 8b2e62d4..b9a9103e 100644 --- a/aiogram/types/message_entity.py +++ b/aiogram/types/message_entity.py @@ -49,29 +49,23 @@ class MessageEntity(base.TelegramObject): entity_text = self.get_text(text) if self.type == MessageEntityType.BOLD: - if as_html: - return markdown.hbold(entity_text) - return markdown.bold(entity_text) + method = markdown.hbold if as_html else markdown.bold + return method(entity_text) if self.type == MessageEntityType.ITALIC: - if as_html: - return markdown.hitalic(entity_text) - return markdown.italic(entity_text) + method = markdown.hitalic if as_html else markdown.italic + return method(entity_text) if self.type == MessageEntityType.PRE: - if as_html: - return markdown.hpre(entity_text) - return markdown.pre(entity_text) + method = markdown.hpre if as_html else markdown.pre + return method(entity_text) if self.type == MessageEntityType.CODE: - if as_html: - return markdown.hcode(entity_text) - return markdown.code(entity_text) + method = markdown.hcode if as_html else markdown.code + return method(entity_text) if self.type == MessageEntityType.URL: - if as_html: - return markdown.hlink(entity_text, entity_text) - return markdown.link(entity_text, entity_text) + method = markdown.hlink if as_html else markdown.link + return method(entity_text, entity_text) if self.type == MessageEntityType.TEXT_LINK: - if as_html: - return markdown.hlink(entity_text, self.url) - return markdown.link(entity_text, self.url) + method = markdown.hlink if as_html else markdown.link + return method(entity_text, self.url) if self.type == MessageEntityType.TEXT_MENTION and self.user: return self.user.get_mention(entity_text) return entity_text