Upload seed_products.py

seed_products.py

@@ -0,0 +1,142 @@
+import json
+import random
+from sqlalchemy.orm import Session
+from app.db.base import SessionLocal
+from app.models.product import Product, Category, ProductImage
+
+# Mapping distinct categories from Excel to a hierarchical structure
+CATEGORY_MAP = {
+    # Parents (Keys) : Subcategories (Values)
+    "الأزياء": ["الملابس الرجالية", "الملابس النسائية"],
+    "الأحذية": ["الأحذية"],
+    "الأطفال والألعاب": ["الأطفال"],
+    "المنزل والمطبخ": ["المنزل والمطبخ"],
+    "العطور والتجميل": ["العطور", "التجميل والعناية"],
+    "الأجهزة الإلكترونية": ["الهواتف والإلكترونيات"],
+}
+
+# Image placeholders by Parent Category for realistic-ish dummy images
+IMAGE_KEYWORDS = {
+    "الأزياء": "clothing,fashion,apparel",
+    "الأحذية": "shoes,sneakers",
+    "الأطفال والألعاب": "kids,toys,baby",
+    "المنزل والمطبخ": "kitchen,home,appliance",
+    "العطور والتجميل": "perfume,beauty,skincare",
+    "الأجهزة الإلكترونية": "electronics,smartphone,gadget",
+}
+
+def get_image_url(parent_category: str, image_index: int):
+    # Use Unsplash Source or Picsum
+    # To ensure varied images, we use random seeds
+    seed = random.randint(1000, 9999) + image_index
+    return f"https://picsum.photos/seed/{seed}/800/800"
+
+def seed_database():
+    db: Session = SessionLocal()
+    try:
+        print("Loading JSON data...")
+        with open("products_dump.json", "r", encoding="utf-8") as f:
+            products_data = json.load(f)
+            
+        print(f"Loaded {len(products_data)} products from JSON.")
+
+        # Optional: Clean existing data (or we can just append, but appending might cause duplicates)
+        # Uncomment if starting fresh is preferred:
+        # print("Clearing existing products and categories...")
+        # db.query(ProductImage).delete()
+        # db.query(Product).delete()
+        # db.query(Category).delete()
+        # db.commit()
+
+        # 1. Create Parent Categories
+        parent_cats = {}
+        for parent_name, sub_names in CATEGORY_MAP.items():
+            parent_cat = db.query(Category).filter(Category.name_ar == parent_name).first()
+            if not parent_cat:
+                parent_cat = Category(
+                    name_ar=parent_name,
+                    name_en=parent_name, # Fallback to Arabic if English is unavailable
+                    icon="grid", # generic icon
+                )
+                db.add(parent_cat)
+                db.commit()
+                db.refresh(parent_cat)
+            parent_cats[parent_name] = parent_cat
+
+        # 2. Create Subcategories
+        sub_cats = {}
+        for parent_name, sub_names in CATEGORY_MAP.items():
+            for sub_name in sub_names:
+                sub_cat = db.query(Category).filter(Category.name_ar == sub_name).first()
+                if not sub_cat:
+                    sub_cat = Category(
+                        name_ar=sub_name,
+                        name_en=sub_name,
+                        icon="list",
+                        parent_id=parent_cats[parent_name].id
+                    )
+                    db.add(sub_cat)
+                    db.commit()
+                    db.refresh(sub_cat)
+                sub_cats[sub_name] = {"model": sub_cat, "parent_name": parent_name}
+
+        # 3. Create Products
+        print("Inserting products...")
+        inserted_count = 0
+        for item in products_data:
+            cat_name = item.get("الفئة")
+            if not cat_name or cat_name not in sub_cats:
+                print(f"Skipping row with missing/unknown category: {cat_name}")
+                continue
+                
+            sub_cat_info = sub_cats[cat_name]
+            
+            # Create product
+            price = float(item.get("سعر البيع (ريال)", 0))
+            compare_price = price * 1.2 # dummy compare price 20% higher
+            
+            prod = Product(
+                name_ar=item.get("اسم المنتج", ""),
+                name_en=item.get("اسم المنتج", ""), # Fallback
+                description_ar=item.get("الوصف التسويقي", ""),
+                description_en=item.get("الوصف التسويقي", ""),
+                price=price,
+                compare_price=round(compare_price, 2),
+                stock=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
+            )
+            db.add(prod)
+            db.flush() # get prod.id
+            
+            # 4. Create multiple images for the product
+            # Request was "توفير صور للمنتجات مناسبة متعددة" -> multiple images
+            num_images = random.randint(2, 4)
+            for i in range(num_images):
+                img_url = get_image_url(sub_cat_info["parent_name"], i)
+                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 % 50 == 0:
+                print(f"Inserted {inserted_count} / {len(products_data)} products...")
+
+        db.commit()
+        print(f"Successfully seeded {inserted_count} products with multiple images and subcategories!")
+
+    except Exception as e:
+        db.rollback()
+        print(f"Error seeding database: {e}")
+    finally:
+        db.close()
+
+if __name__ == "__main__":
+    seed_database()