57 lines
1.8 KiB
Python
57 lines
1.8 KiB
Python
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:
|
|
session,response = await self.api.update_stocks(chunk)
|
|
try:
|
|
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
|
|
finally:
|
|
await session.close()
|
|
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
|