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

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

from fastapi import FastAPI, Request
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.procurement import router as procurement_router
from app.services.procurement_scheduler_service import ProcurementSchedulerService
from shared.service_base import StandardFastAPIService


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

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

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

    async def on_startup(self, app: FastAPI):
        """Custom startup logic for orders service"""
        # Initialize procurement scheduler service
        scheduler_service = ProcurementSchedulerService(settings)
        await scheduler_service.start()
        self.logger.info("Procurement scheduler service started")

        # Store scheduler service in app state
        app.state.scheduler_service = scheduler_service

    async def on_shutdown(self, app: FastAPI):
        """Custom shutdown logic for orders service"""
        # Stop scheduler service
        if hasattr(app.state, 'scheduler_service'):
            await app.state.scheduler_service.stop()
            self.logger.info("Scheduler service stopped")

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


# Create service instance
service = OrdersService()

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

# Setup standard endpoints
service.setup_standard_endpoints()

# Include routers
service.add_router(orders_router)
service.add_router(procurement_router)


@app.post("/test/procurement-scheduler")
async def test_procurement_scheduler():
    """Test endpoint to manually trigger procurement scheduler"""
    try:
        if hasattr(app.state, 'scheduler_service'):
            scheduler_service = app.state.scheduler_service
            await scheduler_service.test_procurement_generation()
            return {"message": "Procurement scheduler test triggered successfully"}
        else:
            return {"error": "Scheduler service not available"}
    except Exception as e:
        logger.error("Error testing procurement scheduler", error=str(e))
        return {"error": f"Failed to trigger scheduler test: {str(e)}"}


@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
    
    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
    )