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.
284 lines
11 KiB
284 lines
11 KiB
1 month ago
|
from module_admin.entity.vo.user_vo import *
|
||
|
|
from module_admin.entity.vo.sscf_vo import *
|
||
|
|
from module_admin.dao.sscf_dao import *
|
||
|
|
from utils.pwd_util import *
|
||
|
|
from utils.common_util import *
|
||
|
|
|
||
|
|
|
||
|
|
class SscfService:
|
||
|
|
"""
|
||
|
|
短句配置模块服务层
|
||
|
|
"""
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def get_sscf_list_services(cls, result_db: Session, query_object: SscfQueryModel, data_scope_sql: str):
|
||
|
|
"""
|
||
|
|
获取短句配置信息service
|
||
|
|
:param result_db: orm对象
|
||
|
|
:param query_object: 查询参数对象
|
||
|
|
:param data_scope_sql: 数据权限对应的查询sql语句
|
||
|
|
:return: 字段列表信息对象
|
||
|
|
"""
|
||
|
|
sscf_list_result = SscfDao.get_sscf_list(result_db, query_object, data_scope_sql)
|
||
|
|
|
||
|
|
return sscf_list_result
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def add_sscf_services(cls, result_db: Session, page_object: AddSscfModel):
|
||
|
|
"""
|
||
|
|
新增短句配置信息service
|
||
|
|
:param result_db: orm对象
|
||
|
|
:param page_object: 新增短句配置对象
|
||
|
|
:return: 新增字段校验结果
|
||
|
|
"""
|
||
|
|
add_sscf = SscfModel(**page_object.dict())
|
||
|
|
sscf = SscfDao.get_sscf_by_info(result_db, SscfModel(**dict(keyword=page_object.keyword)))
|
||
|
|
if sscf:
|
||
|
|
result = dict(is_success=False, message='字段名已存在')
|
||
|
|
else:
|
||
|
|
try:
|
||
|
|
add_result = SscfDao.add_sscf_dao(result_db, add_sscf)
|
||
|
|
onum = add_result.onum
|
||
|
|
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 CrudSscfResponse(**result)
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def edit_sscf_services(cls, result_db: Session, page_object: AddSscfModel):
|
||
|
|
"""
|
||
|
|
编辑短句配置信息service
|
||
|
|
:param result_db: orm对象
|
||
|
|
:param page_object: 编辑短句配置对象
|
||
|
|
:return: 编辑短句配置校验结果
|
||
|
|
"""
|
||
|
|
edit_sscf = page_object.dict(exclude_unset=True)
|
||
|
|
if page_object.type == 'status' or page_object.type == 'avatar':
|
||
|
|
del edit_sscf['type']
|
||
|
|
sscf_info = cls.detail_sscf_services(result_db, edit_sscf.get('onum'))
|
||
|
|
if sscf_info:
|
||
|
|
if page_object.type != 'status' and page_object.type != 'avatar' and sscf_info.sscf.onum != page_object.onum:
|
||
|
|
sscf = SscfDao.get_sscf_by_info(result_db, SscfModel(**dict(onum=page_object.onum)))
|
||
|
|
if sscf:
|
||
|
|
result = dict(is_success=False, message='字段名已存在')
|
||
|
|
return CrudSscfResponse(**result)
|
||
|
|
try:
|
||
|
|
SscfDao.edit_sscf_dao(result_db, edit_sscf)
|
||
|
|
if page_object.type != 'status' and page_object.type != 'avatar':
|
||
|
|
onum_dict = dict(onum=page_object.onum)
|
||
|
|
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 CrudSscfResponse(**result)
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def delete_sscf_services(cls, result_db: Session, page_object: DeleteSscfModel):
|
||
|
|
"""
|
||
|
|
删除短句配置信息service
|
||
|
|
:param result_db: orm对象
|
||
|
|
:param page_object: 删除短句配置对象
|
||
|
|
:return: 删除短句配置校验结果
|
||
|
|
"""
|
||
|
|
|
||
|
|
if page_object.onums.split(','):
|
||
|
|
onum_list = page_object.onums.split(',')
|
||
|
|
|
||
|
|
try:
|
||
|
|
for onum in onum_list:
|
||
|
|
onum_dict = dict(onum=onum, update_by=page_object.update_by, update_time=page_object.update_time) #
|
||
|
|
SscfDao.delete_sscf_dao(result_db, SscfModel(**onum_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='传入短句配置onum为空')
|
||
|
|
return CrudSscfResponse(**result)
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def detail_sscf_services(cls, result_db: Session, onum: int):
|
||
|
|
"""
|
||
|
|
获取短句配置详细信息service
|
||
|
|
:param result_db: orm对象
|
||
|
|
:param onum: 短句配置id
|
||
|
|
:return: 短句配置onum对应的信息
|
||
|
|
"""
|
||
|
|
sscf = SscfDao.get_sscf_detail_by_id(result_db, onum=onum)
|
||
|
|
|
||
|
|
return SscfDetailModel(
|
||
|
|
sscf=sscf.sscf_basic_info,
|
||
|
|
dasset=sscf.sscf_dasset_info
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def export_sscf_list_services(sscf_list: List):
|
||
|
|
"""
|
||
|
|
导出配置表信息service
|
||
|
|
:param sscf_list: 代码信息列表
|
||
|
|
:return: 配置表信息对应excel的二进制数据
|
||
|
|
"""
|
||
|
|
# 创建一个映射代码，将英文键映射到中文键
|
||
|
|
mapping_dict = {
|
||
|
|
"dasset_id": "数据域名",
|
||
|
|
"onum": "序号",
|
||
|
|
#"keywords": "关键词JSON",
|
||
|
|
"keyword": "关键词",
|
||
|
|
"algorithm": "算法",
|
||
|
|
"order": "指令",
|
||
|
|
"whole_sentence": "整句",
|
||
|
|
"type": "类型",
|
||
|
|
"supp_expl": "补充说明",
|
||
|
|
# "bak1": "备注1",
|
||
|
|
# "bak2": "备注2",
|
||
|
|
# "bak3": "备注3",
|
||
|
|
# "bak4": "备注4",
|
||
|
|
# "bak5": "备注5",
|
||
|
|
"status": "状态",
|
||
|
|
}
|
||
|
|
|
||
|
|
#data = [SscfModel(**vars(row)).dict() for row in sscf_list]
|
||
|
|
data = [SscfModel(**row).dict() for row in sscf_list]
|
||
|
|
for item in data:
|
||
|
|
if item.get('status') == '0':
|
||
|
|
item['status'] = '正常'
|
||
|
|
else:
|
||
|
|
item['status'] = '停用'
|
||
|
|
if item.get('dasset_id') == '101':
|
||
|
|
item['dasset_id'] = '资讯域'
|
||
|
|
elif item.get('dasset_id') == '102':
|
||
|
|
item['dasset_id'] = '财务域'
|
||
|
|
else:
|
||
|
|
item['dasset_id'] == '其他域'
|
||
|
|
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
|
||
|
|
def batch_import_sscf_services(cls, result_db: Session, sscf_import: ImportSscfModel, current_user: CurrentUserInfoServiceResponse):
|
||
|
|
"""
|
||
|
|
批量导入短句配置表service
|
||
|
|
:param sscf_import: 短句配置表导入参数对象
|
||
|
|
:param result_db: orm对象
|
||
|
|
:param current_sscf: 当前短句配置表对象
|
||
|
|
:return: 批量导入短句配置表结果
|
||
|
|
"""
|
||
|
|
header_dict = {
|
||
|
|
"数据域名":"dasset_id",
|
||
|
|
#"序号":"onum",
|
||
|
|
#"关键词JSON":"keywords",
|
||
|
|
"关键词":"keyword",
|
||
|
|
"算法":"algorithm",
|
||
|
|
"指令":"order",
|
||
|
|
"整句":"whole_sentence",
|
||
|
|
"类型":"type",
|
||
|
|
"补充说明":"supp_expl",
|
||
|
|
# "备注1":"bak1",
|
||
|
|
# "备注2":"bak2",
|
||
|
|
# "备注3":"bak3",
|
||
|
|
# "备注4":"bak4",
|
||
|
|
# "备注5":"bak5",
|
||
|
|
"状态":"status",
|
||
|
|
}
|
||
|
|
filepath = get_filepath_from_url(sscf_import.url)
|
||
|
|
df = pd.read_excel(filepath)
|
||
|
|
df.rename(columns=header_dict, inplace=True)
|
||
|
|
add_error_result = []
|
||
|
|
count = 0
|
||
|
|
try:
|
||
|
|
for index, row in df.iterrows():
|
||
|
|
count = count + 1
|
||
|
|
if row['dasset_id'] == '资讯域':
|
||
|
|
row['dasset_id'] = '101'
|
||
|
|
if row['dasset_id'] == '财务域':
|
||
|
|
row['dasset_id'] = '102'
|
||
|
|
if row['dasset_id'] == '账户域':
|
||
|
|
row['dasset_id'] = '145'
|
||
|
|
if row['dasset_id'] == '其他域':
|
||
|
|
row['dasset_id'] = '121'
|
||
|
|
if row['status'] == '正常':
|
||
|
|
row['status'] = '0'
|
||
|
|
if row['status'] == '停用':
|
||
|
|
row['status'] = '1'
|
||
|
|
add_sscf = SscfModel(
|
||
|
|
**dict(
|
||
|
|
dasset_id=row['dasset_id'],
|
||
|
|
#onum=row['onum'],
|
||
|
|
#keywords=row['keywords'],
|
||
|
|
keyword=row['keyword'],
|
||
|
|
algorithm=row['algorithm'],
|
||
|
|
order=row['order'],
|
||
|
|
whole_sentence=row['whole_sentence'],
|
||
|
|
supp_expl=row['supp_expl'],
|
||
|
|
type=row['type'],
|
||
|
|
# bak1=row['bak1'],
|
||
|
|
# bak2=row['bak2'],
|
||
|
|
# bak3=row['bak3'],
|
||
|
|
# bak4=row['bak4'],
|
||
|
|
# bak5=row['bak5'],
|
||
|
|
status=row['status'],
|
||
|
|
create_by=current_user.user.user_name,
|
||
|
|
update_by=current_user.user.user_name,
|
||
|
|
)
|
||
|
|
)
|
||
|
|
sscf_info = SscfDao.get_sscf_by_info_imp(result_db, SscfModel(**dict(keyword=row['keyword'])))
|
||
|
|
if sscf_info:
|
||
|
|
if sscf_import.is_update:
|
||
|
|
edit_sscf = SscfModel(
|
||
|
|
**dict(
|
||
|
|
dasset_id=row['dasset_id'],
|
||
|
|
#onum=row['onum'],
|
||
|
|
#keywords=row['keywords'],
|
||
|
|
keyword=row['keyword'],
|
||
|
|
algorithm=row['algorithm'],
|
||
|
|
order=row['order'],
|
||
|
|
whole_sentence=row['whole_sentence'],
|
||
|
|
supp_expl=row['supp_expl'],
|
||
|
|
type=row['type'],
|
||
|
|
# bak1=row['bak1'],
|
||
|
|
# bak2=row['bak2'],
|
||
|
|
# bak3=row['bak3'],
|
||
|
|
# bak4=row['bak4'],
|
||
|
|
# bak5=row['bak5'],
|
||
|
|
status=row['status'],
|
||
|
|
update_by=current_user.user.user_name
|
||
|
|
)
|
||
|
|
).dict(exclude_unset=True)
|
||
|
|
SscfDao.edit_sscf_dao(result_db, edit_sscf)
|
||
|
|
else:
|
||
|
|
add_error_result.append(f"{count}.用户账号{row['user_name']}已存在")
|
||
|
|
else:
|
||
|
|
SscfDao.add_sscf_dao(result_db, add_sscf)
|
||
|
|
result_db.commit()
|
||
|
|
result = dict(is_success=True, message='\n'.join(add_error_result))
|
||
|
|
except Exception as e:
|
||
|
|
result_db.rollback()
|
||
|
|
result = dict(is_success=False, message=str(e))
|
||
|
|
|
||
|
|
return CrudSscfResponse(**result)
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def get_sscf_import_template_services():
|
||
|
|
"""
|
||
|
|
获取短句配置表导入模板service
|
||
|
|
:return: 短句配置表导入模板excel的二进制数据
|
||
|
|
"""
|
||
|
|
# header_list = ["数据域名", "序号", "关键词JSON", "关键词", "算法", "指令", "整句", "类型", "补充说明", "备注1", "备注2", "备注3", "备注4", "备注5", "状态"]
|
||
|
|
# header_list = ["数据域名", "关键词", "算法", "指令", "整句", "类型", "补充说明", "备注1", "备注2", "备注3", "备注4", "备注5", "状态"]
|
||
|
|
header_list = ["数据域名", "关键词", "算法", "指令", "整句", "类型", "补充说明", "状态"]
|
||
|
|
selector_header_list = ["数据域名","状态"]
|
||
|
|
option_list = [{"数据域名": ["资讯域", "财务域","账户域", "其他域"]},{"状态": ["正常", "停用"]}]
|
||
|
|
binary_data = get_excel_template(header_list=header_list, selector_header_list=selector_header_list, option_list=option_list)
|
||
|
|
|
||
|
|
return binary_data
|