initial commit

apps/createDocument/controllers/jl.py

@@ -0,0 +1,154 @@
+# 内置模块导入
+from pathlib import Path
+# 导入框架相关
+from django.db import transaction
+from django.shortcuts import get_object_or_404
+from ninja_extra import api_controller, ControllerBase, route
+# 导入权限相关
+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 InlineImage
+# 导入文档处理相关
+from docxtpl import DocxTemplate
+# 导入自己工具
+from utils.chapter_tools.csx_chapter import create_csx_chapter_dict
+from utils.util import get_ident, get_case_ident, get_testType
+from utils.chen_response import ChenResponse
+from utils.path_utils import project_path
+from apps.createDocument.extensions.parse_rich_text import RichParser
+
+# @api_controller("/generateJL", tags=['生成测试记录系列'], auth=JWTAuth(), permissions=[IsAuthenticated])
+@api_controller("/generateJL", tags=['生成测试记录系列'])
+class GenerateControllerJL(ControllerBase):
+    @route.get("/create/caserecord", url_name="create-caserecord")
+    @transaction.atomic
+    def create_caserecord(self, id: int):
+        """生成测试记录表格"""
+        project_path_str = project_path(id)
+        project_obj = get_object_or_404(Project, id=id)
+        # 测试用例记录模版位置
+        record_template_path = Path.cwd() / 'media' / project_path_str / 'form_template/jl' / '测试用例记录.docx'
+        # 打开模版文档
+        doc = DocxTemplate(record_template_path)
+        # 准备返回的测试类型数组的嵌套
+        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)]  # 每个测试类型组合为一个列表[[],[],[],[]]
+        # 查询出第一轮次
+        round_one = project_obj.pField.filter(key=0).first()
+        # 测试项的章节号预置处理 - 根据轮次生成list和dict的二维数据
+        demand_prefix = '6.2'
+        testType_list, last_chapter_items = create_csx_chapter_dict(round_one)
+        # 找出所有测试项
+        testDemands = 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=68)
+                    # 这里需要对result富文本处理
+                    rich_parser2 = RichParser(one.result)
+                    res_list = rich_parser2.get_final_list(doc, img_size=75)
+                    # 组装用例里面的步骤dict
+                    passed = '通过'
+                    if one.passed == '2':
+                        passed = '未通过'
+                    elif one.passed == '3':
+                        passed = '未执行'
+                    step_dict = {
+                        'index': index,
+                        'operation': desc_list,
+                        'expect': one.expect,
+                        'result': res_list,
+                        'passed': passed,
+                    }
+                    index += 1
+                    step_list.append(step_dict)
+                # 这里判断里面的单个步骤的执行情况，来输出一个整个用例的执行情况
+                exe_noncount = 0
+                execution_str = '已执行'
+                for ste in step_list:
+                    if ste.get('execution') == '3':
+                        exe_noncount += 1
+                if exe_noncount > 0 and exe_noncount != len(step_list):
+                    execution_str = '部分执行'
+                elif exe_noncount == len(step_list):
+                    execution_str = '未执行'
+                else:
+                    execution_str = '已执行'
+                # 查询所有的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]))
+                # 组装用例的dict
+                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
+                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,
+                    'execution': execution_str,
+                    'time': str(case.exe_time) if case.exe_time is not None else str(case.update_datetime),
+                    'problems': "、".join(problem_list),
+                    # 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))  # type:ignore
+            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/jl" / "测试用例记录.docx")
+            return ChenResponse(status=200, code=200, message="文档生成成功！")
+        except PermissionError as e:
+            return ChenResponse(status=400, code=400, message="模版文件已打开，请关闭后再试，{0}".format(e))