Initial commit

api/app/category/models.py

@@ -0,0 +1,51 @@
+from uuid import UUID, uuid4
+from enum import Enum
+
+from sqlmodel import Field, SQLModel, select
+
+class CategoryBase(SQLModel):
+    name: str = Field(index=True)
+
+class CategoryRead(CategoryBase):
+    id: UUID | None = Field(default_factory=uuid4, primary_key=True)
+
+class Category(CategoryRead, table=True):
+
+    @classmethod
+    def create(cls, category, session):
+        category_db = cls.model_validate(category)
+        session.add(category_db)
+        session.commit()
+        session.refresh(category_db)
+
+        return category_db
+
+    @classmethod
+    def list(cls):
+        return select(Category)
+
+    @classmethod
+    def get(cls, session, category_id):
+        return session.get(Category, category_id)
+
+    @classmethod
+    def update(cls, session, category_db, category_data):
+        category_db.sqlmodel_update(category_data)
+        session.add(category_db)
+        session.commit()
+        session.refresh(category_db)
+        return category_db
+
+    @classmethod
+    def delete(cls, session, category):
+        session.delete(category)
+        session.commit()
+
+class CategoryWrite(CategoryBase):
+    pass
+
+class CategoryCreate(CategoryWrite):
+    pass
+
+class CategoryUpdate(CategoryWrite):
+    pass

api/app/category/routes.py

@@ -0,0 +1,47 @@
+from uuid import UUID
+
+
+from fastapi import APIRouter, HTTPException, Depends
+from fastapi_pagination import Page
+from fastapi_pagination.ext.sqlmodel import paginate
+
+from category.models import Category, CategoryCreate, CategoryRead, CategoryUpdate
+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, current_user=Depends(get_current_user)) -> Page[CategoryRead]:
+    return paginate(session, Category.list())
+
+@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}