287 lines
10 KiB
Python
287 lines
10 KiB
Python
"""
|
|
Internal Demo API Endpoints for Orchestrator Service
|
|
Used by demo_session service to clone data for virtual demo tenants
|
|
"""
|
|
|
|
from fastapi import APIRouter, Depends, HTTPException, Header
|
|
from typing import Dict, Any
|
|
from uuid import UUID
|
|
import structlog
|
|
import os
|
|
import json
|
|
|
|
from app.core.database import get_db
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
from sqlalchemy import select, delete, func
|
|
from app.models.orchestration_run import OrchestrationRun, OrchestrationStatus
|
|
import uuid
|
|
from datetime import datetime, timezone, timedelta
|
|
from typing import Optional
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
# Add shared utilities to path
|
|
sys.path.insert(0, str(Path(__file__).parent.parent.parent.parent))
|
|
from shared.utils.demo_dates import adjust_date_for_demo
|
|
|
|
from app.core.config import settings
|
|
|
|
router = APIRouter(prefix="/internal/demo", tags=["internal"])
|
|
logger = structlog.get_logger()
|
|
|
|
|
|
async def ensure_unique_run_number(db: AsyncSession, base_run_number: str) -> str:
|
|
"""Ensure the run number is unique by appending a suffix if needed"""
|
|
proposed_run_number = base_run_number
|
|
|
|
# Check if the proposed run number already exists in the database
|
|
while True:
|
|
result = await db.execute(
|
|
select(OrchestrationRun)
|
|
.where(OrchestrationRun.run_number == proposed_run_number)
|
|
)
|
|
existing_run = result.scalar_one_or_none()
|
|
|
|
if not existing_run:
|
|
# Run number is unique, return it
|
|
return proposed_run_number
|
|
|
|
# Generate a new run number with an additional random suffix
|
|
random_suffix = str(uuid.uuid4())[:4].upper()
|
|
proposed_run_number = f"{base_run_number[:50-len(random_suffix)-1]}-{random_suffix}"
|
|
|
|
|
|
def verify_internal_api_key(x_internal_api_key: str = Header(...)):
|
|
"""Verify internal API key for service-to-service communication"""
|
|
if x_internal_api_key != settings.INTERNAL_API_KEY:
|
|
raise HTTPException(status_code=403, detail="Invalid internal API key")
|
|
return True
|
|
|
|
|
|
async def load_fixture_data_for_tenant(
|
|
db: AsyncSession,
|
|
tenant_uuid: UUID,
|
|
demo_account_type: str,
|
|
reference_time: datetime,
|
|
base_tenant_id: Optional[str] = None
|
|
) -> int:
|
|
"""
|
|
Load orchestration run data from JSON fixture directly into the virtual tenant.
|
|
Returns the number of runs created.
|
|
"""
|
|
from shared.utils.seed_data_paths import get_seed_data_path
|
|
from shared.utils.demo_dates import resolve_time_marker, adjust_date_for_demo
|
|
|
|
# Load fixture data
|
|
if demo_account_type == "enterprise_child" and base_tenant_id:
|
|
json_file = get_seed_data_path("enterprise", "11-orchestrator.json", child_id=base_tenant_id)
|
|
else:
|
|
json_file = get_seed_data_path(demo_account_type, "11-orchestrator.json")
|
|
|
|
with open(json_file, 'r', encoding='utf-8') as f:
|
|
fixture_data = json.load(f)
|
|
|
|
orchestration_run_data = fixture_data.get("orchestration_run")
|
|
if not orchestration_run_data:
|
|
logger.warning("No orchestration_run data in fixture")
|
|
return 0
|
|
|
|
# Parse and adjust dates from fixture to reference_time
|
|
base_started_at = resolve_time_marker(orchestration_run_data.get("started_at"), reference_time)
|
|
base_completed_at = resolve_time_marker(orchestration_run_data.get("completed_at"), reference_time)
|
|
|
|
# Adjust dates to make them appear recent relative to session creation
|
|
started_at = adjust_date_for_demo(base_started_at, reference_time) if base_started_at else reference_time - timedelta(hours=2)
|
|
completed_at = adjust_date_for_demo(base_completed_at, reference_time) if base_completed_at else started_at + timedelta(minutes=15)
|
|
|
|
# Generate unique run number with session context
|
|
current_year = reference_time.year
|
|
unique_suffix = str(uuid.uuid4())[:8].upper()
|
|
run_number = f"ORCH-DEMO-PROF-{current_year}-001-{unique_suffix}"
|
|
|
|
# Create orchestration run for virtual tenant
|
|
new_run = OrchestrationRun(
|
|
id=uuid.uuid4(), # Generate new UUID
|
|
tenant_id=tenant_uuid,
|
|
run_number=run_number,
|
|
status=OrchestrationStatus[orchestration_run_data["status"]],
|
|
run_type=orchestration_run_data.get("run_type", "daily"),
|
|
priority="normal",
|
|
started_at=started_at,
|
|
completed_at=completed_at,
|
|
duration_seconds=orchestration_run_data.get("duration_seconds", 900),
|
|
|
|
# Step statuses from orchestration_results
|
|
forecasting_status="success",
|
|
forecasting_started_at=started_at,
|
|
forecasting_completed_at=started_at + timedelta(minutes=2),
|
|
|
|
production_status="success",
|
|
production_started_at=started_at + timedelta(minutes=2),
|
|
production_completed_at=started_at + timedelta(minutes=5),
|
|
|
|
procurement_status="success",
|
|
procurement_started_at=started_at + timedelta(minutes=5),
|
|
procurement_completed_at=started_at + timedelta(minutes=8),
|
|
|
|
notification_status="success",
|
|
notification_started_at=started_at + timedelta(minutes=8),
|
|
notification_completed_at=completed_at,
|
|
|
|
# Results from orchestration_results
|
|
forecasts_generated=fixture_data.get("orchestration_results", {}).get("forecasts_generated", 10),
|
|
production_batches_created=fixture_data.get("orchestration_results", {}).get("production_batches_created", 18),
|
|
procurement_plans_created=0,
|
|
purchase_orders_created=fixture_data.get("orchestration_results", {}).get("purchase_orders_created", 6),
|
|
notifications_sent=fixture_data.get("orchestration_results", {}).get("notifications_sent", 8),
|
|
|
|
# Metadata
|
|
triggered_by="system",
|
|
created_at=started_at,
|
|
updated_at=completed_at
|
|
)
|
|
|
|
db.add(new_run)
|
|
await db.flush()
|
|
|
|
logger.info(
|
|
"Loaded orchestration run from fixture",
|
|
tenant_id=str(tenant_uuid),
|
|
run_number=new_run.run_number,
|
|
started_at=started_at.isoformat()
|
|
)
|
|
|
|
return 1
|
|
|
|
|
|
@router.post("/clone")
|
|
async def clone_demo_data(
|
|
base_tenant_id: str,
|
|
virtual_tenant_id: str,
|
|
demo_account_type: str,
|
|
session_id: Optional[str] = None,
|
|
session_created_at: Optional[str] = None,
|
|
db: AsyncSession = Depends(get_db),
|
|
_: bool = Depends(verify_internal_api_key)
|
|
):
|
|
"""
|
|
Clone orchestration run demo data from base tenant to virtual tenant
|
|
|
|
This endpoint is called by the demo_session service during session initialization.
|
|
It clones orchestration runs with date adjustments to make them appear recent.
|
|
|
|
If the base tenant has no orchestration runs, it will first seed them from the fixture.
|
|
"""
|
|
|
|
start_time = datetime.now(timezone.utc)
|
|
|
|
# Parse session_created_at or use current time
|
|
if session_created_at:
|
|
try:
|
|
reference_time = datetime.fromisoformat(session_created_at.replace('Z', '+00:00'))
|
|
except:
|
|
reference_time = datetime.now(timezone.utc)
|
|
else:
|
|
reference_time = datetime.now(timezone.utc)
|
|
|
|
logger.info(
|
|
"Starting orchestration runs cloning with date adjustment",
|
|
base_tenant_id=base_tenant_id,
|
|
virtual_tenant_id=virtual_tenant_id,
|
|
demo_account_type=demo_account_type,
|
|
session_id=session_id,
|
|
reference_time=reference_time.isoformat()
|
|
)
|
|
|
|
try:
|
|
virtual_uuid = uuid.UUID(virtual_tenant_id)
|
|
|
|
# Load fixture data directly into virtual tenant (no base tenant cloning)
|
|
runs_created = await load_fixture_data_for_tenant(
|
|
db,
|
|
virtual_uuid,
|
|
demo_account_type,
|
|
reference_time,
|
|
base_tenant_id
|
|
)
|
|
|
|
await db.commit()
|
|
|
|
duration_ms = int((datetime.now(timezone.utc) - start_time).total_seconds() * 1000)
|
|
|
|
logger.info(
|
|
"Orchestration runs loaded from fixture successfully",
|
|
virtual_tenant_id=str(virtual_tenant_id),
|
|
runs_created=runs_created,
|
|
duration_ms=duration_ms
|
|
)
|
|
|
|
return {
|
|
"service": "orchestrator",
|
|
"status": "completed",
|
|
"success": True,
|
|
"records_cloned": runs_created,
|
|
"runs_cloned": runs_created,
|
|
"duration_ms": duration_ms
|
|
}
|
|
|
|
except Exception as e:
|
|
logger.error("Failed to clone orchestration runs", error=str(e), exc_info=True)
|
|
await db.rollback()
|
|
raise HTTPException(status_code=500, detail=f"Failed to clone orchestration runs: {str(e)}")
|
|
|
|
|
|
@router.delete("/tenant/{virtual_tenant_id}")
|
|
async def delete_demo_data(
|
|
virtual_tenant_id: str,
|
|
db: AsyncSession = Depends(get_db),
|
|
_: bool = Depends(verify_internal_api_key)
|
|
):
|
|
"""Delete all orchestration runs for a virtual demo tenant"""
|
|
logger.info("Deleting orchestration runs for virtual tenant", virtual_tenant_id=virtual_tenant_id)
|
|
start_time = datetime.now(timezone.utc)
|
|
|
|
try:
|
|
virtual_uuid = uuid.UUID(virtual_tenant_id)
|
|
|
|
# Count records
|
|
run_count = await db.scalar(
|
|
select(func.count(OrchestrationRun.id))
|
|
.where(OrchestrationRun.tenant_id == virtual_uuid)
|
|
)
|
|
|
|
# Delete orchestration runs
|
|
await db.execute(
|
|
delete(OrchestrationRun)
|
|
.where(OrchestrationRun.tenant_id == virtual_uuid)
|
|
)
|
|
await db.commit()
|
|
|
|
duration_ms = int((datetime.now(timezone.utc) - start_time).total_seconds() * 1000)
|
|
logger.info(
|
|
"Orchestration runs deleted successfully",
|
|
virtual_tenant_id=virtual_tenant_id,
|
|
duration_ms=duration_ms
|
|
)
|
|
|
|
return {
|
|
"service": "orchestrator",
|
|
"status": "deleted",
|
|
"virtual_tenant_id": virtual_tenant_id,
|
|
"records_deleted": {
|
|
"orchestration_runs": run_count,
|
|
"total": run_count
|
|
},
|
|
"duration_ms": duration_ms
|
|
}
|
|
except Exception as e:
|
|
logger.error("Failed to delete orchestration runs", error=str(e), exc_info=True)
|
|
await db.rollback()
|
|
raise HTTPException(status_code=500, detail=str(e))
|
|
|
|
|
|
@router.get("/clone/health")
|
|
async def health_check(_: bool = Depends(verify_internal_api_key)):
|
|
"""Health check for demo cloning endpoint"""
|
|
return {"status": "healthy", "service": "orchestrator"}
|