from module_admin.entity.vo.dasset_vo import * from module_admin.dao.dasset_dao import * class DassetService: """ 部门管理模块服务层 """ @classmethod def get_dasset_tree_services(cls, result_db: Session, page_object: DassetModel, data_scope_sql: str): """ 获取部门树信息service :param result_db: orm对象 :param page_object: 查询参数对象 :param data_scope_sql: 数据权限对应的查询sql语句 :return: 部门树信息对象 """ dasset_list_result = DassetDao.get_dasset_list_for_tree(result_db, page_object, data_scope_sql) dasset_tree_result = cls.list_to_tree(dasset_list_result) return dasset_tree_result @classmethod def get_dasset_tree_for_edit_option_services(cls, result_db: Session, page_object: DassetModel, data_scope_sql: str): """ 获取部门编辑部门树信息service :param result_db: orm对象 :param page_object: 查询参数对象 :param data_scope_sql: 数据权限对应的查询sql语句 :return: 部门树信息对象 """ dasset_list_result = DassetDao.get_dasset_info_for_edit_option(result_db, page_object, data_scope_sql) dasset_tree_result = cls.list_to_tree(dasset_list_result) return dasset_tree_result @classmethod def get_dasset_list_services(cls, result_db: Session, page_object: DassetModel, data_scope_sql: str): """ 获取部门列表信息service :param result_db: orm对象 :param page_object: 分页查询参数对象 :param data_scope_sql: 数据权限对应的查询sql语句 :return: 部门列表信息对象 """ dasset_list_result = DassetDao.get_dasset_list(result_db, page_object, data_scope_sql) return dasset_list_result @classmethod def add_dasset_services(cls, result_db: Session, page_object: DassetModel): """ 新增部门信息service :param result_db: orm对象 :param page_object: 新增部门对象 :return: 新增部门校验结果 """ parent_info = DassetDao.get_dasset_by_id(result_db, page_object.dasset_parent_id) if parent_info: page_object.dasset_ancestors = f'{parent_info.dasset_ancestors},{page_object.dasset_parent_id}' else: page_object.dasset_ancestors = '0' dasset = DassetDao.get_dasset_detail_by_info(result_db, DassetModel(**dict(dasset_parent_id=page_object.dasset_parent_id, dasset_name=page_object.dasset_name))) if dasset: result = dict(is_success=False, message='同一部门下不允许存在同名的部门') else: try: DassetDao.add_dasset_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 CrudDassetResponse(**result) @classmethod def edit_dasset_services(cls, result_db: Session, page_object: DassetModel): """ 编辑资产信息service :param result_db: orm对象 :param page_object: 编辑资产对象 :return: 编辑资产校验结果 """ parent_info = DassetDao.get_dasset_by_id(result_db, page_object.dasset_parent_id) if parent_info: page_object.dasset_ancestors = f'{parent_info.dasset_ancestors},{page_object.dasset_parent_id}' else: page_object.dasset_ancestors = '0' edit_dasset = page_object.dict(exclude_unset=True) dasset_info = cls.detail_dasset_services(result_db, edit_dasset.get('dasset_id')) if dasset_info: if dasset_info.dasset_id != page_object.dasset_id: dasset = DassetDao.get_dasset_detail_by_info(result_db, DassetModel( **dict(dasset_parent_id=page_object.dasset_parent_id, dasset_id=page_object.dasset_id))) if dasset: result = dict(is_success=False, message='存在相同名称的资产') return CrudDassetResponse(**result) try: DassetDao.edit_dasset_dao(result_db, edit_dasset) cls.update_children_info(result_db, DassetModel(dasset_id=page_object.dasset_id, dasset_ancestors=page_object.dasset_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 CrudDassetResponse(**result) @classmethod def delete_dasset_services(cls, result_db: Session, page_object: DeleteDassetModel): """ 删除部门信息service :param result_db: orm对象 :param page_object: 删除部门对象 :return: 删除部门校验结果 """ if page_object.dasset_ids.split(','): dasset_id_list = page_object.dasset_ids.split(',') dasset_ancestors = DassetDao.get_dasset_all_ancestors(result_db) try: for dasset_id in dasset_id_list: for dasset_ancestor in dasset_ancestors: if dasset_id in dasset_ancestor[0]: result = dict(is_success=False, message='该部门下有子部门，不允许删除') return CrudDassetResponse(**result) dasset_id_dict = dict(dasset_id=dasset_id) DassetDao.delete_dasset_dao(result_db, DassetModel(**dasset_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 CrudDassetResponse(**result) @classmethod def detail_dasset_services(cls, result_db: Session, dasset_id: int): """ 获取部门详细信息service :param result_db: orm对象 :param dasset_id: 部门id :return: 部门id对应的信息 """ dasset = DassetDao.get_dasset_detail_by_id(result_db, dasset_id=dasset_id) return dasset @classmethod def list_to_tree(cls, permission_list: list) -> list: """ 工具方法：根据部门列表信息生成树形嵌套数据 :param permission_list: 部门列表信息 :return: 部门树形嵌套数据 """ permission_list = [dict(title=item.dasset_name, key=str(item.dasset_id), value=str(item.dasset_id), dasset_parent_id=str(item.dasset_parent_id)) for item in permission_list] # 转成dasset_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['dasset_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_dasset_tree(cls, pid: int, permission_list: DassetTree): """ 工具方法：根据部门信息生成树形嵌套数据 :param pid: 部门id :param permission_list: 部门列表信息 :return: 部门树形嵌套数据 """ dasset_list = [] for permission in permission_list.dasset_tree: if permission.dasset_parent_id == pid: children = cls.get_dasset_tree(permission.dasset_id, permission_list) dasset_list_data = {} if children: dasset_list_data['title'] = permission.dasset_name dasset_list_data['key'] = str(permission.dasset_id) dasset_list_data['value'] = str(permission.dasset_id) dasset_list_data['children'] = children else: dasset_list_data['title'] = permission.dasset_name dasset_list_data['key'] = str(permission.dasset_id) dasset_list_data['value'] = str(permission.dasset_id) dasset_list.append(dasset_list_data) return dasset_list @classmethod def update_children_info(cls, result_db, page_object): """ 工具方法：递归更新子部门信息 :param result_db: orm对象 :param page_object: 编辑部门对象 :return: """ children_info = DassetDao.get_children_dasset(result_db, page_object.dasset_id) if children_info: for child in children_info: child.dasset_ancestors = f'{page_object.dasset_ancestors},{page_object.dasset_id}' DassetDao.edit_dasset_dao(result_db, dict(dasset_id=child.dasset_id, dasset_ancestors=child.dasset_ancestors, update_by=page_object.update_by, update_time=page_object.update_time ) ) cls.update_children_info(result_db, DassetModel(dasset_id=child.dasset_id, dasset_ancestors=child.dasset_ancestors, update_by=page_object.update_by, update_time=page_object.update_time ))