rewritten crap

sender/wildberries.py

@@ -0,0 +1,54 @@
+import asyncio
+from typing import TYPE_CHECKING
+
+from marketplaces import WildberriesMarketplaceApi
+from sender.base import SendStockStatus, BaseStocksSender
+
+if TYPE_CHECKING:
+    from updaters.base import BaseMarketplaceUpdater
+
+
+class WildberriesStocksSender(BaseStocksSender):
+    def __init__(self, updater: "BaseMarketplaceUpdater"):
+        super().__init__(updater)
+        self.remaining = 1
+        self.sleep_time = 60 / 300
+
+    @property
+    def max_retries(self) -> int:
+        return 5
+
+    async def _process_chunk(self, chunk: list[dict]) -> SendStockStatus:
+        response = await self.api.update_stocks(chunk)
+        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 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
+
+    async def after_chunk_processed(self):
+        if self.remaining <= 0:
+            await asyncio.sleep(self.sleep_time)
+            self.remaining = 1
+
+    async def before_chunk_processed(self):
+        if self.remaining <= 0:
+            await asyncio.sleep(self.sleep_time)
+            self.remaining = 1
+    @property
+    def chunk_size(self) -> int:
+        return 1000