Files
fastapi-best-architecture/backend/plugin/config/service/config_service.py
T

127 lines
3.9 KiB
Python

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
from typing import Sequence
from sqlalchemy import Select
from backend.common.exception import errors
from backend.core.conf import settings
from backend.database.db import async_db_session
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,
SaveBuiltInConfigParam,
UpdateConfigParam,
)
class ConfigService:
"""参数配置服务类"""
@staticmethod
async def get_built_in_config(type: str) -> Sequence[Config]:
"""
获取内置参数配置
:param type: 参数配置类型
:return:
"""
async with async_db_session() as db:
return await config_dao.get_by_type(db, type)
@staticmethod
async def save_built_in_config(objs: list[SaveBuiltInConfigParam], type: str) -> None:
"""
保存内置参数配置
:param objs: 参数配置参数列表
:param type: 参数配置类型
:return:
"""
async with async_db_session.begin() as db:
for obj in objs:
config = await config_dao.get_by_key_and_type(db, obj.key, type)
if config is None:
if await config_dao.get_by_key(db, obj.key):
raise errors.ForbiddenError(msg=f'参数配置 {obj.key} 已存在')
await config_dao.create_model(db, obj, flush=True, type=type)
else:
await config_dao.update_model(db, config.id, obj, type=type)
@staticmethod
async def get(pk: int) -> Config:
"""
获取参数配置详情
:param pk: 参数配置 ID
:return:
"""
async with async_db_session() as db:
config = await config_dao.get(db, pk)
if not config:
raise errors.NotFoundError(msg='参数配置不存在')
return config
@staticmethod
async def get_select(*, name: str | None, type: str | None) -> Select:
"""
获取参数配置列表查询条件
:param name: 参数配置名称
:param type: 参数配置类型
:return:
"""
return await config_dao.get_list(name=name, type=type)
@staticmethod
async def create(*, obj: CreateConfigParam) -> None:
"""
创建参数配置
:param obj: 参数配置创建参数
:return:
"""
async with async_db_session.begin() as db:
if obj.type in settings.CONFIG_BUILT_IN_TYPES:
raise errors.ForbiddenError(msg='非法类型参数')
config = await config_dao.get_by_key(db, obj.key)
if config:
raise errors.ForbiddenError(msg=f'参数配置 {obj.key} 已存在')
await config_dao.create(db, obj)
@staticmethod
async def update(*, pk: int, obj: UpdateConfigParam) -> int:
"""
更新参数配置
:param pk: 参数配置 ID
:param obj: 参数配置更新参数
:return:
"""
async with async_db_session.begin() as db:
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.ForbiddenError(msg=f'参数配置 {obj.key} 已存在')
count = await config_dao.update(db, pk, obj)
return count
@staticmethod
async def delete(*, pk: list[int]) -> int:
"""
删除参数配置
:param pk: 参数配置 ID 列表
:return:
"""
async with async_db_session.begin() as db:
count = await config_dao.delete(db, pk)
return count
config_service: ConfigService = ConfigService()