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}