72 lines
2.2 KiB
Python
72 lines
2.2 KiB
Python
|
# 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}"
|