Browse Source

Рефакторинг

Убрал ненужные импорты
master
Дмитрий 3 years ago
parent
commit
68d0142bd7
  1. 6
      config_object.py
  2. 7
      request_builder.py

6
config_object.py

@ -1,10 +1,12 @@
"""Модуль с объектами для работы с конфигами nginx""" """Модуль с объектами для работы с конфигами nginx"""
import asyncio
import os import os
import aiofiles import aiofiles
class ConfigObject: class ConfigObject:
"""Объект для хранения настроек для хоста."""
def __init__(self, host: str, conf_body: str, path: str): def __init__(self, host: str, conf_body: str, path: str):
self.host = host self.host = host
self.conf_body = conf_body.replace( self.conf_body = conf_body.replace(
@ -35,6 +37,8 @@ class ConfigObject:
class ConfigFactory: class ConfigFactory:
"""Фабрика, чтобы проще было строить однотипные объекты."""
def __init__(self, path_to_template: str, path_to_configs_dir: str): def __init__(self, path_to_template: str, path_to_configs_dir: str):
self.templ = self.__read_config_template_file(path_to_template) self.templ = self.__read_config_template_file(path_to_template)
self.path = path_to_configs_dir self.path = path_to_configs_dir

7
request_builder.py

@ -1,6 +1,5 @@
"""Модуль для конструирования и обработки запросов.""" """Модуль для конструирования и обработки запросов."""
import asyncio import asyncio
from typing import Dict, List, NewType
import aiohttp import aiohttp
from aiohttp import ClientResponse from aiohttp import ClientResponse
@ -9,6 +8,7 @@ from loguru import logger
from app_config import AppConfig from app_config import AppConfig
class RequestBulder: class RequestBulder:
"""Конструктор запросов. """Конструктор запросов.
@ -22,9 +22,10 @@ class RequestBulder:
send_request(url: str, json_body: dict) send_request(url: str, json_body: dict)
Асинхронный метод для отправки запроса. Может принимать пустой json_body. Асинхронный метод для отправки запроса. Может принимать пустой json_body.
""" """
def __init__(self, cfg: AppConfig) -> None: def __init__(self, cfg: AppConfig) -> None:
"""Инициализация. Объект сесси `aiohttp.ClientSession` """Инициализация. Объект сесси `aiohttp.ClientSession`
создается здесь. """ создается здесь."""
self.cfg = cfg self.cfg = cfg
_conn = aiohttp.TCPConnector(limit=cfg.requests_count) _conn = aiohttp.TCPConnector(limit=cfg.requests_count)
self.session = aiohttp.ClientSession(connector=_conn) self.session = aiohttp.ClientSession(connector=_conn)
@ -75,7 +76,7 @@ class RequestBulder:
await asyncio.sleep(self.cfg.frequency_sec) await asyncio.sleep(self.cfg.frequency_sec)
except ServerDisconnectedError: except ServerDisconnectedError:
logger.error(f"Сервер отклонил подключение {_url}") logger.error(f"Сервер отклонил подключение {_url}")
return dict() return {}
async def wait(self) -> None: async def wait(self) -> None:
"""Ждет frequency_sec время.""" """Ждет frequency_sec время."""

Loading…
Cancel
Save