REFACTOR ALL APIs
This commit is contained in:
123
services/external/app/api/traffic_data.py
vendored
Normal file
123
services/external/app/api/traffic_data.py
vendored
Normal file
@@ -0,0 +1,123 @@
|
||||
# services/external/app/api/traffic_data.py
|
||||
"""
|
||||
Traffic Data API - Atomic CRUD operations on TrafficData model
|
||||
"""
|
||||
|
||||
from fastapi import APIRouter, Depends, HTTPException, Query, Path
|
||||
from typing import List, Optional
|
||||
from datetime import date
|
||||
from uuid import UUID
|
||||
import structlog
|
||||
|
||||
from app.schemas.traffic import TrafficDataResponse
|
||||
from app.services.traffic_service import TrafficService
|
||||
from shared.routing.route_builder import RouteBuilder
|
||||
from sqlalchemy.ext.asyncio import AsyncSession
|
||||
from app.core.database import get_db
|
||||
|
||||
route_builder = RouteBuilder('external')
|
||||
router = APIRouter(tags=["traffic-data"])
|
||||
logger = structlog.get_logger()
|
||||
|
||||
|
||||
def get_traffic_service():
|
||||
"""Dependency injection for TrafficService"""
|
||||
return TrafficService()
|
||||
|
||||
|
||||
@router.get(
|
||||
route_builder.build_base_route("traffic-data"),
|
||||
response_model=List[TrafficDataResponse]
|
||||
)
|
||||
async def list_traffic_data(
|
||||
tenant_id: UUID = Path(..., description="Tenant ID"),
|
||||
start_date: Optional[date] = Query(None),
|
||||
end_date: Optional[date] = Query(None),
|
||||
latitude: Optional[float] = Query(None),
|
||||
longitude: Optional[float] = Query(None),
|
||||
limit: int = Query(100, ge=1, le=1000),
|
||||
db: AsyncSession = Depends(get_db),
|
||||
traffic_service: TrafficService = Depends(get_traffic_service)
|
||||
):
|
||||
"""List stored traffic data records"""
|
||||
try:
|
||||
logger.info("Listing traffic data", tenant_id=tenant_id)
|
||||
|
||||
traffic_records = await traffic_service.get_stored_traffic_data(
|
||||
tenant_id=tenant_id,
|
||||
start_date=start_date,
|
||||
end_date=end_date,
|
||||
latitude=latitude,
|
||||
longitude=longitude,
|
||||
limit=limit,
|
||||
db=db
|
||||
)
|
||||
|
||||
return traffic_records
|
||||
|
||||
except Exception as e:
|
||||
logger.error("Failed to list traffic data", error=str(e), tenant_id=tenant_id)
|
||||
raise HTTPException(status_code=500, detail="Failed to retrieve traffic data")
|
||||
|
||||
|
||||
@router.get(
|
||||
route_builder.build_resource_detail_route("traffic-data", "traffic_id"),
|
||||
response_model=TrafficDataResponse
|
||||
)
|
||||
async def get_traffic_data(
|
||||
tenant_id: UUID = Path(..., description="Tenant ID"),
|
||||
traffic_id: UUID = Path(..., description="Traffic data ID"),
|
||||
db: AsyncSession = Depends(get_db),
|
||||
traffic_service: TrafficService = Depends(get_traffic_service)
|
||||
):
|
||||
"""Get a specific traffic data record"""
|
||||
try:
|
||||
logger.info("Getting traffic data", tenant_id=tenant_id, traffic_id=traffic_id)
|
||||
|
||||
traffic_record = await traffic_service.get_traffic_data_by_id(
|
||||
tenant_id=tenant_id,
|
||||
traffic_id=traffic_id,
|
||||
db=db
|
||||
)
|
||||
|
||||
if not traffic_record:
|
||||
raise HTTPException(status_code=404, detail="Traffic data not found")
|
||||
|
||||
return traffic_record
|
||||
|
||||
except HTTPException:
|
||||
raise
|
||||
except Exception as e:
|
||||
logger.error("Failed to get traffic data", error=str(e), tenant_id=tenant_id)
|
||||
raise HTTPException(status_code=500, detail="Failed to retrieve traffic data")
|
||||
|
||||
|
||||
@router.delete(
|
||||
route_builder.build_resource_detail_route("traffic-data", "traffic_id")
|
||||
)
|
||||
async def delete_traffic_data(
|
||||
tenant_id: UUID = Path(..., description="Tenant ID"),
|
||||
traffic_id: UUID = Path(..., description="Traffic data ID"),
|
||||
db: AsyncSession = Depends(get_db),
|
||||
traffic_service: TrafficService = Depends(get_traffic_service)
|
||||
):
|
||||
"""Delete a traffic data record"""
|
||||
try:
|
||||
logger.info("Deleting traffic data", tenant_id=tenant_id, traffic_id=traffic_id)
|
||||
|
||||
success = await traffic_service.delete_traffic_data(
|
||||
tenant_id=tenant_id,
|
||||
traffic_id=traffic_id,
|
||||
db=db
|
||||
)
|
||||
|
||||
if not success:
|
||||
raise HTTPException(status_code=404, detail="Traffic data not found")
|
||||
|
||||
return {"message": "Traffic data deleted successfully"}
|
||||
|
||||
except HTTPException:
|
||||
raise
|
||||
except Exception as e:
|
||||
logger.error("Failed to delete traffic data", error=str(e), tenant_id=tenant_id)
|
||||
raise HTTPException(status_code=500, detail="Failed to delete traffic data")
|
||||
Reference in New Issue
Block a user