mirror of
https://github.com/LonamiWebs/Telethon.git
synced 2024-11-24 02:13:45 +03:00
89 lines
2.6 KiB
Python
89 lines
2.6 KiB
Python
"""
|
|
This module holds all the base and automatically generated errors that the
|
|
Telegram API has. See telethon_generator/errors.json for more.
|
|
"""
|
|
import urllib.request
|
|
import re
|
|
from threading import Thread
|
|
|
|
from .common import (
|
|
ReadCancelledError, TypeNotFoundError, InvalidChecksumError,
|
|
BrokenAuthKeyError, SecurityError, CdnFileTamperedError
|
|
)
|
|
|
|
# This imports the base errors too, as they're imported there
|
|
from .rpc_base_errors import *
|
|
from .rpc_error_list import *
|
|
|
|
|
|
def report_error(code, message, report_method):
|
|
"""
|
|
Reports an RPC error to pwrtelegram.
|
|
|
|
:param code: the integer code of the error (like 400).
|
|
:param message: the message representing the error.
|
|
:param report_method: the constructor ID of the function that caused it.
|
|
"""
|
|
try:
|
|
# Ensure it's signed
|
|
report_method = int.from_bytes(
|
|
report_method.to_bytes(4, 'big'), 'big', signed=True
|
|
)
|
|
url = urllib.request.urlopen(
|
|
'https://rpc.pwrtelegram.xyz?code={}&error={}&method={}'
|
|
.format(code, message, report_method),
|
|
timeout=5
|
|
)
|
|
url.read()
|
|
url.close()
|
|
except:
|
|
"We really don't want to crash when just reporting an error"
|
|
|
|
|
|
def rpc_message_to_error(code, message, report_method=None):
|
|
"""
|
|
Converts a Telegram's RPC Error to a Python error.
|
|
|
|
:param code: the integer code of the error (like 400).
|
|
:param message: the message representing the error.
|
|
:param report_method: if present, the ID of the method that caused it.
|
|
:return: the RPCError as a Python exception that represents this error.
|
|
"""
|
|
if report_method is not None:
|
|
Thread(
|
|
target=report_error,
|
|
args=(code, message, report_method)
|
|
).start()
|
|
|
|
# Try to get the error by direct look-up, otherwise regex
|
|
# TODO Maybe regexes could live in a separate dictionary?
|
|
cls = rpc_errors_all.get(message, None)
|
|
if cls:
|
|
return cls()
|
|
|
|
for msg_regex, cls in rpc_errors_all.items():
|
|
m = re.match(msg_regex, message)
|
|
if m:
|
|
capture = int(m.group(1)) if m.groups() else None
|
|
return cls(capture=capture)
|
|
|
|
if code == 400:
|
|
return BadRequestError(message)
|
|
|
|
if code == 401:
|
|
return UnauthorizedError(message)
|
|
|
|
if code == 403:
|
|
return ForbiddenError(message)
|
|
|
|
if code == 404:
|
|
return NotFoundError(message)
|
|
|
|
if code == 406:
|
|
return AuthKeyError(message)
|
|
|
|
if code == 500:
|
|
return ServerError(message)
|
|
|
|
return RPCError('{} (code {})'.format(message, code))
|