24 lines
1.2 KiB
Python
24 lines
1.2 KiB
Python
from fastapi import APIRouter, Depends, HTTPException
|
|
from sqlalchemy.orm import Session
|
|
from . import services, dto
|
|
from core.database import get_db
|
|
|
|
router = APIRouter()
|
|
|
|
@router.get("/transport-modes/{mode_key}", response_model=dto.TransportModeDTO)
|
|
def get_transport_mode(mode_key: str, db: Session = Depends(get_db)):
|
|
transport_mode = services.TransportModeService.get_transport_mode_by_key(db, mode_key)
|
|
if not transport_mode:
|
|
raise HTTPException(status_code=404, detail="Transport mode not found")
|
|
return transport_mode
|
|
|
|
@router.post("/transport-modes", response_model=dto.TransportModeDTO)
|
|
def create_transport_mode(transport_mode_data: dto.TransportModeCreateDTO, db: Session = Depends(get_db)):
|
|
return services.TransportModeService.create_transport_mode(db, transport_mode_data)
|
|
|
|
@router.delete("/transport-modes/{mode_key}", response_model=dto.TransportModeDTO)
|
|
def delete_transport_mode(mode_key: str, db: Session = Depends(get_db)):
|
|
transport_mode = services.TransportModeService.delete_transport_mode(db, mode_key)
|
|
if not transport_mode:
|
|
raise HTTPException(status_code=404, detail="Transport mode not found")
|
|
return transport_mode |