fastapi-best-architecture/backend/plugin/notice/service/notice_service.py

from collections.abc import Sequence

from sqlalchemy import Select

from backend.common.exception import errors
from backend.database.db import async_db_session
from backend.plugin.notice.crud.crud_notice import notice_dao
from backend.plugin.notice.model import Notice
from backend.plugin.notice.schema.notice import CreateNoticeParam, DeleteNoticeParam, UpdateNoticeParam


class NoticeService:
    """通知公告服务类"""

    @staticmethod
    async def get(*, pk: int) -> Notice:
        """
        获取通知公告

        :param pk: 通知公告 ID
        :return:
        """
        async with async_db_session() as db:
            notice = await notice_dao.get(db, pk)
            if not notice:
                raise errors.NotFoundError(msg='通知公告不存在')
            return notice

    @staticmethod
    async def get_select(title: str | None, type: int | None, status: int | None) -> Select:
        """获取通知公告查询对象"""
        return await notice_dao.get_list(title, type, status)

    @staticmethod
    async def get_all() -> Sequence[Notice]:
        """获取所有通知公告"""
        async with async_db_session() as db:
            notices = await notice_dao.get_all(db)
            return notices

    @staticmethod
    async def create(*, obj: CreateNoticeParam) -> None:
        """
        创建通知公告

        :param obj: 创建通知公告参数
        :return:
        """
        async with async_db_session.begin() as db:
            await notice_dao.create(db, obj)

    @staticmethod
    async def update(*, pk: int, obj: UpdateNoticeParam) -> int:
        """
        更新通知公告

        :param pk: 通知公告 ID
        :param obj: 更新通知公告参数
        :return:
        """
        async with async_db_session.begin() as db:
            notice = await notice_dao.get(db, pk)
            if not notice:
                raise errors.NotFoundError(msg='通知公告不存在')
            count = await notice_dao.update(db, pk, obj)
            return count

    @staticmethod
    async def delete(*, obj: DeleteNoticeParam) -> int:
        """
        批量删除通知公告

        :param obj: 通知公告 ID 列表
        :return:
        """
        async with async_db_session.begin() as db:
            count = await notice_dao.delete(db, obj.pks)
            return count


notice_service: NoticeService = NoticeService()