日常修复内容

This commit is contained in:
2026-04-20 18:00:55 +08:00
parent 518f2f43a7
commit dffc1d5872
188 changed files with 3969 additions and 3175 deletions

File diff suppressed because it is too large Load Diff

File diff suppressed because it is too large Load Diff

View File

@@ -1,212 +1,216 @@
from copy import deepcopy
from pathlib import Path
from ninja_extra import api_controller, ControllerBase, route
from ninja_extra.permissions import IsAuthenticated
from ninja_jwt.authentication import JWTAuth
from django.db import transaction
from django.db.models import QuerySet
from docxtpl import DocxTemplate
from apps.dict.models import Dict
from utils.chen_response import ChenResponse
from django.shortcuts import get_object_or_404
from typing import Union
from docxtpl import InlineImage
from apps.project.models import Dut, Project, Round
from utils.util import get_list_dict, get_str_dict, get_ident, get_case_ident
from utils.chapter_tools.csx_chapter import create_csx_chapter_dict
from utils.path_utils import project_path
from apps.createDocument.extensions.util import delete_dir_files
from apps.createDocument.extensions.parse_rich_text import RichParser
# 导入生成日志记录模块
from apps.createSeiTaiDocument.extensions.logger import GenerateLogger
chinese_round_name: list = ['', '', '', '', '', '', '', '', '', '']
# @api_controller("/generateHSM", tags=['生成回归记录系列文档'], auth=JWTAuth(), permissions=[IsAuthenticated])
@api_controller("/generateHJL", tags=['生成回归记录系列文档'])
class GenerateControllerHJL(ControllerBase):
logger = GenerateLogger('回归测试记录')
# important删除之前的文件
@route.get('/create/deleteHJLDocument', url_name='delete-hjl-document')
def delete_hjl_document(self, id: int):
project_path_str = project_path(id)
save_path = Path.cwd() / 'media' / project_path_str / 'output_dir/hjl'
delete_dir_files(save_path)
@route.get("/create/basicInformation", url_name="create-basicInformation")
@transaction.atomic
def create_basicInformation(self, id: int):
"""生成回归测试记录的被测软件基本信息"""
project_path_str = project_path(id)
tpl_path = Path.cwd() / 'media' / project_path_str / 'form_template/hjl' / '被测软件基本信息.docx'
doc = DocxTemplate(tpl_path)
project_obj = get_object_or_404(Project, id=id)
# 第一轮次对象
round1_obj: Union[Round, None] = project_obj.pField.filter(key='0').first()
# 第一轮源代码被测件对象
round1_so_dut: Union[Dut, None] = round1_obj.rdField.filter(type='SO').first()
languages = get_list_dict('language', project_obj.language)
language_list = [item['ident_version'] for item in languages]
# 取非第一轮次
hround_list: QuerySet = project_obj.pField.exclude(key='0')
if len(hround_list) < 1:
# ***Inspect-start***
self.logger.model = '回归测试记录'
self.logger.write_warning_log('当前文档全部片段', f'该项目没有创建轮次')
# ***Inspect-end***
return ChenResponse(code=400, status=400, message='您未创建轮次,请创建完毕后再试')
context = {
'project_name': project_obj.name,
'language': "".join(language_list),
'soft_type': project_obj.get_soft_type_display(),
'security_level': get_str_dict(project_obj.security_level, 'security_level'),
'runtime': get_str_dict(project_obj.runtime, 'runtime'),
'devplant': get_str_dict(project_obj.devplant, 'devplant'),
'recv_date': project_obj.beginTime.strftime("%Y-%m-%d"),
'dev_unit': project_obj.dev_unit,
}
version_info = [{'version': round1_so_dut.version,
'line_count': int(round1_so_dut.total_lines),
'effective_line': int(round1_so_dut.effective_lines)}]
# 循环回归的轮次
for hround in hround_list:
# 每个轮次独立渲染context
context_round = deepcopy(context)
# 取中文名称
cname = chinese_round_name[int(hround.key)] # 输出二、三...
# 取该轮次源代码版本放入版本列表
so_dut: Dut = hround.rdField.filter(type='SO').first()
if not so_dut:
return ChenResponse(code=400, status=400, message=f'您第{cname}轮次中缺少源代码被测件,请添加')
version_info.append(
{'version': so_dut.version, 'line_count': int(so_dut.total_lines),
'effective_line': int(so_dut.effective_lines)})
context_round['version_info'] = version_info
# 开始渲染每个轮次的二级文档
save_path = Path.cwd() / 'media' / project_path_str / 'output_dir/hjl' / f"{cname}轮被测软件基本信息.docx"
doc.render(context=context_round, autoescape=True)
try:
doc.save(save_path)
except PermissionError:
return ChenResponse(code=400, status=400, message='您打开了生成的文档,请关闭后重试')
return ChenResponse(code=200, status=200, message='多轮回归说明文档基本信息生成完毕')
@route.get("/create/caseinfo", url_name="create-caseinfo")
@transaction.atomic
def create_caseinfo(self, id: int):
"""生成回归测试记录的-{测试用例记录}"""
project_path_str = project_path(id)
tpl_path = Path.cwd() / 'media' / project_path_str / 'form_template/hjl' / '测试用例记录.docx'
doc = DocxTemplate(tpl_path)
project_obj = get_object_or_404(Project, id=id)
hround_list: QuerySet = project_obj.pField.exclude(key='0')
if len(hround_list) < 1:
return None
demand_prefix = '3.1'
# 循环每轮轮次对象
for hround in hround_list:
cname = chinese_round_name[int(hround.key)] # var输出二、三字样
test_type_len = Dict.objects.get(code='testType').dictItem.count() # 测试类型的个数
type_number_list = [i for i in range(1, test_type_len + 1)] # 测试类型编号对应的列表
list_list = [[] for j in range(1, test_type_len + 1)] # 每个测试类型组合为一个列表[[],[],[],[]]
testType_list, last_chapter_items = create_csx_chapter_dict(hround)
testDemands = hround.rtField.all() # 本轮所有测试项
for demand in testDemands:
type_index = type_number_list.index(int(demand.testType))
demand_ident = get_ident(demand)
# ~~~组装测试项~~~
demand_last_chapter = last_chapter_items[demand.testType].index(demand.key) + 1
demand_chapter = ".".join([demand_prefix, str(testType_list.index(demand.testType) + 1),
str(demand_last_chapter)])
demand_dict = {
'name': demand.name,
'ident': demand_ident,
'chapter': demand_chapter,
'item': []
}
# ~~~这里组装测试项里面的测试用例~~~
for case in demand.tcField.all():
step_list = []
index = 1
for one in case.step.all():
# 这里需要对operation富文本处理
rich_parser = RichParser(one.operation)
desc_list = rich_parser.get_final_list(doc, img_size=68)
rich_parser2 = RichParser(one.result)
res_list = rich_parser2.get_final_list(doc, img_size=75)
# 组装用例里面的步骤dict
passed = '通过'
if one.passed == '2':
passed = '未通过'
if one.passed == '3':
passed = '未执行'
step_dict = {
'index': index,
'operation': desc_list,
'expect': one.expect,
'result': res_list,
'passed': passed,
}
step_list.append(step_dict)
index += 1
# 查询所有的problem
problem_list = []
problem_prefix = "PT"
proj_ident = project_obj.ident
for problem in case.caseField.all():
problem_list.append("_".join([problem_prefix, proj_ident, problem.ident]))
# fpga的时序图
rich_parser3 = RichParser(case.timing_diagram)
timing_diagram = rich_parser3.get_final_list(doc, img_size=115, height=50)
has_timing_diagram = False
if len(timing_diagram) > 0:
if isinstance(timing_diagram[0], InlineImage):
has_timing_diagram = True
# 组装用例的dict
case_dict = {
'name': case.name,
'ident': get_case_ident(demand_ident, case),
'summary': case.summarize,
'initialization': case.initialization,
'premise': case.premise,
'design_person': case.designPerson,
'test_person': case.testPerson,
'monitor_person': case.monitorPerson,
'step': step_list,
'time': str(case.exe_time) if case.exe_time is not None else str(case.update_datetime),
'problems': "".join(problem_list),
'round_num_chn': cname,
# 2025年4月24日新增
'has_timing_diagram': has_timing_diagram,
'timing_diagram': timing_diagram,
}
demand_dict['item'].append(case_dict)
list_list[type_index].append(demand_dict)
# 定义渲染上下文
context = {}
output_list = []
for (index, li) in enumerate(list_list):
qs = Dict.objects.get(code="testType").dictItem.get(key=str(index + 1))
context_str = qs.title
sort = qs.sort
table = {
"type": context_str,
"item": li,
"sort": sort
}
output_list.append(table)
# 排序
output_list = sorted(output_list, key=(lambda x: x["sort"]))
context["data"] = output_list
# 最后渲染
save_path = Path.cwd() / 'media' / project_path_str / 'output_dir/hjl' / f"{cname}轮测试用例记录.docx"
doc.render(context, autoescape=True)
try:
doc.save(save_path)
except PermissionError:
return ChenResponse(code=400, status=400, message='您打开了生成的文档,请关闭后重试')
return ChenResponse(code=200, status=200, message='多轮回归测试用例记录生成完毕')
from copy import deepcopy
from pathlib import Path
from ninja_extra import api_controller, ControllerBase, route
from ninja_extra.permissions import IsAuthenticated
from ninja_jwt.authentication import JWTAuth
from django.db import transaction
from django.db.models import QuerySet
from docxtpl import DocxTemplate
from apps.dict.models import Dict
from utils.chen_response import ChenResponse
from django.shortcuts import get_object_or_404
from typing import Union
from docxtpl import InlineImage
from apps.project.models import Dut, Project, Round
from utils.util import get_list_dict, get_str_dict, get_ident, get_case_ident
from utils.chapter_tools.csx_chapter import create_csx_chapter_dict
from utils.path_utils import project_path
from apps.createDocument.extensions.util import delete_dir_files
from apps.createDocument.extensions.parse_rich_text import RichParser
# 导入生成日志记录模块
from apps.createSeiTaiDocument.extensions.logger import GenerateLogger
chinese_round_name: list = ['', '', '', '', '', '', '', '', '', '']
# @api_controller("/generateHSM", tags=['生成回归记录系列文档'], auth=JWTAuth(), permissions=[IsAuthenticated])
@api_controller("/generateHJL", tags=['生成回归记录系列文档'])
class GenerateControllerHJL(ControllerBase):
logger = GenerateLogger('回归测试记录')
# important删除之前的文件
@route.get('/create/deleteHJLDocument', url_name='delete-hjl-document')
def delete_hjl_document(self, id: int):
project_path_str = project_path(id)
save_path = Path.cwd() / 'media' / project_path_str / 'output_dir/hjl'
delete_dir_files(save_path)
@route.get("/create/basicInformation", url_name="create-basicInformation")
@transaction.atomic
def create_basicInformation(self, id: int):
"""生成回归测试记录的被测软件基本信息"""
project_path_str = project_path(id)
tpl_path = Path.cwd() / 'media' / project_path_str / 'form_template/hjl' / '被测软件基本信息.docx'
doc = DocxTemplate(tpl_path)
project_obj = get_object_or_404(Project, id=id)
# 第一轮次对象
round1_obj: Union[Round, None] = project_obj.pField.filter(key='0').first()
# 第一轮源代码被测件对象
round1_so_dut: Union[Dut, None] = round1_obj.rdField.filter(type='SO').first()
languages = get_list_dict('language', project_obj.language)
language_list = [item['ident_version'] for item in languages]
runtimes = get_list_dict('runtime', project_obj.runtime)
runtime_list = [item['ident_version'] for item in runtimes]
devplants = get_list_dict('devplant', project_obj.devplant)
devplant_list = [item['ident_version'] for item in devplants]
# 取非第一轮次
hround_list: QuerySet = project_obj.pField.exclude(key='0')
if len(hround_list) < 1:
# ***Inspect-start***
self.logger.model = '回归测试记录'
self.logger.write_warning_log('当前文档全部片段', f'该项目没有创建轮次')
# ***Inspect-end***
return ChenResponse(code=400, status=400, message='您未创建轮次,请创建完毕后再试')
context = {
'project_name': project_obj.name,
'language': "".join(language_list),
'soft_type': project_obj.get_soft_type_display(),
'security_level': get_str_dict(project_obj.security_level, 'security_level'),
'runtime': "".join(runtime_list),
'devplant': "".join(devplant_list),
'recv_date': project_obj.beginTime.strftime("%Y-%m-%d"),
'dev_unit': project_obj.dev_unit,
}
version_info = [{'version': round1_so_dut.version,
'line_count': int(round1_so_dut.total_lines),
'effective_line': int(round1_so_dut.effective_lines)}]
# 循环回归的轮次
for hround in hround_list:
# 每个轮次独立渲染context
context_round = deepcopy(context)
# 取中文名称
cname = chinese_round_name[int(hround.key)] # 输出二、三...
# 取该轮次源代码版本放入版本列表
so_dut: Dut = hround.rdField.filter(type='SO').first()
if not so_dut:
return ChenResponse(code=400, status=400, message=f'您第{cname}轮次中缺少源代码被测件,请添加')
version_info.append(
{'version': so_dut.version, 'line_count': int(so_dut.total_lines),
'effective_line': int(so_dut.effective_lines)})
context_round['version_info'] = version_info
# 开始渲染每个轮次的二级文档
save_path = Path.cwd() / 'media' / project_path_str / 'output_dir/hjl' / f"{cname}轮被测软件基本信息.docx"
doc.render(context=context_round, autoescape=True)
try:
doc.save(save_path)
except PermissionError:
return ChenResponse(code=400, status=400, message='您打开了生成的文档,请关闭后重试')
return ChenResponse(code=200, status=200, message='多轮回归说明文档基本信息生成完毕')
@route.get("/create/caseinfo", url_name="create-caseinfo")
@transaction.atomic
def create_caseinfo(self, id: int):
"""生成回归测试记录的-{测试用例记录}"""
project_path_str = project_path(id)
tpl_path = Path.cwd() / 'media' / project_path_str / 'form_template/hjl' / '测试用例记录.docx'
doc = DocxTemplate(tpl_path)
project_obj = get_object_or_404(Project, id=id)
hround_list: QuerySet = project_obj.pField.exclude(key='0')
if len(hround_list) < 1:
return None
demand_prefix = '3.1'
# 循环每轮轮次对象
for hround in hround_list:
cname = chinese_round_name[int(hround.key)] # var输出二、三字样
test_type_len = Dict.objects.get(code='testType').dictItem.count() # 测试类型的个数
type_number_list = [i for i in range(1, test_type_len + 1)] # 测试类型编号对应的列表
list_list = [[] for j in range(1, test_type_len + 1)] # 每个测试类型组合为一个列表[[],[],[],[]]
testType_list, last_chapter_items = create_csx_chapter_dict(hround)
testDemands = hround.rtField.all() # 本轮所有测试项
for demand in testDemands:
type_index = type_number_list.index(int(demand.testType))
demand_ident = get_ident(demand)
# ~~~组装测试项~~~
demand_last_chapter = last_chapter_items[demand.testType].index(demand.key) + 1
demand_chapter = ".".join([demand_prefix, str(testType_list.index(demand.testType) + 1),
str(demand_last_chapter)])
demand_dict = {
'name': demand.name,
'ident': demand_ident,
'chapter': demand_chapter,
'item': []
}
# ~~~这里组装测试项里面的测试用例~~~
for case in demand.tcField.all():
step_list = []
index = 1
for one in case.step.all():
# 这里需要对operation富文本处理
rich_parser = RichParser(one.operation)
desc_list = rich_parser.get_final_list(doc, img_size=68)
rich_parser2 = RichParser(one.result)
res_list = rich_parser2.get_final_list(doc, img_size=75)
# 组装用例里面的步骤dict
passed = '通过'
if one.passed == '2':
passed = '未通过'
if one.passed == '3':
passed = '未执行'
step_dict = {
'index': index,
'operation': desc_list,
'expect': one.expect,
'result': res_list,
'passed': passed,
}
step_list.append(step_dict)
index += 1
# 查询所有的problem
problem_list = []
problem_prefix = "PT"
proj_ident = project_obj.ident
for problem in case.caseField.all():
problem_list.append("_".join([problem_prefix, proj_ident, problem.ident]))
# fpga的时序图
rich_parser3 = RichParser(case.timing_diagram)
timing_diagram = rich_parser3.get_final_list(doc, img_size=115, height=50)
has_timing_diagram = False
if len(timing_diagram) > 0:
if isinstance(timing_diagram[0], InlineImage):
has_timing_diagram = True
# 组装用例的dict
case_dict = {
'name': case.name,
'ident': get_case_ident(demand_ident, case),
'summary': case.summarize,
'initialization': case.initialization,
'premise': case.premise,
'design_person': case.designPerson,
'test_person': case.testPerson,
'monitor_person': case.monitorPerson,
'step': step_list,
'time': str(case.exe_time) if case.exe_time is not None else str(case.update_datetime),
'problems': "".join(problem_list),
'round_num_chn': cname,
# 2025年4月24日新增
'has_timing_diagram': has_timing_diagram,
'timing_diagram': timing_diagram,
}
demand_dict['item'].append(case_dict)
list_list[type_index].append(demand_dict)
# 定义渲染上下文
context = {}
output_list = []
for (index, li) in enumerate(list_list):
qs = Dict.objects.get(code="testType").dictItem.get(key=str(index + 1))
context_str = qs.title
sort = qs.sort
table = {
"type": context_str,
"item": li,
"sort": sort
}
output_list.append(table)
# 排序
output_list = sorted(output_list, key=(lambda x: x["sort"]))
context["data"] = output_list
# 最后渲染
save_path = Path.cwd() / 'media' / project_path_str / 'output_dir/hjl' / f"{cname}轮测试用例记录.docx"
doc.render(context, autoescape=True)
try:
doc.save(save_path)
except PermissionError:
return ChenResponse(code=400, status=400, message='您打开了生成的文档,请关闭后重试')
return ChenResponse(code=200, status=200, message='多轮回归测试用例记录生成完毕')

File diff suppressed because it is too large Load Diff

View File

@@ -1,155 +1,155 @@
# 导入内置模块
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))
# 导入内置模块
from pathlib import Path
from django.db.models import QuerySet
# 导入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, Case
# 导入工具
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: QuerySet[Case] = 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(set(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')
# 依据要求-获取其设计需求
print(case_design_list)
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))