Add typehints to all of the decorators classes (hydrogram#34)

MarsBatya · web-flow · commit 9b1524dd8162 · 2024-03-22T20:50:49.000+03:00
diff --git a/pyrogram/methods/decorators/on_callback_query.py b/pyrogram/methods/decorators/on_callback_query.py
@@ -16,17 +16,17 @@
 #  You should have received a copy of the GNU Lesser General Public License
 #  along with Pyrogram.  If not, see <http://www.gnu.org/licenses/>.
 
-from typing import Callable
+from typing import Callable, Optional, Union
 
 import pyrogram
 from pyrogram.filters import Filter
 
 
 class OnCallbackQuery:
     def on_callback_query(
-        self=None,
-        filters=None,
-        group: int = 0
+        self: Union["OnCallbackQuery", Filter, None] = None,
+        filters: Optional[Filter] = None,
+        group: int = 0,
     ) -> Callable:
         """Decorator for handling callback queries.
 
diff --git a/pyrogram/methods/decorators/on_chat_join_request.py b/pyrogram/methods/decorators/on_chat_join_request.py
@@ -16,17 +16,17 @@
 #  You should have received a copy of the GNU Lesser General Public License
 #  along with Pyrogram.  If not, see <http://www.gnu.org/licenses/>.
 
-from typing import Callable
+from typing import Callable, Optional, Union
 
 import pyrogram
 from pyrogram.filters import Filter
 
 
 class OnChatJoinRequest:
     def on_chat_join_request(
-        self=None,
-        filters=None,
-        group: int = 0
+        self: Union["OnChatJoinRequest", Filter, None] = None,
+        filters: Optional[Filter] = None,
+        group: int = 0,
     ) -> Callable:
         """Decorator for handling chat join requests.
 
diff --git a/pyrogram/methods/decorators/on_chat_member_updated.py b/pyrogram/methods/decorators/on_chat_member_updated.py
@@ -16,17 +16,17 @@
 #  You should have received a copy of the GNU Lesser General Public License
 #  along with Pyrogram.  If not, see <http://www.gnu.org/licenses/>.
 
-from typing import Callable
+from typing import Callable, Optional, Union
 
 import pyrogram
 from pyrogram.filters import Filter
 
 
 class OnChatMemberUpdated:
     def on_chat_member_updated(
-        self=None,
-        filters=None,
-        group: int = 0
+        self: Union["OnChatMemberUpdated", Filter, None] = None,
+        filters: Optional[Filter] = None,
+        group: int = 0,
     ) -> Callable:
         """Decorator for handling event changes on chat members.
 
diff --git a/pyrogram/methods/decorators/on_chosen_inline_result.py b/pyrogram/methods/decorators/on_chosen_inline_result.py
@@ -16,17 +16,17 @@
 #  You should have received a copy of the GNU Lesser General Public License
 #  along with Pyrogram.  If not, see <http://www.gnu.org/licenses/>.
 
-from typing import Callable
+from typing import Callable, Optional, Union
 
 import pyrogram
 from pyrogram.filters import Filter
 
 
 class OnChosenInlineResult:
     def on_chosen_inline_result(
-        self=None,
-        filters=None,
-        group: int = 0
+        self: Union["OnChosenInlineResult", Filter, None] = None,
+        filters: Optional[Filter] = None,
+        group: int = 0,
     ) -> Callable:
         """Decorator for handling chosen inline results.
 
diff --git a/pyrogram/methods/decorators/on_deleted_messages.py b/pyrogram/methods/decorators/on_deleted_messages.py
@@ -16,17 +16,17 @@
 #  You should have received a copy of the GNU Lesser General Public License
 #  along with Pyrogram.  If not, see <http://www.gnu.org/licenses/>.
 
-from typing import Callable
+from typing import Callable, Optional, Union
 
 import pyrogram
 from pyrogram.filters import Filter
 
 
 class OnDeletedMessages:
     def on_deleted_messages(
-        self=None,
-        filters=None,
-        group: int = 0
+        self: Union["OnDeletedMessages", Filter, None] = None,
+        filters: Optional[Filter] = None,
+        group: int = 0,
     ) -> Callable:
         """Decorator for handling deleted messages.
 
diff --git a/pyrogram/methods/decorators/on_disconnect.py b/pyrogram/methods/decorators/on_disconnect.py
@@ -16,13 +16,13 @@
 #  You should have received a copy of the GNU Lesser General Public License
 #  along with Pyrogram.  If not, see <http://www.gnu.org/licenses/>.
 
-from typing import Callable
+from typing import Callable, Optional
 
 import pyrogram
 
 
 class OnDisconnect:
-    def on_disconnect(self=None) -> Callable:
+    def on_disconnect(self: Optional["OnDisconnect"] = None) -> Callable:
         """Decorator for handling disconnections.
 
         This does the same thing as :meth:`~pyrogram.Client.add_handler` using the
diff --git a/pyrogram/methods/decorators/on_edited_message.py b/pyrogram/methods/decorators/on_edited_message.py
@@ -16,17 +16,17 @@
 #  You should have received a copy of the GNU Lesser General Public License
 #  along with Pyrogram.  If not, see <http://www.gnu.org/licenses/>.
 
-from typing import Callable
+from typing import Callable, Optional, Union
 
 import pyrogram
 from pyrogram.filters import Filter
 
 
 class OnEditedMessage:
     def on_edited_message(
-        self=None,
-        filters=None,
-        group: int = 0
+        self: Union["OnEditedMessage", Filter, None] = None,
+        filters: Optional[Filter] = None,
+        group: int = 0,
     ) -> Callable:
         """Decorator for handling edited messages.
 
diff --git a/pyrogram/methods/decorators/on_inline_query.py b/pyrogram/methods/decorators/on_inline_query.py
@@ -16,17 +16,17 @@
 #  You should have received a copy of the GNU Lesser General Public License
 #  along with Pyrogram.  If not, see <http://www.gnu.org/licenses/>.
 
-from typing import Callable
+from typing import Callable, Optional, Union
 
 import pyrogram
 from pyrogram.filters import Filter
 
 
 class OnInlineQuery:
     def on_inline_query(
-        self=None,
-        filters=None,
-        group: int = 0
+        self: Union["OnInlineQuery", Filter, None] = None,
+        filters: Optional[Filter] = None,
+        group: int = 0,
     ) -> Callable:
         """Decorator for handling inline queries.
 
diff --git a/pyrogram/methods/decorators/on_message.py b/pyrogram/methods/decorators/on_message.py
@@ -16,17 +16,17 @@
 #  You should have received a copy of the GNU Lesser General Public License
 #  along with Pyrogram.  If not, see <http://www.gnu.org/licenses/>.
 
-from typing import Callable
+from typing import Callable, Optional, Union
 
 import pyrogram
 from pyrogram.filters import Filter
 
 
 class OnMessage:
     def on_message(
-        self=None,
-        filters=None,
-        group: int = 0
+        self: Union["OnMessage", Filter, None] = None,
+        filters: Optional[Filter] = None,
+        group: int = 0,
     ) -> Callable:
         """Decorator for handling new messages.
 
diff --git a/pyrogram/methods/decorators/on_poll.py b/pyrogram/methods/decorators/on_poll.py
@@ -16,17 +16,17 @@
 #  You should have received a copy of the GNU Lesser General Public License
 #  along with Pyrogram.  If not, see <http://www.gnu.org/licenses/>.
 
-from typing import Callable
+from typing import Callable, Optional, Union
 
 import pyrogram
 from pyrogram.filters import Filter
 
 
 class OnPoll:
     def on_poll(
-        self=None,
-        filters=None,
-        group: int = 0
+        self: Union["OnPoll", Filter, None] = None,
+        filters: Optional[Filter] = None,
+        group: int = 0,
     ) -> Callable:
         """Decorator for handling poll updates.
 
diff --git a/pyrogram/methods/decorators/on_raw_update.py b/pyrogram/methods/decorators/on_raw_update.py
@@ -16,15 +16,15 @@
 #  You should have received a copy of the GNU Lesser General Public License
 #  along with Pyrogram.  If not, see <http://www.gnu.org/licenses/>.
 
-from typing import Callable
+from typing import Callable, Optional
 
 import pyrogram
 
 
 class OnRawUpdate:
     def on_raw_update(
-        self=None,
-        group: int = 0
+        self: Optional["OnRawUpdate"] = None,
+        group: int = 0,
     ) -> Callable:
         """Decorator for handling raw updates.
 
diff --git a/pyrogram/methods/decorators/on_story.py b/pyrogram/methods/decorators/on_story.py
@@ -16,17 +16,17 @@
 #  You should have received a copy of the GNU Lesser General Public License
 #  along with Pyrogram.  If not, see <http://www.gnu.org/licenses/>.
 
-from typing import Callable
+from typing import Callable, Optional, Union
 
 import pyrogram
 from pyrogram.filters import Filter
 
 
 class OnStory:
     def on_story(
-        self=None,
-        filters=None,
-        group: int = 0
+        self: Union["OnStory", Filter, None] = None,
+        filters: Optional[Filter] = None,
+        group: int = 0,
     ) -> Callable:
         """Decorator for handling new stories.
 
diff --git a/pyrogram/methods/decorators/on_user_status.py b/pyrogram/methods/decorators/on_user_status.py
@@ -16,17 +16,17 @@
 #  You should have received a copy of the GNU Lesser General Public License
 #  along with Pyrogram.  If not, see <http://www.gnu.org/licenses/>.
 
-from typing import Callable
+from typing import Callable, Optional, Union
 
 import pyrogram
 from pyrogram.filters import Filter
 
 
 class OnUserStatus:
     def on_user_status(
-        self=None,
-        filters=None,
-        group: int = 0
+        self: Union["OnUserStatus", Filter, None] = None,
+        filters: Optional[Filter] = None,
+        group: int = 0,
     ) -> Callable:
         """Decorator for handling user status updates.
         This does the same thing as :meth:`~pyrogram.Client.add_handler` using the
