feat: do not send stocks where disabled

database/sipro/models/general.py

@@ -54,6 +54,7 @@ class Marketplace(BaseSiproModel):
     sell_from_price: Mapped[bool] = mapped_column()
     is_deleted: Mapped[bool] = mapped_column()
     is_paused: Mapped[bool] = mapped_column()
+    send_stocks: Mapped[bool] = mapped_column()
 
     warehouses: Mapped[List["Warehouse"]] = relationship(secondary=marketplace_warehouses)
     warehouse_id: Mapped[str] = mapped_column()

updaters/stocks_updater.py

@@ -47,6 +47,7 @@ class StocksUpdater:
                 Company.is_archived == False,
                 Marketplace.is_deleted == False,
                 Marketplace.is_paused == False,
+                Marketplace.send_stocks==True,
                 Marketplace.base_marketplace.in_([
                     BaseMarketplace.OZON,
                     BaseMarketplace.WILDBERRIES,
@@ -63,6 +64,8 @@ class StocksUpdater:
 
     async def full_update_marketplace(self, marketplace_id: int):
         marketplace = await self.get_marketplace(marketplace_id)
+        if not marketplace.send_stocks:
+            return
         start = time.time()
         updater = UpdaterFactory.get_updater(self.session, marketplace)
         await updater.update_all()
@@ -97,6 +100,8 @@ class StocksUpdater:
 
     async def update_marketplace_products(self, marketplace_id: int, product_ids: list[int]):
         marketplace = await self.get_marketplace(marketplace_id)
+        if not marketplace.send_stocks:
+            return
         start = time.time()
         updater = UpdaterFactory.get_updater(self.session, marketplace)
         await updater.update_products(product_ids)
@@ -118,6 +123,7 @@ class StocksUpdater:
                     Company.is_deleted == False,
                     Company.is_archived == False,
                     Marketplace.is_paused == False,
+                    Marketplace.send_stocks == True
                 )
             )
             stmt_result = await self.session.execute(stmt)