Upload app/api/seed.py

app/api/seed.py

@@ -98,7 +98,11 @@ async def seed_database(clear: bool = False, db: Session = Depends(get_db)):
                 statusCode=200,
             )
 
-    json_path = os.path.join(os.path.dirname(os.path.dirname(os.path.dirname(__file__))), "products_dump.json")
+    json_path = os.path.join(os.path.dirname(os.path.dirname(os.path.dirname(__file__))), "real_products_500.json")
+    # Fallback to products_dump.json if 500 file not found
+    if not os.path.exists(json_path):
+        json_path = os.path.join(os.path.dirname(os.path.dirname(os.path.dirname(__file__))), "products_dump.json")
+        
     if not os.path.exists(json_path):
         return AuthResponse(
             isSuccess=False,
@@ -148,52 +152,64 @@ async def seed_database(clear: bool = False, db: Session = Depends(get_db)):
         used_names = set() # Avoid absolute name duplicates in current seed run
         
         for item in products_data:
-            cat_name = item.get("الفئة")
+            # Handle both formats (real_products_500.json vs products_dump.json)
+            cat_name = item.get("category_name") or item.get("الفئة")
             if not cat_name or cat_name not in sub_cats:
                 continue
             
-            product_name_ar = item.get("اسم المنتج", "").strip()
+            product_name_ar = item.get("name_ar") or item.get("اسم المنتج", "").strip()
             if not product_name_ar or product_name_ar in used_names:
                 continue
             
             sub_cat_info = sub_cats[cat_name]
             used_names.add(product_name_ar)
             
-            price = float(item.get("سعر البيع (ريال)", 0))
-            compare_price = price * 1.2
+            price = float(item.get("price") or item.get("سعر البيع (ريال)", 0))
+            compare_price = float(item.get("compare_price") or (price * 1.2))
             
             prod = Product(
                 name_ar=product_name_ar,
-                name_en=product_name_ar,
-                description_ar=item.get("الوصف التسويقي", ""),
-                description_en=item.get("الوصف التسويقي", ""),
+                name_en=item.get("name_en") or product_name_ar,
+                description_ar=item.get("description_ar") or item.get("الوصف التسويقي", ""),
+                description_en=item.get("description_en") or item.get("الوصف التسويقي", ""),
                 price=price,
                 compare_price=round(compare_price, 2),
-                stock=random.randint(10, 100),
+                stock=item.get("stock") or random.randint(10, 100),
                 category_id=sub_cat_info["model"].id,
                 rating=round(random.uniform(3.5, 5.0), 1),
                 rating_count=random.randint(0, 50),
-                is_featured=random.choice([True, False]),
-                is_active=True
+                is_featured=item.get("is_featured", random.choice([True, False])),
+                is_active=item.get("is_active", True),
+                specs=item.get("specs", {})
             )
             db.add(prod)
             db.flush()
             
-            # 4. Create multiple unique images
-            num_images = random.randint(3, 4) # More images to look professional
-            for i in range(num_images):
-                img_url = get_image_url(sub_cat_info["parent_name"], product_name_ar, i)
-                prod_img = ProductImage(
-                    product_id=prod.id,
-                    image_url=img_url,
-                    alt_text=prod.name_ar,
-                    sort_order=i
-                )
-                db.add(prod_img)
+            # 4. Handle Images
+            raw_images = item.get("images", [])
+            if not raw_images:
+                # Generate fallback images if none provided
+                num_images = random.randint(3, 4)
+                for i in range(num_images):
+                    img_url = get_image_url(sub_cat_info["parent_name"], product_name_ar, i)
+                    prod_img = ProductImage(
+                        product_id=prod.id,
+                        image_url=img_url,
+                        alt_text=prod.name_ar,
+                        sort_order=i
+                    )
+                    db.add(prod_img)
+            else:
+                for i, img_url in enumerate(raw_images):
+                    prod_img = ProductImage(
+                        product_id=prod.id,
+                        image_url=img_url,
+                        alt_text=prod.name_ar,
+                        sort_order=i
+                    )
+                    db.add(prod_img)
             
             inserted_count += 1
-            if inserted_count >= 500: # Maintain a safe limit for demonstration
-                break
 
         # 4. Enrich with Real Discovered Products
         ext_service = ExternalCatalogService(db)