Improve demo seed

services/production/app/api/internal_demo.py

@@ -9,7 +9,7 @@ from sqlalchemy import select
 import structlog
 import uuid
 from datetime import datetime, timezone, timedelta
-from typing import Optional
+from typing import Optional, Dict, Any
 import os
 
 from app.core.database import get_db
@@ -19,6 +19,8 @@ from app.models.production import (
     ProductionStatus, ProductionPriority, ProcessStage,
     EquipmentStatus, EquipmentType
 )
+from shared.utils.demo_dates import adjust_date_for_demo, BASE_REFERENCE_DATE
+from shared.utils.alert_generator import generate_equipment_alerts
 
 logger = structlog.get_logger()
 router = APIRouter(prefix="/internal/demo", tags=["internal"])
@@ -45,6 +47,7 @@ async def clone_demo_data(
     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)
 ):
@@ -64,18 +67,29 @@ async def clone_demo_data(
         virtual_tenant_id: Target virtual tenant UUID
         demo_account_type: Type of demo account
         session_id: Originating session ID for tracing
+        session_created_at: Session creation timestamp for date adjustment
 
     Returns:
         Cloning status and record counts
     """
     start_time = datetime.now(timezone.utc)
 
+    # Parse session creation time for date adjustment
+    if session_created_at:
+        try:
+            session_time = datetime.fromisoformat(session_created_at.replace('Z', '+00:00'))
+        except (ValueError, AttributeError):
+            session_time = start_time
+    else:
+        session_time = start_time
+
     logger.info(
         "Starting production data cloning",
         base_tenant_id=base_tenant_id,
         virtual_tenant_id=virtual_tenant_id,
         demo_account_type=demo_account_type,
-        session_id=session_id
+        session_id=session_id,
+        session_created_at=session_created_at
     )
 
     try:
@@ -90,7 +104,8 @@ async def clone_demo_data(
             "production_capacity": 0,
             "quality_check_templates": 0,
             "quality_checks": 0,
-            "equipment": 0
+            "equipment": 0,
+            "alerts_generated": 0
         }
 
         # ID mappings
@@ -114,6 +129,17 @@ async def clone_demo_data(
             new_equipment_id = uuid.uuid4()
             equipment_id_map[equipment.id] = new_equipment_id
 
+            # Adjust dates relative to session creation time
+            adjusted_install_date = adjust_date_for_demo(
+                equipment.install_date, session_time, BASE_REFERENCE_DATE
+            )
+            adjusted_last_maintenance = adjust_date_for_demo(
+                equipment.last_maintenance_date, session_time, BASE_REFERENCE_DATE
+            )
+            adjusted_next_maintenance = adjust_date_for_demo(
+                equipment.next_maintenance_date, session_time, BASE_REFERENCE_DATE
+            )
+
             new_equipment = Equipment(
                 id=new_equipment_id,
                 tenant_id=virtual_uuid,
@@ -123,9 +149,9 @@ async def clone_demo_data(
                 serial_number=equipment.serial_number,
                 location=equipment.location,
                 status=equipment.status,
-                install_date=equipment.install_date,
-                last_maintenance_date=equipment.last_maintenance_date,
-                next_maintenance_date=equipment.next_maintenance_date,
+                install_date=adjusted_install_date,
+                last_maintenance_date=adjusted_last_maintenance,
+                next_maintenance_date=adjusted_next_maintenance,
                 maintenance_interval_days=equipment.maintenance_interval_days,
                 efficiency_percentage=equipment.efficiency_percentage,
                 uptime_percentage=equipment.uptime_percentage,
@@ -137,8 +163,8 @@ async def clone_demo_data(
                 target_temperature=equipment.target_temperature,
                 is_active=equipment.is_active,
                 notes=equipment.notes,
-                created_at=datetime.now(timezone.utc),
-                updated_at=datetime.now(timezone.utc)
+                created_at=session_time,
+                updated_at=session_time
             )
             db.add(new_equipment)
             stats["equipment"] += 1
@@ -185,8 +211,8 @@ async def clone_demo_data(
                 tolerance_percentage=template.tolerance_percentage,
                 applicable_stages=template.applicable_stages,
                 created_by=template.created_by,
-                created_at=datetime.now(timezone.utc),
-                updated_at=datetime.now(timezone.utc)
+                created_at=session_time,
+                updated_at=session_time
             )
             db.add(new_template)
             stats["quality_check_templates"] += 1
@@ -403,9 +429,18 @@ async def clone_demo_data(
             db.add(new_capacity)
             stats["production_capacity"] += 1
 
-        # Commit all changes
+        # Commit cloned data first
         await db.commit()
 
+        # Generate equipment maintenance and status alerts
+        try:
+            alerts_count = await generate_equipment_alerts(db, virtual_uuid, session_time)
+            stats["alerts_generated"] += alerts_count
+            await db.commit()
+            logger.info(f"Generated {alerts_count} equipment alerts")
+        except Exception as alert_error:
+            logger.warning(f"Alert generation failed: {alert_error}", exc_info=True)
+
         total_records = sum(stats.values())
         duration_ms = int((datetime.now(timezone.utc) - start_time).total_seconds() * 1000)