Add ChatMemberStatus hints

z44d · Oct 9, 2024 · c0581e1 · c0581e1
1 parent 9e31806
commit c0581e1
Show file tree

Hide file tree

Showing 7 changed files with 9 additions and 6 deletions.
diff --git a/tgram/types/__init__.py b/tgram/types/__init__.py
@@ -285,6 +285,9 @@
 
 ParseMode = _Literal["Markdown", "MarkdownV2", "HTML"]
 ChatType = _Literal["private", "group", "supergroup", "channel"]
+ChatMemberStatus = _Literal[
+    "creator", "administrator", "member", "restricted", "left", "kicked"
+]
 MessageEntityType = _Literal[
     "mention",
     "hashtag",

diff --git a/tgram/types/_chat_member_administrator.py b/tgram/types/_chat_member_administrator.py
@@ -79,7 +79,7 @@ class ChatMemberAdministrator(Type_):
 
     def __init__(
         self,
-        status: "str" = None,
+        status: "tgram.types.ChatMemberStatus" = None,
         user: "tgram.types.User" = None,
         can_be_edited: "bool" = None,
         is_anonymous: "bool" = None,

diff --git a/tgram/types/_chat_member_banned.py b/tgram/types/_chat_member_banned.py
@@ -26,7 +26,7 @@ class ChatMemberBanned(Type_):
 
     def __init__(
         self,
-        status: "str" = None,
+        status: "tgram.types.ChatMemberStatus" = None,
         user: "tgram.types.User" = None,
         until_date: "int" = None,
         me: "tgram.TgBot" = None,

diff --git a/tgram/types/_chat_member_left.py b/tgram/types/_chat_member_left.py
@@ -22,7 +22,7 @@ class ChatMemberLeft(Type_):
 
     def __init__(
         self,
-        status: "str" = None,
+        status: "tgram.types.ChatMemberStatus" = None,
         user: "tgram.types.User" = None,
         me: "tgram.TgBot" = None,
         json: "dict" = None,

diff --git a/tgram/types/_chat_member_member.py b/tgram/types/_chat_member_member.py
@@ -25,7 +25,7 @@ class ChatMemberMember(Type_):
 
     def __init__(
         self,
-        status: "str" = None,
+        status: "tgram.types.ChatMemberStatus" = None,
         user: "tgram.types.User" = None,
         until_date: "int" = None,
         me: "tgram.TgBot" = None,

diff --git a/tgram/types/_chat_member_owner.py b/tgram/types/_chat_member_owner.py
@@ -28,7 +28,7 @@ class ChatMemberOwner(Type_):
 
     def __init__(
         self,
-        status: "str" = None,
+        status: "tgram.types.ChatMemberStatus" = None,
         user: "tgram.types.User" = None,
         is_anonymous: "bool" = None,
         custom_title: "str" = None,

diff --git a/tgram/types/_chat_member_restricted.py b/tgram/types/_chat_member_restricted.py
@@ -72,7 +72,7 @@ class ChatMemberRestricted(Type_):
 
     def __init__(
         self,
-        status: "str" = None,
+        status: "tgram.types.ChatMemberStatus" = None,
         user: "tgram.types.User" = None,
         is_member: "bool" = None,
         can_send_messages: "bool" = None,