Add error package

pyrogram/api/errors/__init__.py

@@ -0,0 +1,19 @@
+# Pyrogram - Telegram MTProto API Client Library for Python
+# Copyright (C) 2017 Dan Tès <https://github.com/delivrance>
+#
+# This file is part of Pyrogram.
+#
+# Pyrogram is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Lesser General Public License as published
+# by the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# Pyrogram is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Lesser General Public License for more details.
+#
+# 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 .exceptions import *

pyrogram/api/errors/error.py

@@ -0,0 +1,71 @@
+# Pyrogram - Telegram MTProto API Client Library for Python
+# Copyright (C) 2017 Dan Tès <https://github.com/delivrance>
+#
+# This file is part of Pyrogram.
+#
+# Pyrogram is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Lesser General Public License as published
+# by the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# Pyrogram is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public License
+# along with Pyrogram.  If not, see <http://www.gnu.org/licenses/>.
+
+import re
+from importlib import import_module
+
+from pyrogram.api.types import RpcError
+
+from .exceptions.all import exceptions
+
+
+class Error(Exception):
+    ID = None
+    CODE = None
+    NAME = None
+    MESSAGE = None
+
+    def __init__(self, x: int or RpcError = None, query_type: type = None):
+        super().__init__("[{} {}]: {}".format(self.CODE, self.ID or self.NAME, self.MESSAGE.format(x=x)))
+
+        try:
+            self.x = int(x)
+        except (ValueError, TypeError):
+            self.x = x
+
+        # TODO: Proper log unknown errors
+        if self.CODE == 520:
+            with open("unknown_errors.txt", "a") as f:
+                f.write("{}\t{}\t{}\n".format(x.error_code, x.error_message, query_type))
+
+    @staticmethod
+    def raise_it(rpc_error: RpcError, query_type: type):
+        code = rpc_error.error_code
+
+        if code not in exceptions:
+            raise UnknownError(rpc_error, query_type)
+
+        message = rpc_error.error_message
+        id = re.sub(r"_\d+", "_X", message)
+
+        if id not in exceptions[code]:
+            raise UnknownError(rpc_error, query_type)
+
+        x = re.search(r"_(\d+)", message)
+        x = x.group(1) if x is not None else x
+
+        raise getattr(
+            import_module("pyrogram.api.errors"),
+            exceptions[code][id]
+        )(x)
+
+
+class UnknownError(Error):
+    CODE = 520
+    NAME = "Unknown error"
+    MESSAGE = "{x}"