fastapi-best-architecture/backend/plugin/casbin/service/api_service.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
from typing import Sequence

from fastapi import Request
from sqlalchemy import Select

from backend.common.exception import errors
from backend.database.db import async_db_session
from backend.plugin.casbin.crud.crud_api import api_dao
from backend.plugin.casbin.model import Api
from backend.plugin.casbin.schema.api import CreateApiParam, UpdateApiParam


class ApiService:
    """API 服务类"""

    @staticmethod
    async def get(*, pk: int) -> Api:
        """
        获取 API

        :param pk: API ID
        :return:
        """
        async with async_db_session() as db:
            api = await api_dao.get(db, pk)
            if not api:
                raise errors.NotFoundError(msg='接口不存在')
            return api

    @staticmethod
    async def get_select(*, request: Request, name: str = None, method: str = None, path: str = None) -> Select:
        """
        获取 API 查询对象

        :param request: 请求对象
        :param name: API 名称
        :param method: 请求方法
        :param path: API 路径
        :return:
        """
        return await api_dao.get_list(request=request, name=name, method=method, path=path)

    @staticmethod
    async def get_all() -> Sequence[Api]:
        """获取所有 API"""
        async with async_db_session() as db:
            apis = await api_dao.get_all(db)
            return apis

    @staticmethod
    async def create(*, obj: CreateApiParam) -> None:
        """
        创建 API

        :param obj: 创建 API 参数
        :return:
        """
        async with async_db_session.begin() as db:
            api = await api_dao.get_by_name(db, obj.name)
            if api:
                raise errors.ForbiddenError(msg='接口已存在')
            await api_dao.create(db, obj)

    @staticmethod
    async def update(*, pk: int, obj: UpdateApiParam) -> int:
        """
        更新 API

        :param pk: API ID
        :param obj: 更新 API 参数
        :return:
        """
        async with async_db_session.begin() as db:
            api = await api_dao.get(db, pk)
            if not api:
                raise errors.NotFoundError(msg='接口不存在')
            count = await api_dao.update(db, pk, obj)
            return count

    @staticmethod
    async def delete(*, pk: list[int]) -> int:
        """
        删除 API

        :param pk: API ID 列表
        :return:
        """
        async with async_db_session.begin() as db:
            count = await api_dao.delete(db, pk)
            return count


api_service: ApiService = ApiService()