145 lines
5.0 KiB
Python
145 lines
5.0 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Seed Demo Tenants
|
|
Creates base demo tenant templates with Spanish data
|
|
"""
|
|
|
|
import asyncio
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
project_root = Path(__file__).parent.parent.parent
|
|
sys.path.insert(0, str(project_root))
|
|
|
|
import os
|
|
os.environ.setdefault("TENANT_DATABASE_URL", os.getenv("TENANT_DATABASE_URL"))
|
|
|
|
from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession, async_sessionmaker
|
|
from sqlalchemy import select
|
|
import structlog
|
|
import uuid
|
|
from datetime import datetime, timezone
|
|
|
|
logger = structlog.get_logger()
|
|
|
|
# Demo tenant configurations
|
|
DEMO_TENANTS = [
|
|
{
|
|
"id": "a1b2c3d4-e5f6-47a8-b9c0-d1e2f3a4b5c6",
|
|
"name": "Panadería San Pablo - Demo",
|
|
"subdomain": "demo-sanpablo",
|
|
"business_type": "bakery",
|
|
"business_model": "individual_bakery",
|
|
"owner_id": "c1a2b3c4-d5e6-47a8-b9c0-d1e2f3a4b5c6", # María García
|
|
"address": "Calle Mayor, 15",
|
|
"city": "Madrid",
|
|
"postal_code": "28013",
|
|
"latitude": 40.4168,
|
|
"longitude": -3.7038,
|
|
"phone": "+34 912 345 678",
|
|
"email": "contacto@panaderiasanpablo.com",
|
|
"subscription_tier": "professional",
|
|
"is_active": True,
|
|
"is_demo": True,
|
|
"is_demo_template": True,
|
|
"ml_model_trained": True,
|
|
},
|
|
{
|
|
"id": "b2c3d4e5-f6a7-48b9-c0d1-e2f3a4b5c6d7",
|
|
"name": "Panadería La Espiga - Demo",
|
|
"subdomain": "demo-laespiga",
|
|
"business_type": "bakery",
|
|
"business_model": "central_baker_satellite",
|
|
"owner_id": "d2e3f4a5-b6c7-48d9-e0f1-a2b3c4d5e6f7", # Carlos Martínez
|
|
"address": "Avenida de la Constitución, 42",
|
|
"city": "Barcelona",
|
|
"postal_code": "08001",
|
|
"latitude": 41.3851,
|
|
"longitude": 2.1734,
|
|
"phone": "+34 913 456 789",
|
|
"email": "contacto@panaderialaespiga.com",
|
|
"subscription_tier": "enterprise",
|
|
"is_active": True,
|
|
"is_demo": True,
|
|
"is_demo_template": True,
|
|
"ml_model_trained": True,
|
|
}
|
|
]
|
|
|
|
|
|
async def seed_demo_tenants():
|
|
"""Seed demo tenants into tenant database"""
|
|
|
|
database_url = os.getenv("TENANT_DATABASE_URL")
|
|
if not database_url:
|
|
logger.error("TENANT_DATABASE_URL environment variable not set")
|
|
return False
|
|
|
|
logger.info("Connecting to tenant database", url=database_url.split("@")[-1])
|
|
|
|
engine = create_async_engine(database_url, echo=False)
|
|
session_factory = async_sessionmaker(engine, class_=AsyncSession, expire_on_commit=False)
|
|
|
|
try:
|
|
async with session_factory() as session:
|
|
try:
|
|
from app.models.tenants import Tenant
|
|
except ImportError:
|
|
from services.tenant.app.models.tenants import Tenant
|
|
|
|
for tenant_data in DEMO_TENANTS:
|
|
# Check if tenant already exists
|
|
result = await session.execute(
|
|
select(Tenant).where(Tenant.subdomain == tenant_data["subdomain"])
|
|
)
|
|
existing_tenant = result.scalar_one_or_none()
|
|
|
|
if existing_tenant:
|
|
logger.info(f"Demo tenant already exists: {tenant_data['subdomain']}")
|
|
continue
|
|
|
|
# Create new demo tenant
|
|
tenant = Tenant(
|
|
id=uuid.UUID(tenant_data["id"]),
|
|
name=tenant_data["name"],
|
|
subdomain=tenant_data["subdomain"],
|
|
business_type=tenant_data["business_type"],
|
|
business_model=tenant_data["business_model"],
|
|
owner_id=uuid.UUID(tenant_data["owner_id"]),
|
|
address=tenant_data["address"],
|
|
city=tenant_data["city"],
|
|
postal_code=tenant_data["postal_code"],
|
|
latitude=tenant_data.get("latitude"),
|
|
longitude=tenant_data.get("longitude"),
|
|
phone=tenant_data.get("phone"),
|
|
email=tenant_data.get("email"),
|
|
subscription_tier=tenant_data["subscription_tier"],
|
|
is_active=tenant_data["is_active"],
|
|
is_demo=tenant_data["is_demo"],
|
|
is_demo_template=tenant_data["is_demo_template"],
|
|
ml_model_trained=tenant_data.get("ml_model_trained", False),
|
|
created_at=datetime.now(timezone.utc),
|
|
updated_at=datetime.now(timezone.utc)
|
|
)
|
|
|
|
session.add(tenant)
|
|
logger.info(f"Created demo tenant: {tenant_data['name']}")
|
|
|
|
await session.commit()
|
|
logger.info("Demo tenants seeded successfully")
|
|
return True
|
|
|
|
except Exception as e:
|
|
logger.error(f"Failed to seed demo tenants: {str(e)}")
|
|
import traceback
|
|
traceback.print_exc()
|
|
return False
|
|
|
|
finally:
|
|
await engine.dispose()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
result = asyncio.run(seed_demo_tenants())
|
|
sys.exit(0 if result else 1)
|