chore(error-handler): some doc and typing hints improvements

alissonlauffer · alissonlauffer · commit 2b4d6bcdb723 · 2024-06-30T17:38:42.000-03:00
diff --git a/hydrogram/handlers/error_handler.py b/hydrogram/handlers/error_handler.py
@@ -16,6 +16,9 @@
 #  You should have received a copy of the GNU Lesser General Public License
 #  along with Hydrogram.  If not, see <http://www.gnu.org/licenses/>.
 
+from __future__ import annotations
+
+from collections.abc import Iterable
 from typing import TYPE_CHECKING, Callable
 
 from .handler import Handler
@@ -29,38 +32,40 @@ class ErrorHandler(Handler):
     It is intended to be used with :meth:`~hydrogram.Client.add_handler`
 
     For a nicer way to register this handler, have a look at the
-    :meth:`~hydrogram.Client.on_message` decorator.
+    :meth:`~hydrogram.Client.on_error` decorator.
 
     Parameters:
         callback (``Callable``):
             Pass a function that will be called when a new Error arrives. It takes *(client, error)*
             as positional arguments (look at the section below for a detailed description).
 
-        errors (:obj:`Exception` | List of :obj:`Exception`):
+        errors (``Exception`` | Iterable of ``Exception``, *optional*):
             Pass one or more exception classes to allow only a subset of errors to be passed
             in your callback function.
 
     Other parameters:
         client (:obj:`~hydrogram.Client`):
-            The Client itself, useful when you want to call other API methods inside the message handler.
+            The Client itself, useful when you want to call other API methods inside the error handler.
 
-        error (:obj:`~Exception`):
+        error (``Exception``):
             The error that was raised.
 
         update (:obj:`~hydrogram.Update`):
             The update that caused the error.
     """
 
-    def __init__(self, callback: Callable, errors=None):
+    def __init__(
+        self, callback: Callable, errors: type[Exception] | Iterable[type[Exception]] | None = None
+    ):
         if errors is None:
             errors = [Exception]
-        elif not isinstance(errors, list):
+        elif not isinstance(errors, Iterable):
             errors = [errors]
 
         self.errors = errors
         super().__init__(callback)
 
-    async def check(self, client: "hydrogram.Client", error: Exception):
+    async def check(self, client: hydrogram.Client, error: Exception):
         return any(isinstance(error, e) for e in self.errors)
 
     def check_remove(self, error: Exception):
diff --git a/hydrogram/methods/utilities/remove_error_handler.py b/hydrogram/methods/utilities/remove_error_handler.py
@@ -21,17 +21,19 @@
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
+    from collections.abc import Iterable
+
     import hydrogram
 
 
 class RemoveErrorHandler:
     def remove_error_handler(
-        self: hydrogram.Client, error: Exception | tuple[Exception] = Exception
+        self: hydrogram.Client, error: type[Exception] | Iterable[type[Exception]] = Exception
     ):
         """Remove a previously-registered error handler. (using exception classes)
 
         Parameters:
-            error (``Exception``):
+            error (``Exception`` | Iterable of ``Exception``, *optional*):
                 The error(s) for handlers to be removed.
         """
         for handler in self.dispatcher.error_handlers:
