57 lines
1.9 KiB
Python
57 lines
1.9 KiB
Python
from fastapi import APIRouter, Depends, HTTPException, status
|
|
from sqlalchemy.orm import Session
|
|
|
|
from ..db import get_db
|
|
from ..models import Cemetery
|
|
from ..schemas import CemeteryCreate, CemeteryOut, CemeteryUpdate
|
|
from ..dependencies import require_admin
|
|
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.post("/", response_model=CemeteryOut, dependencies=[Depends(require_admin)])
|
|
def create_cemetery(payload: CemeteryCreate, db: Session = Depends(get_db)):
|
|
cemetery = Cemetery(**payload.dict())
|
|
db.add(cemetery)
|
|
db.commit()
|
|
db.refresh(cemetery)
|
|
return cemetery
|
|
|
|
|
|
@router.get("/", response_model=list[CemeteryOut])
|
|
def list_cemeteries(db: Session = Depends(get_db)):
|
|
return db.query(Cemetery).order_by(Cemetery.id.desc()).all()
|
|
|
|
|
|
@router.get("/{cemetery_id}", response_model=CemeteryOut)
|
|
def get_cemetery(cemetery_id: int, db: Session = Depends(get_db)):
|
|
cemetery = db.query(Cemetery).get(cemetery_id)
|
|
if not cemetery:
|
|
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Not found")
|
|
return cemetery
|
|
|
|
|
|
@router.patch("/{cemetery_id}", response_model=CemeteryOut, dependencies=[Depends(require_admin)])
|
|
def update_cemetery(cemetery_id: int, payload: CemeteryUpdate, db: Session = Depends(get_db)):
|
|
cemetery = db.query(Cemetery).get(cemetery_id)
|
|
if not cemetery:
|
|
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Not found")
|
|
for k, v in payload.dict(exclude_unset=True).items():
|
|
setattr(cemetery, k, v)
|
|
db.commit()
|
|
db.refresh(cemetery)
|
|
return cemetery
|
|
|
|
|
|
@router.delete("/{cemetery_id}", status_code=204, dependencies=[Depends(require_admin)])
|
|
def delete_cemetery(cemetery_id: int, db: Session = Depends(get_db)):
|
|
cemetery = db.query(Cemetery).get(cemetery_id)
|
|
if not cemetery:
|
|
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Not found")
|
|
db.delete(cemetery)
|
|
db.commit()
|
|
return None
|
|
|
|
|