CTToolsOld/need/new_JtoS.py

from PyQt5 import QtCore
from PyQt5.QtCore import pyqtSignal
from pathlib import *  # noqa: F403
from PyQt5.QtWidgets import QMessageBox
from docxtpl import DocxTemplate, InlineImage  # type:ignore
from docx import Document
from docx.table import Table
from docx.text.paragraph import Paragraph
import io


class create_new_JtoS(QtCore.QThread):
    sin_out = pyqtSignal(str)
    PREFIX = r'{http://schemas.openxmlformats.org/wordprocessingml/2006/main}'

    def __init__(self, parent):
        super().__init__()
        self.parent = parent

    def run(self):
        self.sin_out.emit("进入CPU测试记录转说明线程......")
        self.sin_out.emit("开始转换......")
        # 如果没有选择文件路径则退出
        if not self.parent.open_file_name:
            self.sin_out.emit('nofile')
            self.parent.tabWidget.setEnabled(True)
            return

        # 打开模板文件进行渲染，然后就是用docxtpl生成用例
        try:
            tpl_path = Path.cwd(  # noqa: F405
            ) / "need" / "document_templates" / "cpu新记录to说明模版.docx"
            self.sin_out.emit('导入模板文件路径为：' + str(tpl_path))
            tpl = DocxTemplate(tpl_path)  # 模板导入成功

        except:  # noqa: E722
            QMessageBox.warning(self.parent, "出错了", "导入模板出错请检查模板文件是否存在或名字不正确")
            return

        try:
            doc = Document(self.parent.open_file_name[0])
            self.sin_out.emit('已识别到CPU测试记录文件...')
        except:  # noqa: E722
            self.sin_out.emit('open failed:选择的文档')
            self.parent.tabWidget.setEnabled(True)
            return

        self.sin_out.emit('复制测试说明文档模板到本程序所在目录...')
        curpath = Path.cwd() / 'need'  # noqa: F405
        shuoming_path_tmp = curpath / 'document_templates' / 'cpu新记录to说明模版.docx'
        print(shuoming_path_tmp)
        if shuoming_path_tmp.is_file():
            self.sin_out.emit('已检测到有说明模板文件...')
        else:
            self.sin_out.emit('open failed:选择的文档')
            return

        # 获取表格数量
        tables = doc.tables
        tb_count = len(tables)
        self.sin_out.emit('total:' + str(tb_count))
        # 创建一个字典来储存单个用例
        data_list = []
        table_index = 1
        # 获取表格数量
        for ele in doc._element.body.iter():
            data = {'type': ''}
            if ele.tag.endswith('}p'):
                for child in ele.iter():
                    if child.tag.endswith('pStyle'):
                        rank = child.get(f"{self.PREFIX}val")
                        t_ele = ele.findall(
                            f".//{self.PREFIX}t")  # type:ignore
                        title = ''
                        for i in range(len(t_ele)):
                            if not t_ele[i].text.isdigit():
                                title = title + t_ele[i].text
                        data['type'] = rank
                        data['title'] = title
                        data_list.append(data)

            elif ele.tag.endswith('}tbl'):
                data = {
                    'name': '',
                    'biaoshi': '',
                    'zhuizong': [],
                    'zongsu': '',
                    'init': '',
                    'qianti': '',
                    'step': []
                }
                data['type'] = 'table'
                table = Table(ele, doc)
                if table.cell(1, 0).text == '测试用例名称':
                    self.sin_out.emit(str(table_index))

                    try:
                        self.sin_out.emit(str(table_index))
                        self.sin_out.emit(f'正在处理第{table_index}个表格')
                        # 1、获取测试用例名称
                        data['name'] = table.cell(1, 3).text
                        # 2、获取用例标识
                        data['biaoshi'] = table.cell(1, 9).text
                        # 3、获取追踪关系 注意word中换行为\r\x07
                        temp = table.cell(2, 3)
                        for tem in temp.paragraphs:
                            data['zhuizong'].append(tem.text)
                        # 4、获取综述
                        data['zongsu'] = table.cell(3, 3).text.replace(
                            '\n', '\a')
                        # 5、初始化
                        data['init'] = table.cell(4,
                                                  3).text.replace('\n', '\a')
                        # 6、获取前提与约束
                        data['qianti'] = table.cell(5, 3).text.replace(
                            '\n', '\a')
                        # 7、获取步骤信息-总行数减去12为步骤行数
                        row_count = len(table.rows)
                        step_count = row_count - 12
                        for j in range(step_count):
                            buzhou = {
                                'shuru': '',
                                'yuqi': '',
                                'num': '',
                                'image': '',
                                'is_image': '0'
                            }
                            buzhou['num'] = table.rows[8 + j].cells[0].text
                            if buzhou['num'] == '':
                                buzhou['num'] = j + 1
                            buzhou['shuru'] = table.rows[
                                8 + j].cells[2].text.replace('\n', '\a')
                            cel = table.rows[8 + j].cells[2]
                            if len(
                                    cel._element.findall(
                                        './/{http://schemas.openxmlformats.org/wordprocessingml/2006/main}drawing'
                                    )) > 0:
                                img_ele = cel._element.xpath('.//pic:pic')[0]
                                embed = img_ele.xpath('.//a:blip/@r:embed')[0]
                                related_part = doc.part.related_parts[embed]
                                image = related_part.image
                                # blob属性就是二进制图片属性
                                image_bytes = image.blob
                                buzhou['image'] = InlineImage(
                                    tpl, io.BytesIO(image_bytes))
                                buzhou['is_image'] = '1'
                            buzhou['yuqi'] = table.rows[
                                8 + j].cells[4].text.replace('\n', '\a')
                            data['step'].append(buzhou)
                        # 8、最后加入data_list
                        data_list.append(data)
                        table_index += 1
                    except:  # noqa: E722
                        self.sin_out.emit(f'第{table_index}个表格处理错误！')
                        table_index += 1
                        pass
        # 开始渲染模板文件
        try:
            self.sin_out.emit('注意：如果生成文档没有标题，请检查您文档章节号格式是否正确!!!')
            self.sin_out.emit('all_doned:')
            context = {
                "tables": data_list,
                "renyuan": self.parent.lineEdit_17.text(),
            }
            tpl.render(context)
            tpl.save("CPU新版反向生成说明.docx")
            self.sin_out.emit('stopsuccess')
        except:  # noqa: E722
            self.sin_out.emit('stoperror')
            return