refactor(error_handler): enhanced code

HitaloM · HitaloM · commit ced23aa6ad81 · 2024-07-18T20:11:23.000-03:00
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -11,7 +11,7 @@ repos:
       - id: check-yaml
 
   - repo: https://github.com/astral-sh/ruff-pre-commit
-    rev: v0.5.1
+    rev: v0.5.3
     hooks:
       - id: ruff-format
       - id: ruff
diff --git a/docs/source/api/decorators.rst b/docs/source/api/decorators.rst
@@ -61,3 +61,4 @@ Details
 .. autodecorator:: hydrogram.Client.on_poll()
 .. autodecorator:: hydrogram.Client.on_disconnect()
 .. autodecorator:: hydrogram.Client.on_raw_update()
+.. autodecorator:: hydrogram.Client.on_error()
diff --git a/hydrogram/crypto/aes.py b/hydrogram/crypto/aes.py
@@ -50,6 +50,7 @@ def xor(a: bytes, b: bytes) -> bytes:
             len(a),
             "big",
         )
+
 except ImportError:
     import pyaes
 
diff --git a/hydrogram/handlers/error_handler.py b/hydrogram/handlers/error_handler.py
@@ -25,6 +25,7 @@
 
 if TYPE_CHECKING:
     import hydrogram
+    from hydrogram.types import Update
 
 
 class ErrorHandler(Handler):
@@ -47,26 +48,29 @@ class ErrorHandler(Handler):
         client (:obj:`~hydrogram.Client`):
             The Client itself, useful when you want to call other API methods inside the error handler.
 
-        error (``Exception``):
-            The error that was raised.
-
         update (:obj:`~hydrogram.Update`):
             The update that caused the error.
+
+        error (``Exception``):
+            The error that was raised.
     """
 
     def __init__(
-        self, callback: Callable, errors: type[Exception] | Iterable[type[Exception]] | None = None
+        self,
+        callback: Callable,
+        errors: type[Exception] | Iterable[type[Exception]] | None = None,
     ):
-        if errors is None:
-            errors = [Exception]
-        elif not isinstance(errors, Iterable):
-            errors = [errors]
-
-        self.errors = errors
+        self.errors = (
+            tuple(errors)
+            if isinstance(errors, Iterable)
+            else (errors,)
+            if errors
+            else (Exception,)
+        )
         super().__init__(callback)
 
-    async def check(self, client: hydrogram.Client, error: Exception):
-        return any(isinstance(error, e) for e in self.errors)
+    async def check(self, client: hydrogram.Client, update: Update, error: Exception) -> bool:
+        return isinstance(error, self.errors)
 
-    def check_remove(self, error: Exception):
-        return self.errors == error or any(isinstance(error, e) for e in self.errors)
+    def check_remove(self, error: Exception) -> bool:
+        return isinstance(error, self.errors)
diff --git a/pyproject.toml b/pyproject.toml
@@ -51,7 +51,7 @@ build-backend = "hatchling.build"
 [tool.rye]
 managed = true
 dev-dependencies = [
-    "ruff>=0.5.1",
+    "ruff>=0.5.3",
     "pytest>=7.4.3",
     "pytest-asyncio>=0.23.2",
     "pytest-cov>=4.1.0",
@@ -61,8 +61,8 @@ dev-dependencies = [
 
 [project.optional-dependencies]
 docs = [
-    "Sphinx>=7.3.7",
-    "furo>=2024.5.6",
+    "Sphinx>=7.4.6",
+    "furo>=2024.7.18",
     "sphinx-autobuild>=2024.4.16",
     "sphinx-copybutton>=0.5.2",
     "pygments>=2.18.0",
diff --git a/tests/filters/test_command.py b/tests/filters/test_command.py
@@ -17,7 +17,6 @@
 #  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 hydrogram import filters
 from tests.filters import Client, Message
 

Original file line number	Diff line number	Diff line change
`@@ -50,6 +50,7 @@ def xor(a: bytes, b: bytes) -> bytes:`
`50`	`50`	`len(a),`
`51`	`51`	`"big",`
`52`	`52`	`)`
	`53`	`+`
`53`	`54`	`except ImportError:`
`54`	`55`	`import pyaes`
`55`	`56`