156 lines
8.0 KiB
Python
156 lines
8.0 KiB
Python
# 导入内置模块
|
||
from pathlib import Path
|
||
# 导入django、ninja等模块
|
||
from ninja_extra import api_controller, ControllerBase, route
|
||
from django.db import transaction
|
||
from django.shortcuts import get_object_or_404
|
||
# 导入文档处理模块
|
||
from docxtpl import DocxTemplate, InlineImage
|
||
# 导入ORM模型
|
||
from apps.project.models import Project
|
||
# 导入工具
|
||
from utils.util import get_str_abbr, get_str_dict
|
||
from utils.chen_response import ChenResponse
|
||
from utils.path_utils import project_path
|
||
from apps.createDocument.extensions.parse_rich_text import RichParser
|
||
# 导入生成日志记录模块
|
||
from apps.createSeiTaiDocument.extensions.logger import GenerateLogger
|
||
|
||
gloger = GenerateLogger("问题单二段文档")
|
||
|
||
# @api_controller("/generateWtd", tags=['生成问题单文档系列'], auth=JWTAuth(), permissions=[IsAuthenticated])
|
||
@api_controller('/generateWtd', tags=['生成问题单文档系列'])
|
||
class GenerateControllerWtd(ControllerBase):
|
||
@route.get("/create/problem", url_name="create-problem")
|
||
@transaction.atomic
|
||
def create_problem(self, id: int):
|
||
"""生成问题单"""
|
||
project_path_str = project_path(id)
|
||
tpl_path = Path.cwd() / 'media' / project_path_str / 'form_template/wtd' / '问题详情表.docx'
|
||
doc = DocxTemplate(tpl_path)
|
||
project_obj = get_object_or_404(Project, id=id)
|
||
problem_list = list(project_obj.projField.distinct()) # 去掉重复，因为和case是多对多
|
||
problem_list.sort(key=lambda x: int(x.ident))
|
||
data_list = []
|
||
for problem in problem_list:
|
||
problem_dict = {'ident': problem.ident, 'name': problem.name}
|
||
# 1.生成被测对象名称、被测对象标识、被测对象版本
|
||
cases = problem.case.all()
|
||
# generate_log:无关联问题单进入生成日志
|
||
if cases.count() < 1:
|
||
gloger.write_warning_log('单个问题单表格', f'问题单{problem.ident}未关联用例，请检查')
|
||
str_dut_name_list = []
|
||
str_dut_ident_list = []
|
||
str_dut_version_list = []
|
||
# 2.所属用例标识
|
||
case_ident_list = []
|
||
# 3.获取依据要求
|
||
case_design_list = []
|
||
for case in cases:
|
||
if case.test.testType == '8':
|
||
# 1.1.如果为文档审查，提取所属文档名称、文档被测件标识、文档被测件版本
|
||
str_dut_name_list.append(case.dut.name)
|
||
str_dut_ident_list.append(case.dut.ref)
|
||
str_dut_version_list.append(case.dut.version)
|
||
# 对应dut名称，design章节号，design描述
|
||
case_design_list.append("".join([case.dut.name, case.design.chapter]))
|
||
else:
|
||
# 1.2.如果不为文档审查，则提取该轮次源代码dut的信息
|
||
so_dut = case.round.rdField.filter(type='SO').first()
|
||
if so_dut:
|
||
str_dut_name_list.append(project_obj.name + '软件')
|
||
str_dut_ident_list.append(so_dut.ref)
|
||
str_dut_version_list.append(so_dut.version)
|
||
# TODO:如何处理设计需求的内容，暂时设置为取出图片，只保留文字
|
||
p_list = []
|
||
rich_parse_remove_img = RichParser(case.design.description)
|
||
rich_list = rich_parse_remove_img.get_final_list(doc)
|
||
for rich in rich_list:
|
||
if isinstance(rich, dict) or isinstance(rich, InlineImage):
|
||
continue
|
||
else:
|
||
p_list.append(rich)
|
||
|
||
case_design_list.append(
|
||
"-".join([case.dut.name, case.design.chapter + '章节' + ":" + ''.join(p_list)]))
|
||
# 2.用例标识修改-YL_测试项类型_测试项标识_用例key+1
|
||
demand = case.test # 中间变量
|
||
demand_testType = demand.testType # 中间变量
|
||
testType_abbr = get_str_abbr(demand_testType, 'testType') # 输出FT
|
||
case_ident_list.append("_".join(
|
||
['YL', testType_abbr, demand.ident, str(int(case.key[-1]) + 1).rjust(3, '0')]))
|
||
problem_dict['duts_name'] = "/".join(set(str_dut_name_list))
|
||
problem_dict['duts_ref'] = "/".join(set(str_dut_ident_list))
|
||
problem_dict['duts_version'] = "/".join(set(str_dut_version_list))
|
||
temp_name_version = []
|
||
for i in range(len(str_dut_name_list)):
|
||
temp_name_version.append(
|
||
"".join([str_dut_name_list[i] + str_dut_ident_list[i], '/V', str_dut_version_list[i]]))
|
||
problem_dict['dut_name_version'] = "\a".join(temp_name_version)
|
||
problem_dict['case_ident'] = "，".join(set(case_ident_list))
|
||
problem_dict['type'] = get_str_dict(problem.type, 'problemType')
|
||
problem_dict['grade'] = get_str_dict(problem.grade, 'problemGrade')
|
||
|
||
# 依据要求-获取其设计需求
|
||
problem_dict['yaoqiu'] = "\a".join(case_design_list)
|
||
# 问题操作 - HTML解析
|
||
desc_list = ['【问题操作】']
|
||
rich_parser = RichParser(problem.operation)
|
||
desc_list.extend(rich_parser.get_final_list(doc))
|
||
|
||
# 问题影响
|
||
desc_list_result = [f'\a【问题影响】\a{problem.result}']
|
||
desc_list.extend(desc_list_result)
|
||
# 问题描述赋值
|
||
problem_dict['desc'] = desc_list
|
||
|
||
# 4.原因分析
|
||
desc_list_3 = [f'【原因分析】\a{problem.analysis}']
|
||
problem_dict['cause'] = desc_list_3
|
||
|
||
# 5.影响域分析~~~~
|
||
desc_list_4 = [f'【影响域分析】\a{problem.effect_scope}']
|
||
problem_dict['effect_scope'] = desc_list_4
|
||
|
||
# 6.改正措施
|
||
problem_dict['solve'] = problem.solve
|
||
|
||
# 7.回归验证结果
|
||
desc_list_5 = []
|
||
rich_parser5 = RichParser(problem.verify_result)
|
||
desc_list_5.extend(rich_parser5.get_final_list(doc))
|
||
problem_dict['verify_result'] = desc_list_5
|
||
|
||
# 8.其他日期和人员
|
||
problem_dict['postPerson'] = problem.postPerson
|
||
problem_dict['postDate'] = problem.postDate
|
||
close_str = '□修改文档 □修改程序 □不修改'
|
||
if len(problem.closeMethod) < 1:
|
||
close_str = '□修改文档 □修改程序 ■不修改'
|
||
elif len(problem.closeMethod) == 2:
|
||
close_str = '■修改文档 ■修改程序 □不修改'
|
||
else:
|
||
if problem.closeMethod[0] == '1':
|
||
close_str = '■修改文档 □修改程序 □不修改'
|
||
elif problem.closeMethod[0] == '2':
|
||
close_str = '□修改文档 ■修改程序 □不修改'
|
||
else:
|
||
close_str = '□修改文档 □修改程序 □不修改'
|
||
problem_dict['closeMethod'] = close_str
|
||
problem_dict['designer'] = problem.designerPerson
|
||
problem_dict['designDate'] = problem.designDate
|
||
problem_dict['verifyPerson'] = problem.verifyPerson
|
||
problem_dict['verifyDate'] = problem.verifyDate
|
||
data_list.append(problem_dict)
|
||
context = {
|
||
'project_name': project_obj.name,
|
||
'project_ident': project_obj.ident,
|
||
'problem_list': data_list,
|
||
}
|
||
doc.render(context, autoescape=True)
|
||
try:
|
||
doc.save(Path.cwd() / "media" / project_path_str / "output_dir/wtd" / '问题详情表.docx')
|
||
return ChenResponse(status=200, code=200, message="文档生成成功！")
|
||
except PermissionError as e:
|
||
return ChenResponse(status=400, code=400, message="模版文件已打开，请关闭后再试，{0}".format(e))
|