Improve demo seed
This commit is contained in:
Urtzi Alfaro
230
services/orders/scripts/demo/seed_demo_customers.py
Executable file
230
services/orders/scripts/demo/seed_demo_customers.py
Executable file
@@ -0,0 +1,230 @@
|
||||
#!/usr/bin/env python3
|
||||
# -*- coding: utf-8 -*-
|
||||
"""
|
||||
Demo Customer Seeding Script for Orders Service
|
||||
Creates customers for demo template tenants
|
||||
|
||||
This script runs as a Kubernetes init job inside the orders-service container.
|
||||
"""
|
||||
|
||||
import asyncio
|
||||
import uuid
|
||||
import sys
|
||||
import os
|
||||
import json
|
||||
from datetime import datetime, timezone, timedelta
|
||||
from pathlib import Path
|
||||
|
||||
# Add app to path
|
||||
sys.path.insert(0, str(Path(__file__).parent.parent.parent))
|
||||
|
||||
from sqlalchemy.ext.asyncio import AsyncSession, create_async_engine
|
||||
from sqlalchemy.orm import sessionmaker
|
||||
from sqlalchemy import select
|
||||
import structlog
|
||||
|
||||
from app.models.customer import Customer
|
||||
|
||||
# Configure logging
|
||||
logger = structlog.get_logger()
|
||||
|
||||
# Base demo tenant IDs
|
||||
DEMO_TENANT_SAN_PABLO = uuid.UUID("a1b2c3d4-e5f6-47a8-b9c0-d1e2f3a4b5c6") # Individual bakery
|
||||
DEMO_TENANT_LA_ESPIGA = uuid.UUID("b2c3d4e5-f6a7-48b9-c0d1-e2f3a4b5c6d7") # Central bakery
|
||||
|
||||
# Base reference date for date calculations
|
||||
BASE_REFERENCE_DATE = datetime(2025, 1, 15, 12, 0, 0, tzinfo=timezone.utc)
|
||||
|
||||
|
||||
def load_customer_data():
|
||||
"""Load customer data from JSON file"""
|
||||
data_file = Path(__file__).parent / "clientes_es.json"
|
||||
if not data_file.exists():
|
||||
raise FileNotFoundError(f"Customer data file not found: {data_file}")
|
||||
|
||||
with open(data_file, 'r', encoding='utf-8') as f:
|
||||
return json.load(f)
|
||||
|
||||
|
||||
def calculate_date_from_offset(offset_days: int) -> datetime:
|
||||
"""Calculate a date based on offset from BASE_REFERENCE_DATE"""
|
||||
return BASE_REFERENCE_DATE + timedelta(days=offset_days)
|
||||
|
||||
|
||||
async def seed_customers_for_tenant(
|
||||
db: AsyncSession,
|
||||
tenant_id: uuid.UUID,
|
||||
tenant_name: str,
|
||||
customer_list: list
|
||||
):
|
||||
"""Seed customers for a specific tenant"""
|
||||
logger.info(f"Seeding customers for: {tenant_name}", tenant_id=str(tenant_id))
|
||||
|
||||
# Check if customers already exist
|
||||
result = await db.execute(
|
||||
select(Customer).where(Customer.tenant_id == tenant_id).limit(1)
|
||||
)
|
||||
existing = result.scalar_one_or_none()
|
||||
|
||||
if existing:
|
||||
logger.info(f"Customers already exist for {tenant_name}, skipping seed")
|
||||
return {"tenant_id": str(tenant_id), "customers_created": 0, "skipped": True}
|
||||
|
||||
count = 0
|
||||
for customer_data in customer_list:
|
||||
# Calculate dates from offsets
|
||||
first_order_date = None
|
||||
if "first_order_offset_days" in customer_data:
|
||||
first_order_date = calculate_date_from_offset(customer_data["first_order_offset_days"])
|
||||
|
||||
last_order_date = None
|
||||
if "last_order_offset_days" in customer_data:
|
||||
last_order_date = calculate_date_from_offset(customer_data["last_order_offset_days"])
|
||||
|
||||
# Use strings directly (model doesn't use enums)
|
||||
customer_type = customer_data.get("customer_type", "business")
|
||||
customer_segment = customer_data.get("customer_segment", "regular")
|
||||
is_active = customer_data.get("status", "active") == "active"
|
||||
|
||||
# Create customer (using actual model fields)
|
||||
# For San Pablo, use original IDs. For La Espiga, generate new UUIDs
|
||||
if tenant_id == DEMO_TENANT_SAN_PABLO:
|
||||
customer_id = uuid.UUID(customer_data["id"])
|
||||
else:
|
||||
# Generate deterministic UUID for La Espiga based on original ID
|
||||
base_uuid = uuid.UUID(customer_data["id"])
|
||||
# Add a fixed offset to create a unique but deterministic ID
|
||||
customer_id = uuid.UUID(int=base_uuid.int + 0x10000000000000000000000000000000)
|
||||
|
||||
customer = Customer(
|
||||
id=customer_id,
|
||||
tenant_id=tenant_id,
|
||||
customer_code=customer_data["customer_code"],
|
||||
name=customer_data["name"],
|
||||
business_name=customer_data.get("business_name"),
|
||||
customer_type=customer_type,
|
||||
tax_id=customer_data.get("tax_id"),
|
||||
email=customer_data.get("email"),
|
||||
phone=customer_data.get("phone"),
|
||||
address_line1=customer_data.get("billing_address"),
|
||||
city=customer_data.get("billing_city"),
|
||||
state=customer_data.get("billing_state"),
|
||||
postal_code=customer_data.get("billing_postal_code"),
|
||||
country=customer_data.get("billing_country", "España"),
|
||||
is_active=is_active,
|
||||
preferred_delivery_method=customer_data.get("preferred_delivery_method", "delivery"),
|
||||
payment_terms=customer_data.get("payment_terms", "immediate"),
|
||||
credit_limit=customer_data.get("credit_limit"),
|
||||
discount_percentage=customer_data.get("discount_percentage", 0.0),
|
||||
customer_segment=customer_segment,
|
||||
priority_level=customer_data.get("priority_level", "normal"),
|
||||
special_instructions=customer_data.get("special_instructions"),
|
||||
total_orders=customer_data.get("total_orders", 0),
|
||||
total_spent=customer_data.get("total_revenue", 0.0),
|
||||
average_order_value=customer_data.get("average_order_value", 0.0),
|
||||
last_order_date=last_order_date,
|
||||
created_at=BASE_REFERENCE_DATE,
|
||||
updated_at=BASE_REFERENCE_DATE
|
||||
)
|
||||
|
||||
db.add(customer)
|
||||
count += 1
|
||||
logger.debug(f"Created customer: {customer.name}", customer_id=str(customer.id))
|
||||
|
||||
await db.commit()
|
||||
logger.info(f"Successfully created {count} customers for {tenant_name}")
|
||||
|
||||
return {
|
||||
"tenant_id": str(tenant_id),
|
||||
"customers_created": count,
|
||||
"skipped": False
|
||||
}
|
||||
|
||||
|
||||
async def seed_all(db: AsyncSession):
|
||||
"""Seed all demo tenants with customers"""
|
||||
logger.info("Starting demo customer seed process")
|
||||
|
||||
# Load customer data
|
||||
data = load_customer_data()
|
||||
|
||||
results = []
|
||||
|
||||
# Both tenants get the same customer base
|
||||
# (In real scenario, you might want different customer lists)
|
||||
result_san_pablo = await seed_customers_for_tenant(
|
||||
db,
|
||||
DEMO_TENANT_SAN_PABLO,
|
||||
"San Pablo - Individual Bakery",
|
||||
data["clientes"]
|
||||
)
|
||||
results.append(result_san_pablo)
|
||||
|
||||
result_la_espiga = await seed_customers_for_tenant(
|
||||
db,
|
||||
DEMO_TENANT_LA_ESPIGA,
|
||||
"La Espiga - Central Bakery",
|
||||
data["clientes"]
|
||||
)
|
||||
results.append(result_la_espiga)
|
||||
|
||||
total_created = sum(r["customers_created"] for r in results)
|
||||
|
||||
return {
|
||||
"results": results,
|
||||
"total_customers_created": total_created,
|
||||
"status": "completed"
|
||||
}
|
||||
|
||||
|
||||
async def main():
|
||||
"""Main execution function"""
|
||||
# Get database URL from environment
|
||||
database_url = os.getenv("ORDERS_DATABASE_URL")
|
||||
if not database_url:
|
||||
logger.error("ORDERS_DATABASE_URL environment variable must be set")
|
||||
return 1
|
||||
|
||||
# Ensure asyncpg driver
|
||||
if database_url.startswith("postgresql://"):
|
||||
database_url = database_url.replace("postgresql://", "postgresql+asyncpg://", 1)
|
||||
|
||||
# Create async engine
|
||||
engine = create_async_engine(database_url, echo=False)
|
||||
async_session = sessionmaker(engine, class_=AsyncSession, expire_on_commit=False)
|
||||
|
||||
try:
|
||||
async with async_session() as session:
|
||||
result = await seed_all(session)
|
||||
|
||||
logger.info(
|
||||
"Customer seed completed successfully!",
|
||||
total_customers=result["total_customers_created"],
|
||||
status=result["status"]
|
||||
)
|
||||
|
||||
# Print summary
|
||||
print("\n" + "="*60)
|
||||
print("DEMO CUSTOMER SEED SUMMARY")
|
||||
print("="*60)
|
||||
for tenant_result in result["results"]:
|
||||
tenant_id = tenant_result["tenant_id"]
|
||||
count = tenant_result["customers_created"]
|
||||
skipped = tenant_result.get("skipped", False)
|
||||
status = "SKIPPED (already exists)" if skipped else f"CREATED {count} customers"
|
||||
print(f"Tenant {tenant_id}: {status}")
|
||||
print(f"\nTotal Customers Created: {result['total_customers_created']}")
|
||||
print("="*60 + "\n")
|
||||
|
||||
return 0
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Customer seed failed: {str(e)}", exc_info=True)
|
||||
return 1
|
||||
finally:
|
||||
await engine.dispose()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
exit_code = asyncio.run(main())
|
||||
sys.exit(exit_code)
|
||||
Reference in New Issue
Block a user