from pathlib import Path # 导入框架东西 from ninja_extra import ControllerBase, api_controller, route # 框架组件 from django.db import transaction from django.shortcuts import get_object_or_404 from django.db.models import Q # 认证和权限 from ninja_extra.permissions import IsAuthenticated from ninja_jwt.authentication import JWTAuth # 导入模型 from apps.project.models import Project from apps.dict.models import Dict # 导入文档处理类 from docxtpl import DocxTemplate from docx import Document # 导入自己工具 from utils.chen_response import ChenResponse from utils.util import get_ident, get_case_ident, get_testType from utils.chapter_tools.csx_chapter import create_csx_chapter_dict from apps.createDocument.extensions import util from apps.createDocument.extensions.util import create_sm_docx from utils.path_utils import project_path from apps.createDocument.extensions.parse_rich_text import RichParser from apps.createDocument.extensions.documentTime import DocTime # @api_controller("/generateSM", tags=['生成说明文档系列'], auth=JWTAuth(), permissions=[IsAuthenticated]) @api_controller("/generateSM", tags=['生成说明文档系列']) class GenerateControllerSM(ControllerBase): @route.get("/create/techyiju", url_name="create-techyiju") @transaction.atomic def create_techyiju(self, id: int): project_obj = get_object_or_404(Project, id=id) duties_qs = project_obj.pdField.filter(Q(type='XQ') | Q(type='SJ') | Q(type='XY')) std_documents = [] for duty in duties_qs: one_duty = {'doc_name': duty.name, 'ident_version': duty.ref + '-' + duty.version, 'publish_date': duty.release_date, 'source': duty.release_union} std_documents.append(one_duty) # 添加大纲到这里 ## 判断是否为鉴定 doc_name = f'{project_obj.name}软件测评大纲' if project_obj.report_type == '9': doc_name = f'{project_obj.name}软件鉴定测评大纲' # 这里大纲版本升级如何处理 - TODO：1.大纲版本升级后版本处理 # 时间控制类 timer = DocTime(id) dg_duty = {'doc_name': doc_name, 'ident_version': f'PT-{project_obj.ident}-TO-1.00', 'publish_date': timer.dg_cover_time, 'source': project_obj.test_unit} std_documents.append(dg_duty) # 生成二级文档 context = { 'std_documents': std_documents } return create_sm_docx("技术依据文件.docx", context, id) @route.get('/create/caseList', url_name='create-caseList') @transaction.atomic def create_caseList(self, id: int): """创建第一轮文档""" project_path_str = project_path(id) # 生成测试用例需要doc对象 case_template_doc_path = Path.cwd() / 'media' / project_path_str / 'form_template/sm' / '测试用例.docx' doc = DocxTemplate(case_template_doc_path) project_obj = get_object_or_404(Project, id=id) # 先查询dict字典，查出总共有多少个testType test_type_len = Dict.objects.get(code='testType').dictItem.count() type_number_list = [i for i in range(1, test_type_len + 1)] # [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16] list_list = [[] for j in range(1, test_type_len + 1)] # [[], [], [], [], [], [], [], [], [], [], [], [], [], [], [], []] # 先找到第一轮次 project_round_one = project_obj.pField.filter(key=0).first() # 测试项的章节号预置处理 demand_prefix = '6.2' testType_list, last_chapter_items = create_csx_chapter_dict(project_round_one) testDemands = project_round_one.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=70) step_dict = { 'index': index, 'operation': desc_list, 'expect': one.expect, } step_list.append(step_dict) index += 1 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, 'step': step_list } 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 doc.render(context) try: doc.save(Path.cwd() / "media" / project_path_str / "output_dir/sm" / "测试用例.docx") return ChenResponse(status=200, code=200, message="文档生成成功！") except PermissionError as e: return ChenResponse(status=400, code=400, message="模版文件已打开，请关闭后再试，{0}".format(e)) @route.get('/create/caseBreifList', url_name='create-caseBreifList') @transaction.atomic def create_caseBreifList(self, id: int): # 生成第一轮的测试说明 project_obj = get_object_or_404(Project, id=id) # 先查询dict字典，查出总共有多少个testType test_type_len = Dict.objects.get(code='testType').dictItem.count() type_number_list = [i for i in range(1, test_type_len + 1)] # [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16] list_list = [[] for j in range(1, test_type_len + 1)] # [[], [], [], [], [], [], [], [], [], [], [], [], [], [], [], []] # 先找到第一轮次 project_round_one = project_obj.pField.filter(key=0).first() # 找到第一轮的全部测试项 testDemands = project_round_one.rtField.all() for demand in testDemands: type_index = type_number_list.index(int(demand.testType)) demand_ident = get_ident(demand) demand_dict = { 'name': demand.name, 'item': [] } for case in demand.tcField.all(): case_dict = { 'name': case.name, 'ident': get_case_ident(demand_ident, case), 'summary': case.summarize, } 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)) sort = qs.sort table = { "item": li, "sort": sort } output_list.append(table) output_list = sorted(output_list, key=(lambda x: x["sort"])) context["data"] = output_list return create_sm_docx("用例说明.docx", context, id) @route.get('/create/smtrack', url_name='create-smtrack') @transaction.atomic def create_smtrack(self, id: int): """生成说明的需求追踪表""" project_path_str = project_path(id) project_obj = get_object_or_404(Project, id=id) demand_prefix = '6.2' design_list = [] project_round_one = project_obj.pField.filter(key='0').first() if project_round_one: testType_list, last_chapter_items = create_csx_chapter_dict(project_round_one) # 找出第一轮被测件为'SO'的 so_dut = project_round_one.rdField.filter(type='SO').first() if so_dut: so_designs = so_dut.rsField.all() for design in so_designs: design_dict = {'name': design.name, 'chapter': design.chapter, 'test_demand': []} test_items = [] test_items.extend(design.dtField.all()) test_items.extend(design.odField.all()) for test_item in test_items: # 对4个测试类型单独处理：因为这4类肯定没有章节号 if test_item.testType in ['2', '3', '15', '8']: design_dict.update({'name': "/", 'chapter': "/"}) reveal_ident = "_".join( ["XQ", get_testType(test_item.testType, "testType"), test_item.ident]) # 查字典方式确认章节号最后一位 test_item_last_chapter = last_chapter_items[test_item.testType].index(test_item.key) + 1 test_chapter = ".".join([demand_prefix, str(testType_list.index(test_item.testType) + 1), str(test_item_last_chapter)]) test_item_dict = {'name': test_item.name, 'chapter': test_chapter, 'ident': reveal_ident, 'case_list': []} for case in test_item.tcField.all(): case_dict = { 'name': case.name, 'ident': get_case_ident(reveal_ident, case) } test_item_dict['case_list'].append(case_dict) design_dict['test_demand'].append(test_item_dict) design_list.append(design_dict) # 上面找出了源代码被测件，这里找XQ被测件 xq_dut = project_round_one.rdField.filter(type='XQ').first() if xq_dut: xq_designs = xq_dut.rsField.all() for design in xq_designs: design_dict = {'name': design.name, 'chapter': design.chapter, 'test_demand': []} # 获取一个design的所有测试项 # 注意：这里有关联测试项!!!需要多对多关系拼接 test_items = [] test_items.extend(design.dtField.all()) test_items.extend(design.odField.all()) for test_item in test_items: reveal_ident = "_".join( ["XQ", get_testType(test_item.testType, "testType"), test_item.ident]) # 查字典方式确认章节号最后一位 test_item_last_chapter = last_chapter_items[test_item.testType].index(test_item.key) + 1 test_chapter = ".".join([demand_prefix, str(testType_list.index(test_item.testType) + 1), str(test_item_last_chapter)]) test_item_dict = {'name': test_item.name, 'chapter': test_chapter, 'ident': reveal_ident, 'case_list': []} for case in test_item.tcField.all(): case_dict = { 'name': case.name, 'ident': get_case_ident(reveal_ident, case) } test_item_dict['case_list'].append(case_dict) design_dict['test_demand'].append(test_item_dict) design_list.append(design_dict) context = { 'design_list': design_list, } # 手动渲染tpl生成文档 input_file = Path.cwd() / 'media' / project_path_str / 'form_template' / 'sm' / '说明追踪.docx' temporary_file = Path.cwd() / 'media' / project_path_str / 'form_template' / 'sm' / 'temporary' / '说明追踪_temp.docx' out_put_file = Path.cwd() / 'media' / project_path_str / 'output_dir' / 'sm' / '说明追踪.docx' doc = DocxTemplate(input_file) doc.render(context) doc.save(temporary_file) # 通过docx合并单元格 if temporary_file.is_file(): try: docu = Document(temporary_file) # 找到其中的表格 util.merge_all_cell(docu.tables[0]) # 储存到合适位置 docu.save(out_put_file) return ChenResponse(code=200, status=200, message='文档生成成功...') except PermissionError as e: return ChenResponse(code=400, status=400, message='请检查文件是否打开，如果打开则关闭...') else: return ChenResponse(code=400, status=400, message='中间文档未找到，请检查你模版是否存在...')