You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
87 lines
3.1 KiB
87 lines
3.1 KiB
"""Модуль для конструирования и обработки запросов.""" |
|
import asyncio |
|
|
|
import aiohttp |
|
from aiohttp import ClientResponse |
|
from aiohttp.client_exceptions import ClientConnectorError, ServerDisconnectedError |
|
from loguru import logger |
|
|
|
from app_config import AppConfig |
|
|
|
|
|
class RequestBulder: |
|
"""Конструктор запросов. |
|
|
|
Attributes |
|
---------- |
|
cfg : AppConfig |
|
Объект с конфигом для приложения. |
|
|
|
Methods |
|
------- |
|
send_request(url: str, json_body: dict) |
|
Асинхронный метод для отправки запроса. Может принимать пустой json_body. |
|
""" |
|
|
|
def __init__(self, cfg: AppConfig) -> None: |
|
"""Инициализация. Объект сесси `aiohttp.ClientSession` |
|
создается здесь.""" |
|
self.cfg = cfg |
|
_conn = aiohttp.TCPConnector(limit=cfg.requests_count) |
|
self.session = aiohttp.ClientSession(connector=_conn) |
|
|
|
async def __check_resp(self, resp: ClientResponse) -> bool: |
|
if not resp.status == 200: |
|
return False |
|
|
|
response = await resp.json() |
|
|
|
if not response["done"]: |
|
return False |
|
|
|
return True |
|
|
|
async def send_request(self, url: str, json_body: dict) -> dict: |
|
"""Выполняет запрос, при успехе возвращает json с ответом, при |
|
неудаче возвращает пустой dict. |
|
|
|
Parameters |
|
---------- |
|
url : str |
|
Адрес куда слать запрос. Нужна часть не включающая хост, только get/post. |
|
|
|
json_body: Json |
|
Тело хапроса в Json формате. |
|
|
|
Returns |
|
------- |
|
dict |
|
Возвращает либо пустой dict, либо json ответа сервера (заполненный dict). |
|
|
|
Examples |
|
-------- |
|
>>> await send_request('count', json_body={} ) |
|
|
|
""" |
|
_url = f"{self.cfg.central_host_url}/{url}" |
|
try: |
|
async with self.session.get(_url, json=json_body) as raw_resp: |
|
if await self.__check_resp(raw_resp): |
|
json_resp: dict = await raw_resp.json() |
|
json_resp.pop("done") |
|
return json_resp |
|
except ClientConnectorError: |
|
logger.error(f"Ошибка подключения к серверу {_url}") |
|
# не придумал ничего умнее чем подождать frequency_sec из конфига |
|
await asyncio.sleep(self.cfg.frequency_sec) |
|
except ServerDisconnectedError: |
|
logger.error(f"Сервер отклонил подключение {_url}") |
|
return {} |
|
|
|
async def wait(self) -> None: |
|
"""Ждет frequency_sec время.""" |
|
await asyncio.sleep(self.cfg.frequency_sec) |
|
|
|
async def close(self) -> None: |
|
"""Gracefull shutdown connection.""" |
|
await self.session.close()
|
|
|