mirror of
https://github.com/LonamiWebs/Telethon.git
synced 2024-11-11 03:56:36 +03:00
67 lines
1.8 KiB
Python
67 lines
1.8 KiB
Python
import urllib.request
|
|
import re
|
|
from threading import Thread
|
|
|
|
from .common import (
|
|
ReadCancelledError, InvalidParameterError, TypeNotFoundError,
|
|
InvalidChecksumError, BrokenAuthKeyError, SecurityError,
|
|
CdnFileTamperedError
|
|
)
|
|
|
|
# This imports the base errors too, as they're imported there
|
|
from .rpc_error_list import *
|
|
|
|
|
|
def report_error(code, message, report_method):
|
|
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):
|
|
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 == 500:
|
|
return ServerError(message)
|
|
|
|
return RPCError('{} (code {})'.format(message, code))
|