mirror of
https://github.com/fastapi-practices/fastapi-best-architecture.git
synced 2026-09-21 21:15:13 +00:00
Wu Clan
15a9527866
* Refactor the service layer db session call method * Fix lint * Update paging interfaces * Fix some errors
133 lines
3.8 KiB
Python
133 lines
3.8 KiB
Python
from collections.abc import Sequence
|
|
from typing import Any
|
|
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from backend.common.exception import errors
|
|
from backend.common.pagination import paging_data
|
|
from backend.plugin.config.crud.crud_config import config_dao
|
|
from backend.plugin.config.model import Config
|
|
from backend.plugin.config.schema.config import (
|
|
CreateConfigParam,
|
|
UpdateConfigParam,
|
|
UpdateConfigsParam,
|
|
)
|
|
|
|
|
|
class ConfigService:
|
|
"""参数配置服务类"""
|
|
|
|
@staticmethod
|
|
async def get(*, db: AsyncSession, pk: int) -> Config:
|
|
"""
|
|
获取参数配置详情
|
|
|
|
:param db: 数据库会话
|
|
:param pk: 参数配置 ID
|
|
:return:
|
|
"""
|
|
|
|
config = await config_dao.get(db, pk)
|
|
if not config:
|
|
raise errors.NotFoundError(msg='参数配置不存在')
|
|
return config
|
|
|
|
@staticmethod
|
|
async def get_all(*, db: AsyncSession, type: str | None) -> Sequence[Config | None]:
|
|
"""
|
|
获取所有参数配置
|
|
|
|
:param db: 数据库会话
|
|
:param type: 参数配置类型
|
|
:return:
|
|
"""
|
|
|
|
return await config_dao.get_all(db, type)
|
|
|
|
@staticmethod
|
|
async def get_list(*, db: AsyncSession, name: str | None, type: str | None) -> dict[str, Any]:
|
|
"""
|
|
获取参数配置列表
|
|
|
|
:param db: 数据库会话
|
|
:param name: 参数配置名称
|
|
:param type: 参数配置类型
|
|
:return:
|
|
"""
|
|
config_select = await config_dao.get_select(name=name, type=type)
|
|
return await paging_data(db, config_select)
|
|
|
|
@staticmethod
|
|
async def create(*, db: AsyncSession, obj: CreateConfigParam) -> None:
|
|
"""
|
|
创建参数配置
|
|
|
|
:param db: 数据库会话
|
|
:param obj: 参数配置创建参数
|
|
:return:
|
|
"""
|
|
|
|
config = await config_dao.get_by_key(db, obj.key)
|
|
if config:
|
|
raise errors.ConflictError(msg=f'参数配置 {obj.key} 已存在')
|
|
await config_dao.create(db, obj)
|
|
|
|
@staticmethod
|
|
async def update(*, db: AsyncSession, pk: int, obj: UpdateConfigParam) -> int:
|
|
"""
|
|
更新参数配置
|
|
|
|
:param db: 数据库会话
|
|
:param pk: 参数配置 ID
|
|
:param obj: 参数配置更新参数
|
|
:return:
|
|
"""
|
|
|
|
config = await config_dao.get(db, pk)
|
|
if not config:
|
|
raise errors.NotFoundError(msg='参数配置不存在')
|
|
if config.key != obj.key:
|
|
config = await config_dao.get_by_key(db, obj.key)
|
|
if config:
|
|
raise errors.ConflictError(msg=f'参数配置 {obj.key} 已存在')
|
|
count = await config_dao.update(db, pk, obj)
|
|
return count
|
|
|
|
@staticmethod
|
|
async def bulk_update(*, db: AsyncSession, objs: list[UpdateConfigsParam]) -> int:
|
|
"""
|
|
批量更新参数配置
|
|
|
|
:param db: 数据库会话
|
|
:param objs: 参数配置批量更新参数
|
|
:return:
|
|
"""
|
|
|
|
for _batch in range(0, len(objs), 1000):
|
|
for obj in objs:
|
|
config = await config_dao.get(db, obj.id)
|
|
if not config:
|
|
raise errors.NotFoundError(msg='参数配置不存在')
|
|
if config.key != obj.key:
|
|
config = await config_dao.get_by_key(db, obj.key)
|
|
if config:
|
|
raise errors.ConflictError(msg=f'参数配置 {obj.key} 已存在')
|
|
count = await config_dao.bulk_update(db, objs)
|
|
return count
|
|
|
|
@staticmethod
|
|
async def delete(*, db: AsyncSession, pks: list[int]) -> int:
|
|
"""
|
|
批量删除参数配置
|
|
|
|
:param db: 数据库会话
|
|
:param pks: 参数配置 ID 列表
|
|
:return:
|
|
"""
|
|
|
|
count = await config_dao.delete(db, pks)
|
|
return count
|
|
|
|
|
|
config_service: ConfigService = ConfigService()
|