Cheng Zhou
86 lines
2.6 KiB
Python
86 lines
2.6 KiB
Python
import uuid
|
|
from typing import Sequence
|
|
|
|
from sqlalchemy import select, update as sa_update
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.models.faq_category import FaqCategory
|
|
from app.schemas.faq import CategoryCreate, CategoryUpdate
|
|
|
|
|
|
async def create_category(db: AsyncSession, category_in: CategoryCreate) -> FaqCategory:
|
|
category = FaqCategory(
|
|
study_id=category_in.study_id,
|
|
name=category_in.name,
|
|
description=category_in.description,
|
|
sort_order=category_in.sort_order,
|
|
icon=category_in.icon,
|
|
)
|
|
db.add(category)
|
|
await db.commit()
|
|
await db.refresh(category)
|
|
return category
|
|
|
|
|
|
async def list_categories(
|
|
db: AsyncSession,
|
|
study_id: uuid.UUID | None,
|
|
include_global: bool = True,
|
|
) -> Sequence[FaqCategory]:
|
|
stmt = select(FaqCategory)
|
|
if include_global:
|
|
stmt = stmt.where((FaqCategory.study_id == study_id) | (FaqCategory.study_id.is_(None)))
|
|
else:
|
|
stmt = stmt.where(FaqCategory.study_id == study_id)
|
|
stmt = stmt.order_by(FaqCategory.sort_order, FaqCategory.name)
|
|
result = await db.execute(stmt)
|
|
return result.scalars().all()
|
|
|
|
|
|
async def get_category(db: AsyncSession, category_id: uuid.UUID) -> FaqCategory | None:
|
|
result = await db.execute(select(FaqCategory).where(FaqCategory.id == category_id))
|
|
return result.scalar_one_or_none()
|
|
|
|
|
|
async def get_category_by_name(
|
|
db: AsyncSession,
|
|
study_id: uuid.UUID | None,
|
|
name: str,
|
|
) -> FaqCategory | None:
|
|
stmt = select(FaqCategory).where(FaqCategory.name == name)
|
|
if study_id is None:
|
|
stmt = stmt.where(FaqCategory.study_id.is_(None))
|
|
else:
|
|
stmt = stmt.where(FaqCategory.study_id == study_id)
|
|
result = await db.execute(stmt)
|
|
return result.scalar_one_or_none()
|
|
|
|
|
|
async def get_category_by_icon(
|
|
db: AsyncSession,
|
|
study_id: uuid.UUID | None,
|
|
icon: str,
|
|
) -> FaqCategory | None:
|
|
if not icon:
|
|
return None
|
|
stmt = select(FaqCategory).where(FaqCategory.icon == icon)
|
|
if study_id is None:
|
|
stmt = stmt.where(FaqCategory.study_id.is_(None))
|
|
else:
|
|
stmt = stmt.where(FaqCategory.study_id == study_id)
|
|
result = await db.execute(stmt)
|
|
return result.scalar_one_or_none()
|
|
|
|
|
|
async def update_category(db: AsyncSession, category: FaqCategory, category_in: CategoryUpdate) -> FaqCategory:
|
|
update_data = category_in.model_dump(exclude_unset=True)
|
|
if update_data:
|
|
await db.execute(
|
|
sa_update(FaqCategory)
|
|
.where(FaqCategory.id == category.id)
|
|
.values(**update_data)
|
|
)
|
|
await db.commit()
|
|
await db.refresh(category)
|
|
return category
|