diff --git a/src/loader.py b/src/loader.py index bbced86..1e9f5c0 100644 --- a/src/loader.py +++ b/src/loader.py @@ -21,8 +21,8 @@ class EtlLoader: self.etl = EtlConfig() self.api_config = ApiConfig() - @backoff.on_exception(backoff.expo, (asyncio.TimeoutError, ), base=2, factor=1, - max_value=etl_config.max_wait_size, max_tries=None) + @backoff.on_exception(backoff.expo, (asyncio.TimeoutError, aiohttp.ClientConnectorError), base=2, factor=1, + max_value=int(etl_config.max_wait_size), max_tries=None) async def load_data(self, state: State, abitr: Abitr, iblock_id: int, storage: BaseStorage): async with aiohttp.ClientSession() as session: logging.info(f"Информация об абитуриенте: {abitr.FIO}") @@ -33,7 +33,7 @@ class EtlLoader: async with session.post( url=f"{self.etl.protocol}://{self.etl.host}:{self.etl.port}", data=data_dict, - timeout=aiohttp.ClientTimeout(total=30) + timeout=aiohttp.ClientTimeout(total=int(etl_config.max_wait_size)) ) as response: pass state.set_state(f'iblock_{iblock_id}', abitr.ID)