87 lines
2.5 KiB
Python
87 lines
2.5 KiB
Python
from fastapi import APIRouter, Depends, Query, HTTPException
|
|
from sqlmodel import Session, select
|
|
from typing import Optional
|
|
from uuid import UUID
|
|
|
|
from app.core.database import get_session
|
|
from app.models.bus_schedule import BusSchedule
|
|
from app.api.deps import get_current_admin
|
|
|
|
router = APIRouter(prefix="/api/schedules", tags=["schedules"])
|
|
|
|
|
|
@router.get("")
|
|
async def get_schedules(
|
|
route_id: Optional[UUID] = Query(None),
|
|
stop_id: Optional[UUID] = Query(None),
|
|
only_published: bool = Query(True),
|
|
session: Session = Depends(get_session)
|
|
):
|
|
"""Get schedules for a route or stop."""
|
|
statement = select(BusSchedule)
|
|
|
|
if only_published:
|
|
statement = statement.where(BusSchedule.is_published)
|
|
|
|
if route_id:
|
|
statement = statement.where(BusSchedule.route_id == route_id)
|
|
|
|
if stop_id:
|
|
from app.models.route_stop import RouteStop
|
|
statement = statement.join(
|
|
RouteStop, BusSchedule.route_id == RouteStop.route_id
|
|
).where(RouteStop.stop_id == stop_id)
|
|
|
|
schedules = session.exec(statement).all()
|
|
return schedules
|
|
|
|
@router.post("")
|
|
async def create_schedule(
|
|
schedule: BusSchedule,
|
|
session: Session = Depends(get_session),
|
|
_: bool = Depends(get_current_admin)
|
|
):
|
|
"""Create a new bus schedule (Admin only)."""
|
|
db_schedule = BusSchedule.model_validate(schedule)
|
|
session.add(db_schedule)
|
|
session.commit()
|
|
session.refresh(db_schedule)
|
|
return db_schedule
|
|
|
|
@router.put("/{schedule_id}")
|
|
async def update_schedule(
|
|
schedule_id: UUID,
|
|
schedule_update: dict,
|
|
session: Session = Depends(get_session),
|
|
_: bool = Depends(get_current_admin)
|
|
):
|
|
"""Update a bus schedule (Admin only)."""
|
|
db_schedule = session.get(BusSchedule, schedule_id)
|
|
if not db_schedule:
|
|
raise HTTPException(status_code=404, detail="Schedule not found")
|
|
|
|
for key, value in schedule_update.items():
|
|
if hasattr(db_schedule, key):
|
|
setattr(db_schedule, key, value)
|
|
|
|
session.add(db_schedule)
|
|
session.commit()
|
|
session.refresh(db_schedule)
|
|
return db_schedule
|
|
|
|
@router.delete("/{schedule_id}")
|
|
async def delete_schedule(
|
|
schedule_id: UUID,
|
|
session: Session = Depends(get_session),
|
|
_: bool = Depends(get_current_admin)
|
|
):
|
|
"""Delete a bus schedule (Admin only)."""
|
|
db_schedule = session.get(BusSchedule, schedule_id)
|
|
if not db_schedule:
|
|
raise HTTPException(status_code=404, detail="Schedule not found")
|
|
|
|
session.delete(db_schedule)
|
|
session.commit()
|
|
return {"ok": True}
|
|
|