feat: a lot of a lot

marketplaces/wildberries/core.py

@@ -0,0 +1,76 @@
+import time
+
+from sqlalchemy import select
+
+from external.marketplace.wildberries.core import WildberriesMarketplaceApi
+from marketplaces.base.core import BaseMarketplaceController
+from models import Product, ProductBarcode, ProductImage, WildberriesProduct
+
+
+class WildberriesController(BaseMarketplaceController):
+    api: WildberriesMarketplaceApi
+
+    async def synchronize_products(self):
+        products = []
+        barcodes = []
+        images = []
+        wildberries_products = []
+
+        marketplace_id: int = self.marketplace.id
+        synchronized_nm_uuids = set(
+            (
+                await self.session.scalars(
+                    select(
+                        WildberriesProduct.nm_uuid
+                    )
+                    .where(
+                        WildberriesProduct.marketplace_id == marketplace_id
+                    )
+                )
+            ).all()
+        )
+
+        async for card in self.api.get_all_products():
+            nm_uuid = card['nmUUID']
+            if nm_uuid in synchronized_nm_uuids:
+                continue
+            sizes: list[dict] = card.get('sizes') or []
+            for size in sizes:
+                tech_size = size.get('techSize')
+                wb_size = size.get('wbSize')
+                size_value = tech_size or wb_size
+                product = Product(
+                    client_id=self.marketplace.client_id,
+                    name=card['title'],
+                    article=card['vendorCode'],
+                    size=size_value
+                )
+                skus = size.get('skus') or []
+                for sku in skus:
+                    barcode = ProductBarcode(
+                        product=product,
+                        barcode=sku
+                    )
+                    barcodes.append(barcode)
+                photos = card.get('photos') or []
+                for photo in photos:
+                    image = ProductImage(
+                        product=product,
+                        image_url=photo['big']
+                    )
+                    images.append(image)
+                    break
+                wildberries_product = WildberriesProduct(
+                    marketplace_id=self.marketplace.id,
+                    product=product,
+                    nm_uuid=nm_uuid
+                )
+                wildberries_products.append(
+                    wildberries_product
+                )
+                products.append(product)
+        instances = products + wildberries_products + barcodes + images
+        start = time.time()
+        self.session.add_all(instances)
+        await self.session.commit()
+        print(f'Add and commit elapsed: {time.time() - start}')