Add app/services/category_service.py
This commit is contained in:
39
app/services/category_service.py
Normal file
39
app/services/category_service.py
Normal file
@@ -0,0 +1,39 @@
|
|||||||
|
from sqlalchemy.orm import Session
|
||||||
|
from app.models.category import Category
|
||||||
|
from app.schemas.category import CategoryCreate, CategoryUpdate
|
||||||
|
|
||||||
|
|
||||||
|
def create_category(db: Session, data: CategoryCreate) -> Category:
|
||||||
|
cat = Category(name=data.name, description=data.description)
|
||||||
|
db.add(cat)
|
||||||
|
db.commit()
|
||||||
|
db.refresh(cat)
|
||||||
|
return cat
|
||||||
|
|
||||||
|
|
||||||
|
def get_categories(db: Session):
|
||||||
|
return db.query(Category).all()
|
||||||
|
|
||||||
|
|
||||||
|
def get_category(db: Session, category_id: str):
|
||||||
|
return db.query(Category).filter(Category.id == category_id).first()
|
||||||
|
|
||||||
|
|
||||||
|
def update_category(db: Session, category_id: str, data: CategoryUpdate):
|
||||||
|
cat = get_category(db, category_id)
|
||||||
|
if not cat:
|
||||||
|
return None
|
||||||
|
for field, value in data.model_dump(exclude_unset=True).items():
|
||||||
|
setattr(cat, field, value)
|
||||||
|
db.commit()
|
||||||
|
db.refresh(cat)
|
||||||
|
return cat
|
||||||
|
|
||||||
|
|
||||||
|
def delete_category(db: Session, category_id: str):
|
||||||
|
cat = get_category(db, category_id)
|
||||||
|
if not cat:
|
||||||
|
return False
|
||||||
|
db.delete(cat)
|
||||||
|
db.commit()
|
||||||
|
return True
|
||||||
Reference in New Issue
Block a user