82 lines
2.6 KiB
Python
82 lines
2.6 KiB
Python
# Pyrogram - Telegram MTProto API Client Library for Python
|
|
# Copyright (C) 2017-2019 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 as RawRPCError
|
|
from .exceptions.all import exceptions
|
|
|
|
|
|
class RPCError(Exception):
|
|
"""This is the base exception class for all Telegram API related errors.
|
|
For a finer grained control, see the specific errors below.
|
|
"""
|
|
ID = None
|
|
CODE = None
|
|
NAME = None
|
|
MESSAGE = None
|
|
|
|
def __init__(self, x: int or RawRPCError = None, query_type: type = None):
|
|
super().__init__("[{} {}]: {}".format(
|
|
self.CODE,
|
|
self.ID or self.NAME,
|
|
str(self) or 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", encoding="utf-8") as f:
|
|
f.write("{}\t{}\t{}\n".format(x.error_code, x.error_message, query_type))
|
|
|
|
@staticmethod
|
|
def raise_it(rpc_error: RawRPCError, query_type: type):
|
|
code = rpc_error.error_code
|
|
|
|
if code not in exceptions:
|
|
raise UnknownError(x=rpc_error, query_type=query_type)
|
|
|
|
message = rpc_error.error_message
|
|
id = re.sub(r"_\d+", "_X", message)
|
|
|
|
if id not in exceptions[code]:
|
|
raise UnknownError(x=rpc_error, query_type=query_type)
|
|
|
|
x = re.search(r"_(\d+)", message)
|
|
x = x.group(1) if x is not None else x
|
|
|
|
raise getattr(
|
|
import_module("pyrogram.errors"),
|
|
exceptions[code][id]
|
|
)(x=x)
|
|
|
|
|
|
class UnknownError(RPCError):
|
|
"""This object represents an Unknown Error, that is, an error which
|
|
Pyrogram does not know anything about, yet.
|
|
"""
|
|
CODE = 520
|
|
""":obj:`int`: Error code"""
|
|
NAME = "Unknown error"
|
|
MESSAGE = "{x}"
|