add Chat type hint

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

Hide file tree

Showing 4 changed files with 5 additions and 3 deletions.
diff --git a/tgram/types/__init__.py b/tgram/types/__init__.py
@@ -284,6 +284,7 @@
 ]
 
 ParseMode = _Literal["Markdown", "MarkdownV2", "HTML"]
+ChatType = _Literal["private", "group", "supergroup", "channel"]
 
 __all__ = [
     "Animation",
@@ -511,4 +512,5 @@
     "WebhookInfo",
     "WriteAccessAllowed",
     "ParseMode",
+    "ChatType",
 ]
diff --git a/tgram/types/_chat.py b/tgram/types/_chat.py
@@ -18,7 +18,7 @@ class Chat(Type_, bound.ChatB):
     def __init__(
         self,
         id: "int" = None,
-        type: "str" = None,
+        type: "tgram.types.ChatType" = None,
         title: "str" = None,
         username: "str" = None,
         first_name: "str" = None,

diff --git a/tgram/types/_chat_full_info.py b/tgram/types/_chat_full_info.py
@@ -154,7 +154,7 @@ class ChatFullInfo(Type_, bound.ChatB):
     def __init__(
         self,
         id: "int" = None,
-        type: "str" = None,
+        type: "tgram.types.ChatType" = None,
         accent_color_id: "int" = None,
         max_reaction_count: "int" = None,
         title: "str" = None,

diff --git a/tgram/types/_inline_query.py b/tgram/types/_inline_query.py
@@ -42,7 +42,7 @@ def __init__(
         from_user: "tgram.types.User" = None,
         query: "str" = None,
         offset: "str" = None,
-        chat_type: "str" = None,
+        chat_type: "tgram.types.ChatType" = None,
         location: "tgram.types.Location" = None,
         me: "tgram.TgBot" = None,
         json: "dict" = None,