51 lines
2.0 KiB
Python
51 lines
2.0 KiB
Python
from uuid import UUID
|
|
|
|
from fastapi import APIRouter, HTTPException, Depends
|
|
from fastapi_filter import FilterDepends
|
|
from fastapi_pagination import Page
|
|
from fastapi_pagination.ext.sqlmodel import paginate
|
|
from pydantic import BaseModel
|
|
|
|
from category.models import Category, CategoryCreate, CategoryRead, CategoryUpdate, CategoryFilters
|
|
from db import SessionDep
|
|
from user.manager import get_current_user
|
|
|
|
router = APIRouter()
|
|
|
|
@router.post("")
|
|
def create_category(category: CategoryCreate, session: SessionDep, current_user=Depends(get_current_user)) -> CategoryRead:
|
|
Category.create(category, session)
|
|
return category
|
|
|
|
@router.get("")
|
|
def read_categories(session: SessionDep,
|
|
filters: CategoryFilters = FilterDepends(CategoryFilters),
|
|
current_user=Depends(get_current_user)) -> Page[CategoryRead]:
|
|
return paginate(session, Category.list(filters))
|
|
|
|
@router.get("/{category_id}")
|
|
def read_category(category_id: UUID, session: SessionDep, current_user=Depends(get_current_user)) -> CategoryRead:
|
|
category = Category.get(session, category_id)
|
|
if not category:
|
|
raise HTTPException(status_code=404, detail="Category not found")
|
|
return category
|
|
|
|
@router.put("/{category_id}")
|
|
def update_category(category_id: UUID, category: CategoryUpdate, session: SessionDep, current_user=Depends(get_current_user)) -> CategoryRead:
|
|
db_category = Category.get(session, category_id)
|
|
if not db_category:
|
|
raise HTTPException(status_code=404, detail="Category not found")
|
|
|
|
category_data = category.model_dump(exclude_unset=True)
|
|
category = Category.update(session, db_category, category_data)
|
|
return category
|
|
|
|
@router.delete("/{category_id}")
|
|
def delete_category(category_id: UUID, session: SessionDep, current_user=Depends(get_current_user)):
|
|
category = Category.get(session, category_id)
|
|
if not category:
|
|
raise HTTPException(status_code=404, detail="Category not found")
|
|
|
|
Category.delete(session, category)
|
|
return {"ok": True}
|