321

cdTMP/src/views/testmanage/projmanage/cpns/progress.vue

@@ -8,7 +8,7 @@
             :on-before-ok="handleClickConfirmButton"
             @cancel="handleClickConfirmButton"
         >
-            <template #title>生成大纲进度</template>
+            <template #title>正在生成{{ text }}</template>
             <div><a-progress :percent="percent" size="large" /></div>
         </a-modal>
     </div>
@@ -26,6 +26,10 @@ const props = defineProps({
     isComplete: {
         type: Boolean,
         default: false
+    },
+    text: {
+        type: String,
+        default: "测评大纲"
     }
 })
 const percent = ref(0)

cdTMP/src/views/testmanage/projmanage/index.vue

@@ -4,17 +4,32 @@
             <!-- ma-crud组件 -->
             <ma-crud :options="crudOptions" :columns="crudColumns" ref="crudRef">
                 <template #operationBeforeExtend="{ record }">
+                    <a-popover title="文档生成组合按钮" trigger="click">
+                        <a-button type="primary" size="mini">
+                            <template #icon>
+                                <icon-plus />
+                            </template>
+                            文档生成
+                        </a-button>
+                        <template #content>
+                            <p><a-link @click="createDgItem(record)">大纲二段文档</a-link></p>
+                            <p><a-link @click="createSmItem(record)">说明二段文档</a-link></p>
+                            <p><a-link @click="createSeitaiDagang(record)"><icon-eye />[测试]生成最后大纲</a-link></p>
+                            <p><a-link @click="createSeitaiShuoming(record)"><icon-eye />[测试]生成最后说明</a-link></p>
+                        </template>
+                    </a-popover>
                     <a-link @click="enterWorkPlant(record)">进入工作区</a-link>
                     <a-link @click="previewRef.open(record, crudColumns)"><icon-eye />预览</a-link>
-                    <a-link @click="createDgItem(record)">大纲二段文档</a-link>
-                    <a-link @click="createSmItem(record)">说明二段文档</a-link>
-                    <a-link @click="createSeitaiDagang(record)"><icon-eye />[测试]生成最后大纲</a-link>
-                    <a-link @click="createSeitaiShuoming(record)"><icon-eye />[测试]生成最后说明</a-link>
                 </template>
             </ma-crud>
         </div>
         <preview ref="previewRef"></preview>
-        <Progress :visible="visible" :isComplete="isComplete" @clickConfirm="handleModalConfirmClick"></Progress>
+        <Progress
+            :visible="visible"
+            :isComplete="isComplete"
+            :text="ptext"
+            @clickConfirm="handleModalConfirmClick"
+        ></Progress>
     </div>
 </template>
 <script lang="jsx" setup>
@@ -40,13 +55,20 @@ const enterWorkPlant = function (record) {
 // 这里放弹出进度条组件变量
 const visible = ref(false)
 const isComplete = ref(false)
+const ptext = ref("测评大纲")
 const handleModalConfirmClick = () => {
     visible.value = false
 }
 // ~~~~~~~~测试说明生成文档~~~~~~~~
 const createSeitaiShuoming = async (record) => {
-    const st = await seitaiGenerateApi.createShuomingSeiTai({ id: record.id })
-
+    ptext.value = "测试说明"
+    visible.value = true
+    isComplete.value = false
+    const st = await seitaiGenerateApi.createShuomingSeiTai({ id: record.id }).catch((err) => {
+        isComplete.value = true
+        visible.value = false
+    })
+    isComplete.value = true
     Message.success(st.message)
 }
 
@@ -54,6 +76,7 @@ const createSeitaiShuoming = async (record) => {
 const createSeitaiDagang = async (record) => {
     // 根据一系列文档生成大纲 - 这里有进度条组件、a-modal组件
     // 1.打开进度条组件
+    ptext.value = "测评大纲"
     visible.value = true
     isComplete.value = false
     const st = await seitaiGenerateApi.createDagangSeiTai({ id: record.id }).catch((err) => {
@@ -88,7 +111,6 @@ const createSmItem = async (record) => {
     const st9 = await smGenerateApi.createSMCaseBreifList({ id: record.id })
     // 生成说明追踪
     const st10 = await smGenerateApi.createSMTrack({ id: record.id })
-
     Message.success(st10.message)
 }
 // 大纲生成二级文档

cdTMP/temp.py

@@ -1,154 +0,0 @@
-from ninja_extra import api_controller, route, ControllerBase
-from django.db import transaction
-from pathlib import Path
-from django.shortcuts import get_object_or_404
-# 文档处理相关库
-from docxtpl import DocxTemplate
-from docx import Document
-from docx.text.paragraph import Paragraph
-from docx.table import Table
-from docx.oxml.table import CT_Tbl
-from docx.oxml.text.paragraph import CT_P
-from docx.oxml.text.run import CT_R
-from docx.oxml.shape import CT_Picture
-from docx.parts.image import ImagePart
-from docx.text.run import Run
-from docx.shared import Cm
-from docx.enum.text import WD_PARAGRAPH_ALIGNMENT
-
-@route.get('/smDocument', url_name='create-smDocument')
-@transaction.atomic
-def create_smDocument(self, id: int):
-    """生成最后说明文档"""
-    # 获取项目对象
-    project_obj = get_object_or_404(Project, id=id)
-    # 首先第二层模版所需变量
-    member = project_obj.member[0] if len(project_obj.member) > 0 else project_obj.duty_person
-    context = {'name': project_obj.name, 'is_JD': False, 'ident': project_obj.ident, 'sec_title': "公开",
-                'duty_person': project_obj.duty_person, 'member': member}
-    if project_obj.report_type == '9':
-        context['is_JD'] = True
-    # 提取第一轮测试中源代码 - 用户标识
-    round_1 = project_obj.pField.filter(key='0').first()
-    duty_so = round_1.rdField.filter(type='SO').first()
-    if not duty_so:
-        return ChenResponse(code=400, status=400, message="未找到第一轮测试中源代码被测件请添加")
-    context['user_ident'] = duty_so.ref
-
-    # ~~~~~~定义模版路径~~~~~~~
-    ## 定义说明最后模版
-    sm_template_file = Path.cwd() / 'media' / 'form_template' / 'products' / '测试说明.docx'
-    # docx生成后给docxtpl文件路径
-    sm_to_tpl_file = Path.cwd() / 'media' / 'temp' / '测试说明.docx'
-    # 生成最后文档路径
-    sm_seitai_final_file = Path.cwd() / 'media' / 'final_seitai' / '测试说明.docx'
-    # output_dir根路径
-    output_files_path = Path.cwd() / 'media' / 'output_dir'
-    # 获取可能使用的被复制的docx文件
-    dg_copied_files = []
-    sm_copied_files = []
-    for file in output_files_path.iterdir():
-        if file.is_file():
-            if file.suffix == '.docx':
-                dg_copied_files.append(file)
-        elif file.is_dir():
-            if file.stem == 'sm':
-                for f in file.iterdir():
-                    if f.suffix == '.docx':
-                        sm_copied_files.append(f)
-    # 打开文档找到所有的“域”
-    doc = Document(sm_template_file.as_posix())
-    body = doc.element.body
-    sdt_element_list = body.xpath('./w:sdt')
-    # 找到sdt域的名称 -> 为了对应output_dir文件 / 储存所有output_dir图片
-    area_name_list = []
-    image_part_list = []
-    # 遍历所有控件 -> 放入area_name_list【这里准备提取公共代码】
-    for sdt_ele in sdt_element_list:
-        for elem in sdt_ele.iterchildren():
-            # 获取“域”的名称
-            if elem.tag.endswith('sdtPr'):
-                for el in elem.getchildren():
-                    if el.tag.endswith('alias'):
-                        if len(el.attrib.values()) > 0:
-                            area_name = el.attrib.values()[0]
-                            area_name_list.append(area_name)
-            # 开始替换里面的“域”
-            if elem.tag.endswith('sdtContent'):
-                elem.clear()
-                if len(area_name_list) > 0:
-                    area_pop_name = area_name_list.pop(0)
-                    # 取到“域名称”，这里先去找media/output_dir/sm下文件，然后找media/output下文件
-                    copied_file_path = ""
-                    for file in sm_copied_files:
-                        if file.stem == area_pop_name:
-                            copied_file_path = file
-                    # 这里判断是否copied_file_path没取到文件，然后遍历output_dir下文件
-                    if not copied_file_path:
-                        for file in dg_copied_files:
-                            if file.stem == area_pop_name:
-                                copied_file_path = file
-                    # 找到所需的文件，将其数据复制到对应area_name的“域”
-                    if copied_file_path:
-                        doc_copied = Document(copied_file_path)
-                        copied_element_list = []
-                        element_list = doc_copied.element.body.inner_content_elements
-                        for elet in element_list:
-                            if isinstance(elet, CT_P):
-                                copied_element_list.append(Paragraph(elet, doc_copied))
-                            if isinstance(elet, CT_Tbl):
-                                copied_element_list.append(Table(elet, doc_copied))
-                        for para_copied in copied_element_list:
-                            elem.append(para_copied._element)
-
-                        # 下面代码就是将图片全部提取到image_part_list，以便后续插入
-                        doc_copied = Document(copied_file_path)  # 需要重新获取否则namespace错误
-                        copied_body = doc_copied.element.body
-                        img_node_list = copied_body.xpath('.//pic:pic')
-                        if not img_node_list:
-                            pass
-                        else:
-                            for img_node in img_node_list:
-                                img: CT_Picture = img_node
-                                # 根据节点找到图片的关联id
-                                embed = img.xpath('.//a:blip/@r:embed')[0]
-                                # 这里得到ImagePart -> 马上要给新文档添加
-                                related_part: ImagePart = doc_copied.part.related_parts[embed]
-                                # doc_copied.part.related_parts是一个字典
-                                image_part_list.append(related_part)
-
-    # 新文档添加sdt_element_list的图片
-    graph_node_list = body.xpath('.//pic:pic')
-    img_count = 0
-    if len(graph_node_list) == len(image_part_list):
-        for graph_node in graph_node_list:
-            image_run_node = getParentRunNode(graph_node)
-            image_run_node.clear()
-            copied_bytes_io = BytesIO(image_part_list[img_count].image.blob)
-            r_element = Run(image_run_node, doc)
-            inline_shape = r_element.add_picture(copied_bytes_io)
-            # 设置图片位置尺寸
-            source_width = inline_shape.width
-            inline_shape.width = Cm(12)
-            inline_shape.height = int(inline_shape.height * (inline_shape.width / source_width))
-            # 设置图片所在段落居中对齐
-            r_element.alignment = WD_PARAGRAPH_ALIGNMENT.CENTER
-            img_count += 1
-    else:
-        return ChenResponse(code=400, status=400, message='注意模版里面有自定义图片，请删除后重试!!!')
-
-    try:
-        doc.save(sm_to_tpl_file)
-    except PermissionError as e:
-        return ChenResponse(status=400, code=400, message="注意你打开了生成的文档，请关闭后再试，{0}".format(e))
-
-    doc = DocxTemplate(sm_to_tpl_file)
-    start = time.time()
-    doc.render(context)  # 耗时最长，TODO:异步任务处理？或前端等待？
-    end = time.time()
-    print('渲染耗时：', end - start)
-    try:
-        doc.save(sm_seitai_final_file)
-        return ChenResponse(status=200, code=200, message="最终大纲生成成功！")
-    except PermissionError as e:
-        return ChenResponse(status=400, code=400, message="模版文件已打开，请关闭后再试，{0}".format(e))