Add message entitiy type hint

z44d · Sep 27, 2024 · a6b163e · a6b163e
1 parent bca0322
commit a6b163e
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 1 deletion.
diff --git a/tgram/types/__init__.py b/tgram/types/__init__.py
@@ -285,6 +285,27 @@
 
 ParseMode = _Literal["Markdown", "MarkdownV2", "HTML"]
 ChatType = _Literal["private", "group", "supergroup", "channel"]
+MessageEntityType = _Literal[
+    "mention",
+    "hashtag",
+    "cashtag",
+    "bot_command",
+    "url",
+    "email",
+    "phone_number",
+    "bold",
+    "italic",
+    "underline",
+    "strikethrough",
+    "spoiler",
+    "blockquote",
+    "expandable_blockquote",
+    "code",
+    "pre",
+    "text_link",
+    "text_mention",
+    "custom_emoji",
+]
 
 __all__ = [
     "Animation",
@@ -513,4 +534,5 @@
     "WriteAccessAllowed",
     "ParseMode",
     "ChatType",
+    "MessageEntityType",
 ]
diff --git a/tgram/types/_message_entity.py b/tgram/types/_message_entity.py
@@ -43,7 +43,7 @@ class MessageEntity(Type_):
 
     def __init__(
         self,
-        type: "str" = None,
+        type: "tgram.types.MessageEntityType" = None,
         offset: "int" = None,
         length: "int" = None,
         url: "str" = None,