feat: ability to reset marketplace

2024-11-11 10:41:51 +03:00
parent 8cb94d06bf
commit 4841578dee
7 changed files with 47 additions and 2 deletions
--- a/updaters/base.py
+++ b/updaters/base.py
@@ -57,3 +57,16 @@ class BaseMarketplaceUpdater(ABC):
            marketplace_update = self.get_update_for_marketplace(stock_data)
            marketplace_updates.append(marketplace_update)
        await self.marketplace_api.update_stocks(marketplace_updates)
+
+    async def reset(self):
+        if not self.marketplace_api:
+            return
+        stock_data_list = await queries.general.get_stocks_data(
+            session=self.session,
+            marketplace=self.marketplace
+        )
+        marketplace_updates = []
+        for stock_data in stock_data_list:
+            marketplace_update = self.get_update_for_marketplace(stock_data)
+            marketplace_updates.append(marketplace_update)
+        await self.marketplace_api.update_stocks(marketplace_updates)