Upload app/api/seed.py

app/api/seed.py

@@ -1,5 +1,8 @@
 from fastapi import APIRouter, Depends
 from sqlalchemy.orm import Session
+import json
+import random
+import os
 
 from app.db.base import get_db
 from app.models.product import Product, ProductImage, Category
@@ -10,222 +13,36 @@ from app.schemas.auth import AuthResponse
 
 router = APIRouter(prefix="/seed", tags=["Seed Data"])
 
-
-DEMO_CATEGORIES = [
-    {"name_ar": "إلكترونيات", "name_en": "Electronics", "icon": "💻", "sort_order": 1},
-    {"name_ar": "ملابس", "name_en": "Clothing", "icon": "👕", "sort_order": 2},
-    {"name_ar": "أحذية", "name_en": "Shoes", "icon": "👟", "sort_order": 3},
-    {"name_ar": "ساعات", "name_en": "Watches", "icon": "⌚", "sort_order": 4},
-    {"name_ar": "حقائب", "name_en": "Bags", "icon": "👜", "sort_order": 5},
-    {"name_ar": "إكسسوارات", "name_en": "Accessories", "icon": "💎", "sort_order": 6},
-]
-
-DEMO_PRODUCTS = [
-    {
-        "name_ar": "سماعات لاسلكية برو",
-        "name_en": "Wireless Pro Headphones",
-        "description_ar": "سماعات رأس لاسلكية عالية الجودة مع إلغاء الضوضاء النشط وعمر بطارية يصل إلى 30 ساعة. صوت محيطي 360 درجة مع ميكروفون مدمج للمكالمات.",
-        "description_en": "Premium wireless over-ear headphones with active noise cancellation and up to 30 hours battery life. Features 360° spatial audio and built-in microphone for crystal-clear calls.",
-        "price": 299.99,
-        "compare_price": 399.99,
-        "stock": 50,
-        "category_idx": 0,
-        "rating": 4.8,
-        "rating_count": 234,
-        "is_featured": True,
-        "images": [
-            "https://images.unsplash.com/photo-1505740420928-5e560c06d30e?w=600",
-            "https://images.unsplash.com/photo-1484704849700-f032a568e944?w=600",
-        ],
-    },
-    {
-        "name_ar": "ساعة ذكية ألترا",
-        "name_en": "Ultra Smart Watch",
-        "description_ar": "ساعة ذكية متقدمة مع شاشة AMOLED ومقاومة للماء حتى 50 متر. تتبع اللياقة البدنية ومراقبة معدل ضربات القلب والنوم.",
-        "description_en": "Advanced smartwatch with AMOLED display and 50m water resistance. Comprehensive fitness tracking, heart rate monitoring, sleep analysis, and GPS navigation.",
-        "price": 449.99,
-        "compare_price": 599.99,
-        "stock": 30,
-        "category_idx": 3,
-        "rating": 4.6,
-        "rating_count": 156,
-        "is_featured": True,
-        "images": [
-            "https://images.unsplash.com/photo-1523275335684-37898b6baf30?w=600",
-            "https://images.unsplash.com/photo-1546868871-af0de0ae72be?w=600",
-        ],
-    },
-    {
-        "name_ar": "حقيبة ظهر جلدية",
-        "name_en": "Leather Backpack",
-        "description_ar": "حقيبة ظهر من الجلد الطبيعي الفاخر مع جيوب متعددة وبطانة مقاومة للماء. مثالية للعمل والسفر.",
-        "description_en": "Premium genuine leather backpack with multiple compartments and water-resistant lining. Perfect for work and travel with laptop sleeve up to 15 inches.",
-        "price": 189.99,
-        "compare_price": 249.99,
-        "stock": 45,
-        "category_idx": 4,
-        "rating": 4.5,
-        "rating_count": 89,
-        "is_featured": True,
-        "images": [
-            "https://images.unsplash.com/photo-1553062407-98eeb64c6a62?w=600",
-            "https://images.unsplash.com/photo-1622560480605-d83c853bc5c3?w=600",
-        ],
-    },
-    {
-        "name_ar": "حذاء رياضي إير ماكس",
-        "name_en": "Air Max Running Shoes",
-        "description_ar": "حذاء رياضي خفيف الوزن مع تقنية التهوية المتقدمة ونعل مرن يوفر أقصى درجات الراحة أثناء الجري.",
-        "description_en": "Lightweight running shoes with advanced ventilation technology and flexible sole for maximum comfort during runs. Responsive cushioning and premium mesh upper.",
-        "price": 159.99,
-        "compare_price": 199.99,
-        "stock": 80,
-        "category_idx": 2,
-        "rating": 4.7,
-        "rating_count": 312,
-        "is_featured": True,
-        "images": [
-            "https://images.unsplash.com/photo-1542291026-7eec264c27ff?w=600",
-            "https://images.unsplash.com/photo-1608231387042-66d1773070a5?w=600",
-        ],
-    },
-    {
-        "name_ar": "لابتوب ألتراسليم برو",
-        "name_en": "UltraSlim Pro Laptop",
-        "description_ar": "لابتوب احترافي بشاشة 14 بوصة 4K ومعالج من الجيل الأخير. خفيف الوزن 1.2 كغ مع بطارية تدوم 18 ساعة.",
-        "description_en": "Professional laptop with 14-inch 4K display and latest-gen processor. Ultra-lightweight at 1.2kg with 18-hour battery life. 16GB RAM, 512GB SSD.",
-        "price": 1299.99,
-        "compare_price": 1599.99,
-        "stock": 20,
-        "category_idx": 0,
-        "rating": 4.9,
-        "rating_count": 178,
-        "is_featured": True,
-        "images": [
-            "https://images.unsplash.com/photo-1496181133206-80ce9b88a853?w=600",
-            "https://images.unsplash.com/photo-1525547719571-a2d4ac8945e2?w=600",
-        ],
-    },
-    {
-        "name_ar": "نظارات شمسية كلاسيك",
-        "name_en": "Classic Sunglasses",
-        "description_ar": "نظارات شمسية بتصميم كلاسيكي أنيق مع عدسات بولارايزد توفر حماية كاملة من أشعة UV. إطار معدني خفيف.",
-        "description_en": "Classic design sunglasses with polarized lenses providing full UV protection. Lightweight metal frame with adjustable nose pads for perfect fit.",
-        "price": 79.99,
-        "compare_price": 129.99,
-        "stock": 100,
-        "category_idx": 5,
-        "rating": 4.4,
-        "rating_count": 67,
-        "is_featured": False,
-        "images": [
-            "https://images.unsplash.com/photo-1572635196237-14b3f281503f?w=600",
-        ],
-    },
-    {
-        "name_ar": "قميص قطن بريميوم",
-        "name_en": "Premium Cotton Shirt",
-        "description_ar": "قميص من القطن المصري الفاخر بتصميم عصري مريح. مناسب للمناسبات الرسمية وغير الرسمية.",
-        "description_en": "Premium Egyptian cotton shirt with modern comfortable design. Suitable for both formal and casual occasions. Breathable fabric with wrinkle-resistant finish.",
-        "price": 89.99,
-        "compare_price": None,
-        "stock": 120,
-        "category_idx": 1,
-        "rating": 4.3,
-        "rating_count": 45,
-        "is_featured": False,
-        "images": [
-            "https://images.unsplash.com/photo-1602810318383-e386cc2a3ccf?w=600",
-        ],
-    },
-    {
-        "name_ar": "سماعات أذن لاسلكية",
-        "name_en": "Wireless Earbuds",
-        "description_ar": "سماعات أذن لاسلكية مع علبة شحن توفر 24 ساعة من الاستخدام. مقاومة للماء والعرق مع صوت عالي الدقة.",
-        "description_en": "True wireless earbuds with charging case providing 24 hours of use. Water and sweat resistant (IPX5) with high-definition sound and active noise cancellation.",
-        "price": 129.99,
-        "compare_price": 179.99,
-        "stock": 65,
-        "category_idx": 0,
-        "rating": 4.5,
-        "rating_count": 198,
-        "is_featured": False,
-        "images": [
-            "https://images.unsplash.com/photo-1590658268037-6bf12f032f55?w=600",
-        ],
-    },
-    {
-        "name_ar": "محفظة جلدية فاخرة",
-        "name_en": "Luxury Leather Wallet",
-        "description_ar": "محفظة من الجلد الإيطالي الفاخر مع حماية RFID. تصميم أنيق مع فتحات متعددة للبطاقات وجيب للعملات.",
-        "description_en": "Premium Italian leather wallet with RFID protection. Elegant design with multiple card slots, bill compartment, and coin pocket. Gift box included.",
-        "price": 69.99,
-        "compare_price": 99.99,
-        "stock": 90,
-        "category_idx": 5,
-        "rating": 4.6,
-        "rating_count": 112,
-        "is_featured": False,
-        "images": [
-            "https://images.unsplash.com/photo-1627123424574-724758594e93?w=600",
-        ],
-    },
-    {
-        "name_ar": "جاكيت شتوي مقاوم للماء",
-        "name_en": "Waterproof Winter Jacket",
-        "description_ar": "جاكيت شتوي دافئ ومقاوم للماء والرياح. حشوة داون طبيعية مع قلنسوة قابلة للإزالة وجيوب متعددة.",
-        "description_en": "Warm waterproof and windproof winter jacket. Natural down insulation with removable hood and multiple pockets. Breathable membrane technology.",
-        "price": 249.99,
-        "compare_price": 349.99,
-        "stock": 35,
-        "category_idx": 1,
-        "rating": 4.7,
-        "rating_count": 76,
-        "is_featured": False,
-        "images": [
-            "https://images.unsplash.com/photo-1544923246-77307dd270b1?w=600",
-        ],
-    },
-    {
-        "name_ar": "كاميرا ميرورليس احترافية",
-        "name_en": "Professional Mirrorless Camera",
-        "description_ar": "كاميرا ميرورليس احترافية بدقة 45 ميجابكسل وتصوير فيديو 8K. نظام تركيز تلقائي متقدم مع تثبيت صورة 5 محاور.",
-        "description_en": "Professional mirrorless camera with 45MP sensor and 8K video recording. Advanced autofocus system with 5-axis image stabilization. Weather-sealed body.",
-        "price": 2499.99,
-        "compare_price": 2999.99,
-        "stock": 10,
-        "category_idx": 0,
-        "rating": 4.9,
-        "rating_count": 54,
-        "is_featured": True,
-        "images": [
-            "https://images.unsplash.com/photo-1516035069371-29a1b244cc32?w=600",
-            "https://images.unsplash.com/photo-1502920917128-1aa500764cbd?w=600",
-        ],
-    },
-    {
-        "name_ar": "حذاء رسمي جلدي",
-        "name_en": "Formal Leather Shoes",
-        "description_ar": "حذاء رسمي من الجلد الطبيعي المصقول يدوياً. نعل مطاطي مقاوم للانزلاق مع بطانة جلدية مريحة.",
-        "description_en": "Hand-polished genuine leather formal shoes. Non-slip rubber sole with comfortable leather lining. Classic Oxford design with Goodyear welt construction.",
-        "price": 199.99,
-        "compare_price": None,
-        "stock": 40,
-        "category_idx": 2,
-        "rating": 4.4,
-        "rating_count": 38,
-        "is_featured": False,
-        "images": [
-            "https://images.unsplash.com/photo-1614252369475-531eba835eb1?w=600",
-        ],
-    },
-]
-
+# Mapping distinct categories from Excel to a hierarchical structure
+CATEGORY_MAP = {
+    # Parents (Keys) : Subcategories (Values)
+    "الأزياء": ["الملابس الرجالية", "الملابس النسائية"],
+    "الأحذية": ["الأحذية"],
+    "الأطفال والألعاب": ["الأطفال"],
+    "المنزل والمطبخ": ["المنزل والمطبخ"],
+    "العطور والتجميل": ["العطور", "التجميل والعناية"],
+    "الأجهزة الإلكترونية": ["الهواتف والإلكترونيات"],
+}
+
+# Image placeholders by Parent Category
+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):
+    # Generating consistent but different images based on a seed
+    seed = random.randint(1000, 9999) + image_index
+    return f"https://picsum.photos/seed/{seed}/800/800"
 
 @router.post("", response_model=AuthResponse)
 async def seed_database(db: Session = Depends(get_db)):
-    """Populate database with demo categories, products, and an admin user."""
-    # Always ensure admin user exists (even on re-seed)
+    """Populate database with proper product data from json dump and an admin user."""
+    # Always ensure admin user exists
     existing_admin = db.query(User).filter(User.email == "admin@vortex.com").first()
     if not existing_admin:
         admin_user = User(
@@ -247,54 +64,123 @@ async def seed_database(db: Session = Depends(get_db)):
             statusCode=200,
         )
 
-    # Create categories
-    categories = []
-    for cat_data in DEMO_CATEGORIES:
-        cat = Category(**cat_data)
-        db.add(cat)
-        categories.append(cat)
-    db.flush()
-
-    # Create products
-    for prod_data in DEMO_PRODUCTS:
-        images_data = prod_data.pop("images")
-        category_idx = prod_data.pop("category_idx")
-
-        product = Product(
-            **prod_data,
-            category_id=categories[category_idx].id,
+    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,
+            value={"message": f"Source file {json_path} not found."},
+            statusCode=404,
+            error="File not found"
         )
-        db.add(product)
-        db.flush()
-
-        for i, img_url in enumerate(images_data):
-            img = ProductImage(
-                product_id=product.id,
-                image_url=img_url,
-                sort_order=i,
+        
+    try:
+        with open(json_path, "r", encoding="utf-8") as f:
+            products_data = json.load(f)
+            
+        # 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,
+                    icon="grid",
+                )
+                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
+        inserted_count = 0
+        for item in products_data:
+            cat_name = item.get("الفئة")
+            if not cat_name or cat_name not in sub_cats:
+                continue
+                
+            sub_cat_info = sub_cats[cat_name]
+            
+            price = float(item.get("سعر البيع (ريال)", 0))
+            compare_price = price * 1.2
+            
+            prod = Product(
+                name_ar=item.get("اسم المنتج", ""),
+                name_en=item.get("اسم المنتج", ""),
+                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(img)
+            db.add(prod)
+            db.flush()
+            
+            # 4. Create 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
+
+        # Seed Coupons
+        DEMO_COUPONS = [
+            {"code": "WELCOME10", "discount_percent": 10.0, "is_active": True},
+            {"code": "VORTEX20", "discount_percent": 20.0, "is_active": True},
+            {"code": "SUMMER50", "discount_percent": 50.0, "is_active": False},
+        ]
+
+        for coupon_data in DEMO_COUPONS:
+            coupon = Coupon(**coupon_data)
+            db.add(coupon)
 
-    # Seed Coupons
-    DEMO_COUPONS = [
-        {"code": "WELCOME10", "discount_percent": 10.0, "is_active": True},
-        {"code": "VORTEX20", "discount_percent": 20.0, "is_active": True},
-        {"code": "SUMMER50", "discount_percent": 50.0, "is_active": False},
-    ]
-
-    for coupon_data in DEMO_COUPONS:
-        coupon = Coupon(**coupon_data)
-        db.add(coupon)
+        db.commit()
 
-    db.commit()
+        return AuthResponse(
+            isSuccess=True,
+            value={
+                "message": "Database seeded successfully",
+                "categories_count": db.query(Category).count(),
+                "products_count": inserted_count,
+                "coupons_count": len(DEMO_COUPONS),
+            },
+            statusCode=201,
+        )
 
-    return AuthResponse(
-        isSuccess=True,
-        value={
-            "message": "Database seeded successfully",
-            "categories_count": len(DEMO_CATEGORIES),
-            "products_count": len(DEMO_PRODUCTS),
-            "coupons_count": len(DEMO_COUPONS),
-        },
-        statusCode=201,
-    )
+    except Exception as e:
+        db.rollback()
+        return AuthResponse(
+            isSuccess=False,
+            value={"message": str(e)},
+            statusCode=500,
+            error=str(e)
+        )