Fix Telethon 0.18.1 compatibility. Fixes #96

mautrix_telegram/formatter/from_matrix.py

@@ -22,20 +22,12 @@ import math
 import re
 import logging
 
-from telethon_aio.tl.types import (MessageEntityMention,
-                                   InputMessageEntityMentionName, MessageEntityEmail,
-                                   MessageEntityUrl, MessageEntityTextUrl, MessageEntityBold,
-                                   MessageEntityItalic, MessageEntityCode, MessageEntityPre,
-                                   MessageEntityBotCommand, MessageEntityHashtag,
-                                   MessageEntityMentionName, InputUser)
-
-try:
-    from telethon_aio.tl.types import TypeMessageEntity
-except ImportError:
-    TypeMessageEntity = Union[
-        MessageEntityMention, MessageEntityHashtag, MessageEntityBotCommand, MessageEntityUrl,
-        MessageEntityEmail, MessageEntityBold, MessageEntityItalic, MessageEntityCode,
-        MessageEntityPre, MessageEntityTextUrl, MessageEntityMentionName]
+from telethon.tl.types import (MessageEntityMention,
+                               InputMessageEntityMentionName, MessageEntityEmail,
+                               MessageEntityUrl, MessageEntityTextUrl, MessageEntityBold,
+                               MessageEntityItalic, MessageEntityCode, MessageEntityPre,
+                               MessageEntityBotCommand, MessageEntityHashtag,
+                               MessageEntityMentionName, InputUser, TypeMessageEntity)
 
 from ..context import Context
 from .. import user as u, puppet as pu, portal as po

mautrix_telegram/formatter/from_telegram.py

@@ -24,19 +24,11 @@ except ImportError:
 import logging
 import re
 
-from telethon_aio.tl.types import (MessageEntityMention, MessageEntityMentionName,
-                                   MessageEntityEmail, MessageEntityUrl, MessageEntityTextUrl,
-                                   MessageEntityBold, MessageEntityItalic, MessageEntityCode,
-                                   MessageEntityPre, MessageEntityBotCommand, Message, PeerChannel,
-                                   MessageEntityHashtag)
-
-try:
-    from telethon_aio.tl.types import TypeMessageEntity
-except ImportError:
-    TypeMessageEntity = Union[
-        MessageEntityMention, MessageEntityHashtag, MessageEntityBotCommand, MessageEntityUrl,
-        MessageEntityEmail, MessageEntityBold, MessageEntityItalic, MessageEntityCode,
-        MessageEntityPre, MessageEntityTextUrl, MessageEntityMentionName]
+from telethon.tl.types import (MessageEntityMention, MessageEntityMentionName,
+                               MessageEntityEmail, MessageEntityUrl, MessageEntityTextUrl,
+                               MessageEntityBold, MessageEntityItalic, MessageEntityCode,
+                               MessageEntityPre, MessageEntityBotCommand, Message, PeerChannel,
+                               MessageEntityHashtag, TypeMessageEntity)
 
 from mautrix_appservice import MatrixRequestError
 from mautrix_appservice.intent_api import IntentAPI