from fastapi import Request from sqlalchemy.ext.asyncio import AsyncSession from typing import List from config.constant import CommonConstant from config.enums import RedisInitKeyConfig from exceptions.exception import ServiceException from module_admin.dao.config_dao import ConfigDao from module_admin.entity.vo.common_vo import CrudResponseModel from module_admin.entity.vo.config_vo import ConfigModel, ConfigPageQueryModel, DeleteConfigModel from utils.common_util import CamelCaseUtil, export_list2excel class ConfigService: """ 参数配置管理模块服务层 """ @classmethod async def get_config_list_services( cls, query_db: AsyncSession, query_object: ConfigPageQueryModel, is_page: bool = False ): """ 获取参数配置列表信息service :param query_db: orm对象 :param query_object: 查询参数对象 :param is_page: 是否开启分页 :return: 参数配置列表信息对象 """ config_list_result = await ConfigDao.get_config_list(query_db, query_object, is_page) return config_list_result @classmethod async def init_cache_sys_config_services(cls, query_db: AsyncSession, redis): """ 应用初始化：获取所有参数配置对应的键值对信息并缓存service :param query_db: orm对象 :param redis: redis对象 :return: """ # 获取以sys_config:开头的键列表 keys = await redis.keys(f'{RedisInitKeyConfig.SYS_CONFIG.key}:*') # 删除匹配的键 if keys: await redis.delete(*keys) config_all = await ConfigDao.get_config_list(query_db, ConfigPageQueryModel(**dict()), is_page=False) for config_obj in config_all: await redis.set( f"{RedisInitKeyConfig.SYS_CONFIG.key}:{config_obj.get('configKey')}", config_obj.get('configValue'), ) @classmethod async def query_config_list_from_cache_services(cls, redis, config_key: str): """ 从缓存获取参数键名对应值service :param redis: redis对象 :param config_key: 参数键名 :return: 参数键名对应值 """ result = await redis.get(f'{RedisInitKeyConfig.SYS_CONFIG.key}:{config_key}') return result @classmethod async def check_config_key_unique_services(cls, query_db: AsyncSession, page_object: ConfigModel): """ 校验参数键名是否唯一service :param query_db: orm对象 :param page_object: 参数配置对象 :return: 校验结果 """ config_id = -1 if page_object.config_id is None else page_object.config_id config = await ConfigDao.get_config_detail_by_info(query_db, ConfigModel(configKey=page_object.config_key)) if config and config.config_id != config_id: return CommonConstant.NOT_UNIQUE return CommonConstant.UNIQUE @classmethod async def add_config_services(cls, request: Request, query_db: AsyncSession, page_object: ConfigModel): """ 新增参数配置信息service :param request: Request对象 :param query_db: orm对象 :param page_object: 新增参数配置对象 :return: 新增参数配置校验结果 """ if not await cls.check_config_key_unique_services(query_db, page_object): raise ServiceException(message=f'新增参数{page_object.config_name}失败，参数键名已存在') else: try: await ConfigDao.add_config_dao(query_db, page_object) await query_db.commit() await request.app.state.redis.set( f'{RedisInitKeyConfig.SYS_CONFIG.key}:{page_object.config_key}', page_object.config_value ) return CrudResponseModel(is_success=True, message='新增成功') except Exception as e: await query_db.rollback() raise e @classmethod async def edit_config_services(cls, request: Request, query_db: AsyncSession, page_object: ConfigModel): """ 编辑参数配置信息service :param request: Request对象 :param query_db: orm对象 :param page_object: 编辑参数配置对象 :return: 编辑参数配置校验结果 """ edit_config = page_object.model_dump(exclude_unset=True) config_info = await cls.config_detail_services(query_db, page_object.config_id) if config_info.config_id: if not await cls.check_config_key_unique_services(query_db, page_object): raise ServiceException(message=f'修改参数{page_object.config_name}失败，参数键名已存在') else: try: await ConfigDao.edit_config_dao(query_db, edit_config) await query_db.commit() if config_info.config_key != page_object.config_key: await request.app.state.redis.delete( f'{RedisInitKeyConfig.SYS_CONFIG.key}:{config_info.config_key}' ) await request.app.state.redis.set( f'{RedisInitKeyConfig.SYS_CONFIG.key}:{page_object.config_key}', page_object.config_value ) return CrudResponseModel(is_success=True, message='更新成功') except Exception as e: await query_db.rollback() raise e else: raise ServiceException(message='参数配置不存在') @classmethod async def delete_config_services(cls, request: Request, query_db: AsyncSession, page_object: DeleteConfigModel): """ 删除参数配置信息service :param request: Request对象 :param query_db: orm对象 :param page_object: 删除参数配置对象 :return: 删除参数配置校验结果 """ if page_object.config_ids: config_id_list = page_object.config_ids.split(',') try: delete_config_key_list = [] for config_id in config_id_list: config_info = await cls.config_detail_services(query_db, int(config_id)) if config_info.config_type == CommonConstant.YES: raise ServiceException(message=f'内置参数{config_info.config_key}不能删除') else: await ConfigDao.delete_config_dao(query_db, ConfigModel(configId=int(config_id))) delete_config_key_list.append(f'{RedisInitKeyConfig.SYS_CONFIG.key}:{config_info.config_key}') await query_db.commit() if delete_config_key_list: await request.app.state.redis.delete(*delete_config_key_list) return CrudResponseModel(is_success=True, message='删除成功') except Exception as e: await query_db.rollback() raise e else: raise ServiceException(message='传入参数配置id为空') @classmethod async def config_detail_services(cls, query_db: AsyncSession, config_id: int): """ 获取参数配置详细信息service :param query_db: orm对象 :param config_id: 参数配置id :return: 参数配置id对应的信息 """ config = await ConfigDao.get_config_detail_by_id(query_db, config_id=config_id) if config: result = ConfigModel(**CamelCaseUtil.transform_result(config)) else: result = ConfigModel(**dict()) return result @staticmethod async def export_config_list_services(config_list: List): """ 导出参数配置信息service :param config_list: 参数配置信息列表 :return: 参数配置信息对应excel的二进制数据 """ # 创建一个映射字典，将英文键映射到中文键 mapping_dict = { 'configId': '参数主键', 'configName': '参数名称', 'configKey': '参数键名', 'configValue': '参数键值', 'configType': '系统内置', 'createBy': '创建者', 'createTime': '创建时间', 'updateBy': '更新者', 'updateTime': '更新时间', 'remark': '备注', } data = config_list for item in data: if item.get('configType') == 'Y': item['configType'] = '是' else: item['configType'] = '否' new_data = [ {mapping_dict.get(key): value for key, value in item.items() if mapping_dict.get(key)} for item in data ] binary_data = export_list2excel(new_data) return binary_data @classmethod async def refresh_sys_config_services(cls, request: Request, query_db: AsyncSession): """ 刷新字典缓存信息service :param request: Request对象 :param query_db: orm对象 :return: 刷新字典缓存校验结果 """ await cls.init_cache_sys_config_services(query_db, request.app.state.redis) return CrudResponseModel(is_success=True, message='刷新成功')