From b597f1e5655b543c2f8c41604283f30ef61527d8 Mon Sep 17 00:00:00 2001 From: Alex Root Junior Date: Wed, 2 May 2018 18:09:51 +0300 Subject: [PATCH] Fixed the RetryAfter and MigrateToChat errors --- aiogram/bot/api.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/aiogram/bot/api.py b/aiogram/bot/api.py index f5abfc23..342ec62c 100644 --- a/aiogram/bot/api.py +++ b/aiogram/bot/api.py @@ -59,13 +59,14 @@ async def _check_result(method_name, response): result_json = {} description = result_json.get('description') or body + parameters = types.ResponseParameters(**result_json.get('parameters', {}) or {}) if HTTPStatus.OK <= response.status <= HTTPStatus.IM_USED: return result_json.get('result') - elif 'retry_after' in result_json: - raise exceptions.RetryAfter(result_json['retry_after']) - elif 'migrate_to_chat_id' in result_json: - raise exceptions.MigrateToChat(result_json['migrate_to_chat_id']) + elif parameters.retry_after: + raise exceptions.RetryAfter(parameters.retry_after) + elif parameters.migrate_to_chat_id: + raise exceptions.MigrateToChat(parameters.migrate_to_chat_id) elif response.status == HTTPStatus.BAD_REQUEST: exceptions.BadRequest.detect(description) elif response.status == HTTPStatus.NOT_FOUND: