You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 

237 lines
11 KiB

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
))