feat: get stocks
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user