diff --git a/src/extractor.py b/src/extractor.py index bbbbab9..460394b 100644 --- a/src/extractor.py +++ b/src/extractor.py @@ -19,7 +19,8 @@ class ApiExtractor: self.headers = {"Authorization-Token": self.api_config.token} self.fields = ["ID", "IBLOCK_ID", "NAME", "CODE", "SECTION_ID"] - @backoff.on_exception(backoff.expo, (aiohttp.ClientResponseError, aiohttp.ClientConnectorError), base=2, factor=1, + @backoff.on_exception(backoff.expo, (aiohttp.ClientResponseError, aiohttp.ClientConnectorError, + aiohttp.ServerDisconnectedError), base=2, factor=1, max_value=5, max_tries=None) async def get_extract_data(self, state: State, iblock_id: int, fields: list = None, **kwargs) -> list[Abitr]: