setup_demo.py

#!/usr/bin/env python3
"""
Setup demo environment with working credentials
"""
import sys
import os

sys.path.append(os.path.dirname(os.path.abspath(__file__)))

from app.core.tenant import tenant_registry
from app.core.auth import user_manager
from app.utils.seeder import DatabaseSeeder
from app.core.database import get_db_session


def setup_demo():
    """Setup demo environment"""
    tenant_id = "ui_test_tenant"

    print(f"🚀 Setting up demo environment...")

    try:
        # Register tenant
        print(f"📁 Registering tenant: {tenant_id}")
        success = tenant_registry.register_tenant(
            tenant_id,
            {
                "features": {
                    "inventory_tracking": True,
                    "product_variants": True,
                    "customer_accounts": True,
                    "order_management": True,
                }
            },
        )
        print(f"✅ Tenant registration: {success}")

        # Check if data already exists
        print(f"📦 Checking existing data...")
        try:
            db = next(get_db_session(tenant_id))
            from app.models import Category, Product

            categories = db.query(Category).count()
            products = db.query(Product).count()

            print(f"   Categories: {categories}")
            print(f"   Products: {products}")

            if categories == 0:
                print(f"   Seeding categories...")
                seeder = DatabaseSeeder(db)
                seeder.seed_categories()
                db.commit()

            if products == 0:
                print(f"   Seeding products...")
                seeder = DatabaseSeeder(db)
                seeder.seed_products()
                db.commit()

            db.close()
            print(f"✅ Data check complete")
        except Exception as e:
            print(f"⚠️ Data check failed (but continuing): {e}")

        # Test authentication
        print(f"🔐 Testing authentication...")
        user = user_manager.authenticate_tenant_user(tenant_id, "admin", "admin123")
        if user:
            print(f"✅ Authentication working!")
            print(f"   User: {user.get('username')}")
            print(f"   Roles: {user.get('roles')}")
        else:
            print(f"❌ Authentication failed")

        print(f"\n🎉 Demo setup complete!")
        print(f"   Frontend: http://localhost:5173")
        print(f"   Backend: http://localhost:8000")
        print(f"   Tenant ID: {tenant_id}")
        print(f"   Username: admin")
        print(f"   Password: admin123")

    except Exception as e:
        print(f"❌ Error setting up demo: {e}")
        import traceback

        traceback.print_exc()


if __name__ == "__main__":
    setup_demo()