rewritten crap

sender/ozon.py

@@ -18,6 +18,7 @@ class OzonStocksSender(BaseStocksSender):
     async def _process_chunk(self, chunk: list[dict]) -> SendStockStatus:
         session, response = await self.api.update_stocks(data=chunk)
         try:
+            async with response:
                 status_code = response.status
                 if status_code == 200:
                     return SendStockStatus.SUCCESS
@@ -37,8 +38,8 @@ class OzonStocksSender(BaseStocksSender):
                     return SendStockStatus.SHOULD_RETRY
                 return SendStockStatus.ERROR
         finally:
-            await response.close()
             await session.close()
+
     async def after_chunk_processed(self):
         return await asyncio.sleep(80 / 100)
 

sender/wildberries.py

@@ -21,6 +21,7 @@ class WildberriesStocksSender(BaseStocksSender):
     async def _process_chunk(self, chunk: list[dict]) -> SendStockStatus:
         session, response = await self.api.update_stocks(chunk)
         try:
+            async with response:
                 headers = response.headers
                 status_code = response.status
 
@@ -41,8 +42,8 @@ class WildberriesStocksSender(BaseStocksSender):
                 self.remaining = int(headers.get('X-Ratelimit-Remaining', 0))
                 return SendStockStatus.SUCCESS
         finally:
-            await response.close()
             await session.close()
+
     async def after_chunk_processed(self):
         if self.remaining <= 0:
             await asyncio.sleep(self.sleep_time)
@@ -52,6 +53,7 @@ class WildberriesStocksSender(BaseStocksSender):
         if self.remaining <= 0:
             await asyncio.sleep(self.sleep_time)
             self.remaining = 1
+
     @property
     def chunk_size(self) -> int:
         return 1000

sender/yandexmarket.py

@@ -28,6 +28,7 @@ class YandexmarketStocksSender(BaseStocksSender):
     async def _process_chunk(self, chunk: list[dict]) -> SendStockStatus:
         session, response = await self.api.update_stocks(chunk)
         try:
+            async with response:
                 status_code = response.status
                 if status_code == 200:
                     self.total_stocks_sent += len(chunk)
@@ -40,7 +41,6 @@ class YandexmarketStocksSender(BaseStocksSender):
                 logging.error(f'[{self.updater.marketplace.id}]: {response_text}')
                 return SendStockStatus.ERROR
         finally:
-            await response.close()
             await session.close()
 
     async def after_chunk_processed(self):

updaters/base.py

@@ -73,7 +73,6 @@ class BaseMarketplaceUpdater(ABC):
             return {}
 
     async def filter_stocks_data(self, stock_data_list: list[StockData]) -> list[StockData]:
-        return stock_data_list
         cached_stocks: dict = await self.redis_client.hgetall(self.get_cache_key())
         cached_stocks = {int(k): int(v) for k, v in cached_stocks.items()}
         result = []