python/aiogram
1
0
Fork 0
mirror of https://github.com/aiogram/aiogram.git synced 2025-12-11 09:55:21 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Small changes.

Browse source
This commit is contained in:
Alex Root Junior 2017-10-22 14:15:19 +03:00
parent 61e7ee3220
commit 1b87532221
1 changed files with 14 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

29
aiogram/bot/api.py
View file

@ -5,9 +5,8 @@ from http import HTTPStatus
import aiohttp import aiohttp
from ..utils import exceptions
from ..utils import json from ..utils import json
from ..utils.exceptions import BadRequest, ConflictError, MigrateToChat, NetworkError, RetryAfter, TelegramAPIError, \
Unauthorized, ValidationError
from ..utils.helper import Helper, HelperMode, Item from ..utils.helper import Helper, HelperMode, Item
# Main aiogram logger # Main aiogram logger
@ -26,11 +25,11 @@ def check_token(token: str) -> bool:
:return: :return:
""" """
if any(x.isspace() for x in token): if any(x.isspace() for x in token):
raise ValidationError('Token is invalid!') raise exceptions.ValidationError('Token is invalid!')
left, sep, right = token.partition(':') left, sep, right = token.partition(':')
if (not sep) or (not left.isdigit()) or (len(left) < 3): if (not sep) or (not left.isdigit()) or (len(left) < 3):
raise ValidationError('Token is invalid!') raise exceptions.ValidationError('Token is invalid!')
return True return True
@ -61,21 +60,21 @@ async def _check_result(method_name, response):
if HTTPStatus.OK <= response.status <= HTTPStatus.IM_USED: if HTTPStatus.OK <= response.status <= HTTPStatus.IM_USED:
return result_json.get('result') return result_json.get('result')
elif 'retry_after' in result_json: elif 'retry_after' in result_json:
raise RetryAfter(result_json['retry_after']) raise exceptions.RetryAfter(result_json['retry_after'])
elif 'migrate_to_chat_id' in result_json: elif 'migrate_to_chat_id' in result_json:
raise MigrateToChat(result_json['migrate_to_chat_id']) raise exceptions.MigrateToChat(result_json['migrate_to_chat_id'])
elif response.status == HTTPStatus.BAD_REQUEST: elif response.status == HTTPStatus.BAD_REQUEST:
raise BadRequest(description) raise exceptions.BadRequest(description)
elif response.status == HTTPStatus.CONFLICT: elif response.status == HTTPStatus.CONFLICT:
raise ConflictError(description) raise exceptions.ConflictError(description)
elif response.status in [HTTPStatus.UNAUTHORIZED, HTTPStatus.FORBIDDEN]: elif response.status in [HTTPStatus.UNAUTHORIZED, HTTPStatus.FORBIDDEN]:
raise Unauthorized(description) raise exceptions.Unauthorized(description)
elif response.status == HTTPStatus.REQUEST_ENTITY_TOO_LARGE: elif response.status == HTTPStatus.REQUEST_ENTITY_TOO_LARGE:
raise NetworkError('File too large for uploading. ' raise exceptions.NetworkError('File too large for uploading. '
'Check telegram api limits https://core.telegram.org/bots/api#senddocument') 'Check telegram api limits https://core.telegram.org/bots/api#senddocument')
elif response.status >= HTTPStatus.INTERNAL_SERVER_ERROR: elif response.status >= HTTPStatus.INTERNAL_SERVER_ERROR:
raise TelegramAPIError(description) raise exceptions.TelegramAPIError(description)
raise TelegramAPIError(f"{description} [{response.status}]") raise exceptions.TelegramAPIError(f"{description} [{response.status}]")
def _guess_filename(obj): def _guess_filename(obj):
@ -152,8 +151,8 @@ async def request(session, token, method, data=None, files=None, continue_retry=
async with session.post(url, data=data, **kwargs) as response: async with session.post(url, data=data, **kwargs) as response:
return await _check_result(method, response) return await _check_result(method, response)
except aiohttp.ClientError as e: except aiohttp.ClientError as e:
raise NetworkError(f"aiohttp client throws an error: {e.__class__.__name__}: {e}") raise exceptions.NetworkError(f"aiohttp client throws an error: {e.__class__.__name__}: {e}")
except RetryAfter as e: except exceptions.RetryAfter as e:
if continue_retry: if continue_retry:
await asyncio.sleep(e.timeout) await asyncio.sleep(e.timeout)
return await request(session, token, method, data, files, **kwargs) return await request(session, token, method, data, files, **kwargs)