bakery-ia/services/orders/app/main.py

# ================================================================
# services/orders/app/main.py
# ================================================================
"""
Orders Service - FastAPI Application
Customer orders management service
"""

from fastapi import FastAPI, Request
from sqlalchemy import text
from app.core.config import settings
from app.core.database import database_manager
from app.api.orders import router as orders_router
from app.api.customers import router as customers_router
from app.api.order_operations import router as order_operations_router
from app.api import audit, internal_demo
from shared.service_base import StandardFastAPIService


class OrdersService(StandardFastAPIService):
    """Orders Service with standardized setup"""

    expected_migration_version = "00001"

    async def on_startup(self, app):
        """Custom startup logic including migration verification"""
        await self.verify_migrations()
        await super().on_startup(app)

    async def verify_migrations(self):
        """Verify database schema matches the latest migrations."""
        try:
            async with self.database_manager.get_session() as session:
                result = await session.execute(text("SELECT version_num FROM alembic_version"))
                version = result.scalar()
                if version != self.expected_migration_version:
                    self.logger.error(f"Migration version mismatch: expected {self.expected_migration_version}, got {version}")
                    raise RuntimeError(f"Migration version mismatch: expected {self.expected_migration_version}, got {version}")
                self.logger.info(f"Migration verification successful: {version}")
        except Exception as e:
            self.logger.error(f"Migration verification failed: {e}")
            raise

    def __init__(self):
        # Define expected database tables for health checks
        orders_expected_tables = [
            'customers', 'customer_contacts', 'customer_orders', 'order_items',
            'order_status_history', 'audit_logs'
        ]

        super().__init__(
            service_name="orders-service",
            app_name=settings.APP_NAME,
            description=settings.DESCRIPTION,
            version=settings.VERSION,
            api_prefix="",  # Empty because RouteBuilder already includes /api/v1
            database_manager=database_manager,
            expected_tables=orders_expected_tables
        )

    async def on_startup(self, app: FastAPI):
        """Custom startup logic for orders service"""
        # REMOVED: Procurement scheduler service initialization
        # Procurement scheduling is now handled by the Orchestrator Service
        # which calls the Procurement Service's /auto-generate endpoint
        pass

    async def on_shutdown(self, app: FastAPI):
        """Custom shutdown logic for orders service"""
        # REMOVED: Scheduler service shutdown
        pass

    def get_service_features(self):
        """Return orders-specific features"""
        return [
            "customer_management",
            "order_processing",
            "order_tracking"
        ]


# Create service instance
service = OrdersService()

# Create FastAPI app with standardized setup
app = service.create_app()

# Setup standard endpoints
service.setup_standard_endpoints()

# Include routers - organized by ATOMIC and BUSINESS operations
# IMPORTANT: Register specific routes (audit, customers) BEFORE parameterized routes (orders)
# to avoid route matching conflicts where {order_id} would match literal paths like "audit-logs"

# AUDIT: Audit log retrieval endpoints - Must be registered FIRST
service.add_router(audit.router)

# ATOMIC: Direct CRUD operations
# NOTE: Register customers_router BEFORE orders_router to ensure /customers
# matches before the parameterized /{order_id} route
service.add_router(customers_router)
service.add_router(orders_router)

# BUSINESS: Complex operations and workflows
service.add_router(order_operations_router)

# INTERNAL: Service-to-service endpoints - DEPRECATED: Replaced by script-based seed data loading
service.add_router(internal_demo.router, tags=["internal-demo"])

# REMOVED: test_procurement_scheduler endpoint
# Procurement scheduling is now triggered by the Orchestrator Service


@app.middleware("http")
async def logging_middleware(request: Request, call_next):
    """Add request logging middleware"""
    import time

    start_time = time.time()
    response = await call_next(request)
    process_time = time.time() - start_time

    service.logger.info("HTTP request processed",
                method=request.method,
                url=str(request.url),
                status_code=response.status_code,
                process_time=round(process_time, 4))

    return response


if __name__ == "__main__":
    import uvicorn
    uvicorn.run(
        "main:app",
        host="0.0.0.0",
        port=8000,
        reload=settings.DEBUG
    )