# 内置模块导入
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))