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

alissonlauffer · alissonlauffer · commit 8253923ad359 · 2024-06-30T17:22:16.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,20 +32,20 @@ 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 (:obj:`Exception` | List of :obj:`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`):
             The error that was raised.
@@ -51,16 +54,18 @@ class ErrorHandler(Handler):
             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):
