Upload app/api/seed.py

app/api/seed.py

@@ -1,5 +1,6 @@
 from fastapi import APIRouter, Depends
 from sqlalchemy.orm import Session
+from sqlalchemy import text
 import json
 import random
 import os
@@ -11,6 +12,7 @@ from app.models.cart import Coupon
 from app.models.user import User, UserRole, AuthProvider
 from app.core.security import hash_password
 from app.schemas.auth import AuthResponse
+from app.services.external_catalog import ExternalCatalogService
 
 router = APIRouter(prefix="/seed", tags=["Seed Data"])
 
@@ -53,11 +55,29 @@ def get_image_url(parent_category: str, product_name: str, index: int):
     return f"https://source.unsplash.com/featured/800x1000?{brand_keyword},{parent_category}&sig={unique_sig}"
 
 @router.post("", response_model=AuthResponse)
-async def seed_database(db: Session = Depends(get_db)):
-    """Populate database with proper product data from json dump and an admin user."""
+async def seed_database(clear: bool = False, db: Session = Depends(get_db)):
+    """Populate database with proper product data, optionally clearing existing data first."""
+    if clear:
+        # Clear existing data using raw SQL to avoid materialization errors if enums changed
+        db.execute(text("DELETE FROM product_images"))
+        db.execute(text("DELETE FROM products"))
+        db.execute(text("DELETE FROM categories"))
+        db.execute(text("DELETE FROM coupons"))
+        # Optional: clear users if you want a fresh start, or just fix the admin
+        db.execute(text("UPDATE users SET role = 'ADMIN' WHERE role = 'admin'")) # Fix legacy enums
+        
+        # Patch schema: add specs column if missing (SQLite doesn't support IF NOT EXISTS in ALTER)
+        try:
+            db.execute(text("ALTER TABLE products ADD COLUMN specs JSON"))
+        except Exception:
+            pass # Already exists or other issue
+            
+        db.commit()
+
     # Always ensure admin user exists
-    existing_admin = db.query(User).filter(User.email == "admin@vortex.com").first()
-    if not existing_admin:
+    # Use raw SQL to check for admin to avoid LookupError if data is inconsistent
+    admin_exists = db.execute(text("SELECT id FROM users WHERE email = 'admin@vortex.com'")).fetchone()
+    if not admin_exists:
         admin_user = User(
             name="Admin",
             email="admin@vortex.com",
@@ -68,14 +88,15 @@ async def seed_database(db: Session = Depends(get_db)):
         db.add(admin_user)
         db.commit()
 
-    # Check if products are already seeded
-    existing = db.query(Category).count()
-    if existing > 0:
-        return AuthResponse(
-            isSuccess=True,
-            value={"message": "Database already has data. Admin user ensured."},
-            statusCode=200,
-        )
+    # Check if products are already seeded (Skip if clearing)
+    if not clear:
+        existing = db.query(Category).count()
+        if existing > 0:
+            return AuthResponse(
+                isSuccess=True,
+                value={"message": "Database already has data. Admin user ensured."},
+                statusCode=200,
+            )
 
     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):
@@ -174,6 +195,22 @@ async def seed_database(db: Session = Depends(get_db)):
             if inserted_count >= 500: # Maintain a safe limit for demonstration
                 break
 
+        # 4. Enrich with Real Discovered Products
+        ext_service = ExternalCatalogService(db)
+        discovered_data = []
+        if os.path.exists(ext_service.discovery_file):
+            with open(ext_service.discovery_file, "r", encoding="utf-8") as f:
+                discovered_data = json.load(f)
+        
+        real_count = 0
+        for ext_item in discovered_data:
+            # Avoid duplicates by name
+            if ext_item.get("name_ar") in used_names or ext_item.get("title") in used_names:
+                continue
+            await ext_service.create_product_from_external(ext_item)
+            used_names.add(ext_item.get("name_ar") or ext_item.get("title"))
+            real_count += 1
+
         # Seed Coupons
         DEMO_COUPONS = [
             {"code": "WELCOME10", "discount_percent": 10.0, "is_active": True},
@@ -190,9 +227,10 @@ async def seed_database(db: Session = Depends(get_db)):
         return AuthResponse(
             isSuccess=True,
             value={
-                "message": "Database seeded successfully",
+                "message": "Database seeded with real products successfully",
                 "categories_count": db.query(Category).count(),
-                "products_count": inserted_count,
+                "dummy_products": inserted_count,
+                "real_products": real_count,
                 "coupons_count": len(DEMO_COUPONS),
             },
             statusCode=201,