from module_admin.entity.vo.dscatalog_vo import * from module_admin.dao.dscatalog_dao import * class DscatalogService: """ 标准管理模块服务层 """ @classmethod def get_dscatalog_tree_services(cls, result_db: Session, page_object: DscatalogModel, data_scope_sql: str): """ 获取标准树信息service :param result_db: orm对象 :param page_object: 查询参数对象 :param data_scope_sql: 数据权限对应的查询sql语句 :return: 标准树信息对象 """ dscatalog_list_result = DscatalogDao.get_dscatalog_list_for_tree(result_db, page_object, data_scope_sql) dscatalog_tree_result = cls.list_to_tree(dscatalog_list_result) return dscatalog_tree_result @classmethod def get_dscatalog_tree_for_edit_option_services(cls, result_db: Session, page_object: DscatalogModel, data_scope_sql: str): """ 获取标准编辑标准树信息service :param result_db: orm对象 :param page_object: 查询参数对象 :param data_scope_sql: 数据权限对应的查询sql语句 :return: 标准树信息对象 """ dscatalog_list_result = DscatalogDao.get_dscatalog_info_for_edit_option(result_db, page_object, data_scope_sql) dscatalog_tree_result = cls.list_to_tree(dscatalog_list_result) return dscatalog_tree_result @classmethod def get_dscatalog_list_services(cls, result_db: Session, page_object: DscatalogModel, data_scope_sql: str): """ 获取标准列表信息service :param result_db: orm对象 :param page_object: 分页查询参数对象 :param data_scope_sql: 数据权限对应的查询sql语句 :return: 标准列表信息对象 """ dscatalog_list_result = DscatalogDao.get_dscatalog_list(result_db, page_object, data_scope_sql) return dscatalog_list_result @classmethod def add_dscatalog_services(cls, result_db: Session, page_object: DscatalogModel): """ 新增标准信息service :param result_db: orm对象 :param page_object: 新增标准对象 :return: 新增标准校验结果 """ parent_info = DscatalogDao.get_dscatalog_by_id(result_db, page_object.dscatalog_parent_id) if parent_info: page_object.dscatalog_ancestors = f'{parent_info.dscatalog_ancestors},{page_object.dscatalog_parent_id}' else: page_object.dscatalog_ancestors = '0' dscatalog = DscatalogDao.get_dscatalog_detail_by_info(result_db, DscatalogModel(**dict(dscatalog_parent_id=page_object.dscatalog_parent_id, dscatalog_name=page_object.dscatalog_name))) if dscatalog: result = dict(is_success=False, message='同一标准下不允许存在同名的标准') else: try: DscatalogDao.add_dscatalog_dao(result_db, page_object) result_db.commit() result = dict(is_success=True, message='新增成功') except Exception as e: result_db.rollback() result = dict(is_success=False, message=str(e)) return CrudDscatalogResponse(**result) @classmethod def edit_dscatalog_services(cls, result_db: Session, page_object: DscatalogModel): """ 编辑标准信息service :param result_db: orm对象 :param page_object: 编辑标准对象 :return: 编辑标准校验结果 """ parent_info = DscatalogDao.get_dscatalog_by_id(result_db, page_object.dscatalog_parent_id) if parent_info: page_object.dscatalog_ancestors = f'{parent_info.dscatalog_ancestors},{page_object.dscatalog_parent_id}' else: page_object.dscatalog_ancestors = '0' edit_dscatalog = page_object.dict(exclude_unset=True) dscatalog_info = cls.detail_dscatalog_services(result_db, edit_dscatalog.get('dscatalog_id')) if dscatalog_info: if dscatalog_info.dscatalog_id != page_object.dscatalog_id: dscatalog = DscatalogDao.get_dscatalog_detail_by_info(result_db, DscatalogModel( **dict(dscatalog_parent_id=page_object.dscatalog_parent_id, dscatalog_name=page_object.dscatalog_name))) if dscatalog: result = dict(is_success=False, message='同一标准下不允许存在同名的标准') return CrudDscatalogResponse(**result) try: DscatalogDao.edit_dscatalog_dao(result_db, edit_dscatalog) cls.update_children_info(result_db, DscatalogModel(dscatalog_id=page_object.dscatalog_id, dscatalog_ancestors=page_object.dscatalog_ancestors, update_by=page_object.update_by, update_time=page_object.update_time ) ) result_db.commit() result = dict(is_success=True, message='更新成功') except Exception as e: result_db.rollback() result = dict(is_success=False, message=str(e)) else: result = dict(is_success=False, message='标准不存在') return CrudDscatalogResponse(**result) @classmethod def delete_dscatalog_services(cls, result_db: Session, page_object: DeleteDscatalogModel): """ 删除标准信息service :param result_db: orm对象 :param page_object: 删除标准对象 :return: 删除标准校验结果 """ if page_object.dscatalog_ids.split(','): dscatalog_id_list = page_object.dscatalog_ids.split(',') dscatalog_ancestors = DscatalogDao.get_dscatalog_all_ancestors(result_db) try: for dscatalog_id in dscatalog_id_list: for dscatalog_ancestor in dscatalog_ancestors: if dscatalog_id in dscatalog_ancestor[0]: result = dict(is_success=False, message='该标准下有子标准，不允许删除') return CrudDscatalogResponse(**result) dscatalog_id_dict = dict(dscatalog_id=dscatalog_id) DscatalogDao.delete_dscatalog_dao(result_db, DscatalogModel(**dscatalog_id_dict)) result_db.commit() result = dict(is_success=True, message='删除成功') except Exception as e: result_db.rollback() result = dict(is_success=False, message=str(e)) else: result = dict(is_success=False, message='传入标准id为空') return CrudDscatalogResponse(**result) @classmethod def detail_dscatalog_services(cls, result_db: Session, dscatalog_id: int): """ 获取标准详细信息service :param result_db: orm对象 :param dscatalog_id: 标准id :return: 标准id对应的信息 """ dscatalog = DscatalogDao.get_dscatalog_detail_by_id(result_db, dscatalog_id=dscatalog_id) return dscatalog @classmethod def list_to_tree(cls, permission_list: list) -> list: """ 工具方法：根据标准列表信息生成树形嵌套数据 :param permission_list: 标准列表信息 :return: 标准树形嵌套数据 """ permission_list = [dict(title=item.dscatalog_name, key=str(item.dscatalog_id), value=str(item.dscatalog_id), dscatalog_parent_id=str(item.dscatalog_parent_id)) for item in permission_list] # 转成dscatalog_id为key的字典 mapping: dict = dict(zip([i['key'] for i in permission_list], permission_list)) # 树容器 container: list = [] for d in permission_list: # 如果找不到父级项，则是根节点 parent: dict = mapping.get(d['dscatalog_parent_id']) if parent is None: container.append(d) else: children: list = parent.get('children') if not children: children = [] children.append(d) parent.update({'children': children}) return container @classmethod def get_dscatalog_tree(cls, pid: int, permission_list: DscatalogTree): """ 工具方法：根据标准信息生成树形嵌套数据 :param pid: 标准id :param permission_list: 标准列表信息 :return: 标准树形嵌套数据 """ dscatalog_list = [] for permission in permission_list.dscatalog_tree: if permission.dscatalog_parent_id == pid: children = cls.get_dscatalog_tree(permission.dscatalog_id, permission_list) dscatalog_list_data = {} if children: dscatalog_list_data['title'] = permission.dscatalog_name dscatalog_list_data['key'] = str(permission.dscatalog_id) dscatalog_list_data['value'] = str(permission.dscatalog_id) dscatalog_list_data['children'] = children else: dscatalog_list_data['title'] = permission.dscatalog_name dscatalog_list_data['key'] = str(permission.dscatalog_id) dscatalog_list_data['value'] = str(permission.dscatalog_id) dscatalog_list.append(dscatalog_list_data) return dscatalog_list @classmethod def update_children_info(cls, result_db, page_object): """ 工具方法：递归更新子标准信息 :param result_db: orm对象 :param page_object: 编辑标准对象 :return: """ children_info = DscatalogDao.get_children_dscatalog(result_db, page_object.dscatalog_id) if children_info: for child in children_info: child.dscatalog_ancestors = f'{page_object.dscatalog_ancestors},{page_object.dscatalog_id}' DscatalogDao.edit_dscatalog_dao(result_db, dict(dscatalog_id=child.dscatalog_id, dscatalog_ancestors=child.dscatalog_ancestors, update_by=page_object.update_by, update_time=page_object.update_time ) ) cls.update_children_info(result_db, DscatalogModel(dscatalog_id=child.dscatalog_id, dscatalog_ancestors=child.dscatalog_ancestors, update_by=page_object.update_by, update_time=page_object.update_time ))