feat: get stocks

This commit is contained in:
2025-03-10 05:17:14 +03:00
parent a91703ecd8
commit 677a1c8b9c
3 changed files with 36 additions and 1 deletions

View File

@@ -58,6 +58,21 @@ class BaseMarketplaceUpdater(ABC):
marketplace_updates.append(marketplace_update)
await self.marketplace_api.update_stocks(marketplace_updates)
async def get_all_stocks(self, only_available: bool) -> List[StockData]:
if not self.marketplace_api:
return []
stock_data_list = await queries.general.get_stocks_data(
session=self.session,
marketplace=self.marketplace
)
if only_available:
stock_data_list = list(filter(lambda x: x["full_stock"] > 0, stock_data_list))
for idx, stock_data in enumerate(stock_data_list):
stock_data['product_id'] = stock_data['marketplace_product'].product_id
del stock_data["marketplace_product"]
stock_data_list[idx] = stock_data
return stock_data_list
async def reset(self):
if not self.marketplace_api:
return