rewritten crap

sender/wildberries.py

@@ -19,30 +19,31 @@ class WildberriesStocksSender(BaseStocksSender):
         return 5
 
     async def _process_chunk(self, chunk: list[dict]) -> SendStockStatus:
-        session,response = await self.api.update_stocks(chunk)
+        session, response = await self.api.update_stocks(chunk)
         try:
-            headers = response.headers
-            status_code = response.status
+            async with response:
+                headers = response.headers
+                status_code = response.status
 
-            if status_code in [
-                401,  # Invalid token
-                403,  # Access denied
-                404,  # Not found
-                400,  # Other
-            ]:
-                return SendStockStatus.ERROR
+                if status_code in [
+                    401,  # Invalid token
+                    403,  # Access denied
+                    404,  # Not found
+                    400,  # Other
+                ]:
+                    return SendStockStatus.ERROR
 
-            # If there is rate limit
-            if status_code == 429:
-                delay_time = float(headers.get('X-Ratelimit-Reset', self.sleep_time))
-                await asyncio.sleep(delay_time)
-                self.remaining = int(headers.get('X-Ratelimit-Limit', 1))
-                return SendStockStatus.SHOULD_RETRY
-            self.remaining = int(headers.get('X-Ratelimit-Remaining', 0))
-            return SendStockStatus.SUCCESS
+                # If there is rate limit
+                if status_code == 429:
+                    delay_time = float(headers.get('X-Ratelimit-Reset', self.sleep_time))
+                    await asyncio.sleep(delay_time)
+                    self.remaining = int(headers.get('X-Ratelimit-Limit', 1))
+                    return SendStockStatus.SHOULD_RETRY
+                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