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.
265 lines
10 KiB
265 lines
10 KiB
from module_admin.entity.vo.user_vo import *
|
|
from module_admin.entity.vo.tsmcb_vo import *
|
|
from module_admin.dao.tsmcb_dao import *
|
|
from utils.common_util import export_list2excel
|
|
from utils.common_util import *
|
|
|
|
|
|
|
|
|
|
class TsmcbService:
|
|
"""
|
|
词性文本替换管理模块服务层
|
|
"""
|
|
|
|
@classmethod
|
|
def get_tsmcb_select_option_services(cls, result_db: Session):
|
|
"""
|
|
获取词性文本替换列表不分页信息service
|
|
:param result_db: orm对象
|
|
:return: 词性文本替换列表不分页信息对象
|
|
"""
|
|
tsmcb_list_result = TsmcbDao.get_tsmcb_select_option_dao(result_db)
|
|
|
|
return tsmcb_list_result
|
|
|
|
@classmethod
|
|
def get_tsmcb_list_services(cls, result_db: Session, query_object: TsmcbModel):
|
|
"""
|
|
获取词性文本替换列表信息service
|
|
:param result_db: orm对象
|
|
:param query_object: 查询参数对象
|
|
:return: 词性文本替换列表信息对象
|
|
"""
|
|
tsmcb_list_result = TsmcbDao.get_tsmcb_list(result_db, query_object)
|
|
|
|
return tsmcb_list_result
|
|
|
|
|
|
|
|
@classmethod
|
|
async def query_tsmcb_list_from_cache_services(cls, redis, onum: str):
|
|
"""
|
|
从缓存获取词性文本替换列表信息service
|
|
:param redis: redis对象
|
|
:param : 字典类型
|
|
:return: 词性文本替换列表信息对象
|
|
"""
|
|
result = []
|
|
dict_data_list_result = onum
|
|
print('测试使用，没有实际意义的方法！')
|
|
if dict_data_list_result:
|
|
result = onum
|
|
|
|
return result
|
|
|
|
|
|
|
|
@classmethod
|
|
def add_tsmcb_services(cls, result_db: Session, page_object: TsmcbModel):
|
|
"""
|
|
新增词性文本替换信息service
|
|
:param result_db: orm对象
|
|
:param page_object: 新增词性文本替换对象
|
|
:return: 新增词性文本替换校验结果
|
|
"""
|
|
tsmcb = TsmcbDao.get_tsmcb_detail_by_info(result_db, TsmcbModel(**dict(pos=page_object.pos)))
|
|
|
|
if tsmcb:
|
|
result = dict(is_success=False, message='词性文本替换名称已存在')
|
|
else:
|
|
try:
|
|
TsmcbDao.add_tsmcb_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 CrudTsmcbResponse(**result)
|
|
|
|
@classmethod
|
|
def edit_tsmcb_services(cls, result_db: Session, page_object: TsmcbModel):
|
|
"""
|
|
编辑词性文本替换信息service
|
|
:param result_db: orm对象
|
|
:param page_object: 编辑词性文本替换信息对象
|
|
:return: 编辑词性文本替换信息校验结果
|
|
"""
|
|
edit_tsmcb = page_object.dict(exclude_unset=True)
|
|
|
|
tsmcb_info = cls.detail_tsmcb_services(result_db, edit_tsmcb.get('onum'))
|
|
if tsmcb_info:
|
|
if tsmcb_info.onum != page_object.onum:
|
|
tsmcb = TsmcbDao.get_tsmcb_by_info(result_db, TsmcbModel(**dict(onum=page_object.onum)))
|
|
if tsmcb:
|
|
result = dict(is_success=False, message='字段名已存在')
|
|
return CrudTsmcbResponse(**result)
|
|
try:
|
|
TsmcbDao.edit_tsmcb_dao(result_db, edit_tsmcb)
|
|
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 CrudTsmcbResponse(**result)
|
|
|
|
@classmethod
|
|
def delete_tsmcb_services(cls, result_db: Session, page_object: DeleteTsmcbModel):
|
|
"""
|
|
删除词性文本替换信息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)
|
|
TsmcbDao.delete_tsmcb_dao(result_db, TsmcbModel(**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 CrudTsmcbResponse(**result)
|
|
|
|
@classmethod
|
|
def detail_tsmcb_services(cls, result_db: Session, onum: int):
|
|
"""
|
|
获取词性文本替换详细信息service
|
|
:param result_db: orm对象
|
|
:param onum: 词性文本替换id
|
|
:return: 词性文本替换id对应的信息
|
|
"""
|
|
tsmcb = TsmcbDao.get_tsmcb_detail_by_id(result_db, onum=onum)
|
|
|
|
return tsmcb
|
|
|
|
@staticmethod
|
|
def export_tsmcb_list_services(tsmcb_list: List):
|
|
"""
|
|
导出词性文本替换信息service
|
|
:param tsmcb_list: 词性文本替换信息列表
|
|
:return: 词性文本替换信息对应excel的二进制数据
|
|
"""
|
|
# 创建一个映射词性文本替换，将英文键映射到中文键
|
|
mapping_dict = {
|
|
"onum": "序号",
|
|
"pos": "词性",
|
|
"type": "类型",
|
|
"std_rpl_str": "标准化替换字符串",
|
|
"prefix_supp_str": "前缀补充字符串",
|
|
"suffix_supp_str": "后缀补充字符串",
|
|
"status": "状态",
|
|
#"create_by": "创建者",
|
|
#"create_time": "创建时间",
|
|
"update_by": "负责人",
|
|
"update_time": "更新时间",
|
|
}
|
|
|
|
data = [TsmcbModel(**vars(row)).dict() for row in tsmcb_list]
|
|
|
|
for item in data:
|
|
if item.get('status') == '0':
|
|
item['status'] = '正常'
|
|
else:
|
|
item['status'] = '停用'
|
|
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_tsmcb_services(cls, result_db: Session, tsmcb_import: ImportTsmcbModel, current_user: CurrentUserInfoServiceResponse):
|
|
"""
|
|
批量导入词性文本替换表service
|
|
:param tsmcb_import: 词性文本替换表导入参数对象
|
|
:param result_db: orm对象
|
|
:param current_tsmcb: 当前词性文本替换表对象
|
|
:return: 批量导入词性文本替换表结果
|
|
"""
|
|
header_dict = {
|
|
#"序号": "onum",
|
|
"词性": "pos",
|
|
"类型": "type",
|
|
"标准化替换字符串": "std_rpl_str",
|
|
"前缀补充字符串": "prefix_supp_str",
|
|
"后缀补充字符串": "suffix_supp_str",
|
|
"状态": "status"
|
|
}
|
|
filepath = get_filepath_from_url(tsmcb_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['status'] == '正常':
|
|
row['status'] = '0'
|
|
if row['status'] == '停用':
|
|
row['status'] = '1'
|
|
add_tsmcb = TsmcbModel(
|
|
**dict(
|
|
#onum=row['onum'],
|
|
pos=row['pos'],
|
|
type=row['type'],
|
|
std_rpl_str=row['std_rpl_str'],
|
|
prefix_supp_str=row['prefix_supp_str'],
|
|
suffix_supp_str=row['suffix_supp_str'],
|
|
status=row['status'],
|
|
create_by=current_user.user.user_name,
|
|
update_by=current_user.user.user_name,
|
|
)
|
|
)
|
|
tsmcb_infos = TsmcbDao.get_tsmcb_by_info_imp(result_db, TsmcbModel(**dict(pos=row['pos'])))
|
|
if tsmcb_infos:
|
|
if tsmcb_import.is_update:
|
|
edit_tsmcb = TsmcbModel(
|
|
**dict(
|
|
#onum=row['onum'],
|
|
#onum=int(tsmcb_info.onum),
|
|
pos=row['pos'],
|
|
type=row['type'],
|
|
std_rpl_str=row['std_rpl_str'],
|
|
prefix_supp_str=row['prefix_supp_str'],
|
|
suffix_supp_str=row['suffix_supp_str'],
|
|
status=row['status'],
|
|
update_by=current_user.user.user_name
|
|
)
|
|
).dict(exclude_unset=True)
|
|
|
|
TsmcbDao.edit_tsmcb_dao(result_db, edit_tsmcb)
|
|
else:
|
|
add_error_result.append(f"{count}.用户账号{row['user_name']}已存在")
|
|
else:
|
|
TsmcbDao.add_tsmcb_dao(result_db, add_tsmcb)
|
|
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 CrudTsmcbResponse(**result)
|
|
|
|
@staticmethod
|
|
def get_tsmcb_import_template_services():
|
|
"""
|
|
获取词性文本替换表导入模板service
|
|
:return: 词性文本替换表导入模板excel的二进制数据
|
|
"""
|
|
#header_list = ["序号", "词性", "类型", "标准化替换字符串", "前缀补充字符串", "后缀补充字符串", "状态"]
|
|
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
|