REFACTOR API gateway

services/data/app/api/traffic.py

@@ -3,11 +3,12 @@
 # ================================================================
 """Traffic data API endpoints with improved error handling"""
 
-from fastapi import APIRouter, Depends, HTTPException, Query
-from sqlalchemy.ext.asyncio import AsyncSession
+from fastapi import APIRouter, Depends, HTTPException, Query, Path
 from typing import List, Dict, Any
 from datetime import datetime, timedelta
 import structlog
+from uuid import UUID
+from sqlalchemy.ext.asyncio import AsyncSession
 
 from app.core.database import get_db
 from app.services.traffic_service import TrafficService
@@ -23,14 +24,15 @@ from shared.auth.decorators import (
     get_current_tenant_id_dep
 )
 
-router = APIRouter()
+router = APIRouter(tags=["traffic"])
 traffic_service = TrafficService()
 logger = structlog.get_logger()
 
-@router.get("/current", response_model=TrafficDataResponse)
+@router.get("/tenants/{tenant_id}/current", response_model=TrafficDataResponse)
 async def get_current_traffic(
     latitude: float = Query(..., description="Latitude"),
     longitude: float = Query(..., description="Longitude"),
+    tenant_id: UUID = Path(..., description="Tenant ID"),
     current_user: Dict[str, Any] = Depends(get_current_user_dep),
 ):
     """Get current traffic data for location"""
@@ -69,13 +71,14 @@ async def get_current_traffic(
         logger.error("Traffic API traceback", traceback=traceback.format_exc())
         raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
 
-@router.get("/historical", response_model=List[TrafficDataResponse])
+@router.get("/tenants/{tenant_id}/historical", response_model=List[TrafficDataResponse])
 async def get_historical_traffic(
     latitude: float = Query(..., description="Latitude"),
     longitude: float = Query(..., description="Longitude"),
     start_date: datetime = Query(..., description="Start date"),
     end_date: datetime = Query(..., description="End date"),
     db: AsyncSession = Depends(get_db),
+    tenant_id: UUID = Path(..., description="Tenant ID"),
     current_user: Dict[str, Any] = Depends(get_current_user_dep),
 ):
     """Get historical traffic data"""
@@ -115,11 +118,12 @@ async def get_historical_traffic(
         logger.error("Unexpected error in historical traffic API", error=str(e))
         raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
 
-@router.post("/store")
+@router.post("/tenants/{tenant_id}/store")
 async def store_traffic_data(
     latitude: float = Query(..., description="Latitude"),
     longitude: float = Query(..., description="Longitude"),
     db: AsyncSession = Depends(get_db),
+    tenant_id: UUID = Path(..., description="Tenant ID"),
     current_user: Dict[str, Any] = Depends(get_current_user_dep)
 ):
     """Store current traffic data to database"""