Updated claude big changes

main.py

@@ -1,9 +1,11 @@
 import logging
 import sys
+import uuid
 from app.routers import fixstuff_auto
-from fastapi import FastAPI
+from fastapi import FastAPI, Request
 from app.routers import main, delivery, confirm, fixstuff_customer
 from fastapi.middleware.cors import CORSMiddleware
+from starlette.middleware.base import BaseHTTPMiddleware
 import os
 from config import load_config
 from sqlalchemy import create_engine, text
@@ -37,8 +39,14 @@ def setup_logging():
 
 logger = setup_logging()
 
-# Database setup
-engine = create_engine(ApplicationConfig.SQLALCHEMY_DATABASE_URI)
+# Database setup with connection pooling
+engine = create_engine(
+    ApplicationConfig.SQLALCHEMY_DATABASE_URI,
+    pool_pre_ping=True,
+    pool_size=5,
+    max_overflow=10,
+    pool_recycle=3600,
+)
 SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
 
 def check_db_connection():
@@ -56,19 +64,29 @@ def check_db_connection():
 app = FastAPI()
 
 
+# Request ID middleware for request tracking/correlation
+class RequestIDMiddleware(BaseHTTPMiddleware):
+    async def dispatch(self, request: Request, call_next):
+        request_id = request.headers.get("X-Request-ID") or str(uuid.uuid4())[:8]
+        request.state.request_id = request_id
+        response = await call_next(request)
+        response.headers["X-Request-ID"] = request_id
+        return response
+
+app.add_middleware(RequestIDMiddleware)
+
 app.include_router(main.router)
 app.include_router(delivery.router)
 app.include_router(confirm.router)
 app.include_router(fixstuff_auto.router)
 app.include_router(fixstuff_customer.router)
 
-# print(ApplicationConfig.origins)
 app.add_middleware(
     CORSMiddleware,
     allow_origins=ApplicationConfig.origins,
     allow_credentials=True,
-    allow_methods=["*"],
-    allow_headers=["*"],
+    allow_methods=["GET", "POST", "PUT", "DELETE", "OPTIONS"],
+    allow_headers=["Authorization", "Content-Type", "Accept", "Origin", "X-Requested-With"],
 )
 
 
@@ -87,7 +105,11 @@ async def startup_event():
         logger.info("🤖🤖🤖🤖🤖 Mode: Development 🤖🤖🤖🤖🤖")
     elif mode in ['PRODUCTION', 'PROD']:
         logger.info("💀💀💀💀💀💀💀💀💀💀 ⚠️ WARNING PRODUCTION 💀💀💀💀💀💀💀💀💀💀")
-    logger.info(f"DB: {ApplicationConfig.SQLALCHEMY_DATABASE_URI[:30]}...")
+    # Sanitize DB URI to avoid logging credentials
+    db_uri = ApplicationConfig.SQLALCHEMY_DATABASE_URI
+    if '@' in db_uri:
+        db_uri = db_uri.split('@')[-1]  # Only show host/db portion
+    logger.info(f"DB: ...@{db_uri[:50]}")
     logger.info(f"CORS: {len(ApplicationConfig.origins)} origins configured")
 
     # Test database connection