PyQt5+Python-docx实现一键生成测试报告

pip install PyQt5 python-docx

import sys
from PyQt5.QtWidgets import QApplication, QWidget, QLabel, QLineEdit, QDateEdit, QPushButton, QVBoxLayout, QHBoxLayout, QFormLayout

class TestReportTool(QWidget):
    def __init__(self):
        super().__init__()
        self.setWindowTitle("测试报告一键生成工具")
        self.init_ui()

    def init_ui(self):
        # 表单布局
        form_layout = QFormLayout()

        # 输入字段（积木模块）
        self.system_url = QLineEdit()
        self.pm_name = QLineEdit()
        self.tm_name = QLineEdit()
        self.tl_name = QLineEdit()
        self.tester_name = QLineEdit()
        self.start_date = QDateEdit()
        self.end_date = QDateEdit()
        self.func_cases = QLineEdit()
        self.func_bugs = QLineEdit()
        self.regression_cases = QLineEdit()
        self.regression_bugs = QLineEdit()

        # 添加字段到表单
        form_layout.addRow("系统测试环境地址:", self.system_url)
        form_layout.addRow("项目经理:", self.pm_name)
        form_layout.addRow("测试经理:", self.tm_name)
        form_layout.addRow("测试组长:", self.tl_name)
        form_layout.addRow("测试人员:", self.tester_name)
        form_layout.addRow("测试起始时间:", self.start_date)
        form_layout.addRow("测试结束时间:", self.end_date)
        form_layout.addRow("功能测试用例数:", self.func_cases)
        form_layout.addRow("功能测试Bug数:", self.func_bugs)
        form_layout.addRow("回归测试用例数:", self.regression_cases)
        form_layout.addRow("回归测试缺陷数:", self.regression_bugs)

        # 生成按钮
        self.generate_btn = QPushButton("生成测试报告")
        self.generate_btn.clicked.connect(self.generate_report)

        # 主布局
        main_layout = QVBoxLayout()
        main_layout.addLayout(form_layout)
        main_layout.addWidget(self.generate_btn)
        self.setLayout(main_layout)

    def generate_report(self):
        # 收集数据
        data = {
            "system_url": self.system_url.text(),
            "pm_name": self.pm_name.text(),
            "tm_name": self.tm_name.text(),
            "tl_name": self.tl_name.text(),
            "tester_name": self.testers_name.text(),
            "start_date": self.start_date.date().toString("yyyy-MM-dd"),
            "end_date": self.end_date.date().toString("yyyy-MM-dd"),
            "func_cases": self.func_cases.text(),
            "func_bugs": self.func_bugs.text(),
            "regression_cases": self.regression_cases.text(),
            "regression_bugs": self.regression_bugs.text(),
        }

        # 计算时间块并添加到data
        time_blocks = calculate_time_blocks(data["start_date"], data["end_date"])
        for phase, dates in time_blocks.items():
            key = f"{phase}_time"
            data[key] = dates

        # 计算通过率
        func_pass_rate = round((1 - int(data["func_bugs"]) / int(data["func_cases"])) * 100, 2)
        data["func_pass_rate"] = f"{func_pass_rate}%"
        reg_pass_rate = round((1 - int(data["regression_bugs"]) / int(data["regression_cases"])) * 100, 2)
        data["reg_pass_rate"] = f"{reg_pass_rate}%"

        # 调用文档生成函数
        from docx_generator import generate_docx
        generate_docx(data)
        print("测试报告生成成功！")

if __name__ == "__main__":
    app = QApplication(sys.argv)
    window = TestReportTool()
    window.show()
    sys.exit(app.exec_())

from docx import Document
from datetime import datetime, timedelta

def calculate_time_blocks(start_date_str, end_date_str):
    start_date = datetime.strptime(start_date_str, "%Y-%m-%d")
    end_date = datetime.strptime(end_date_str, "%Y-%m-%d")
    total_days = (end_date - start_date).days
    if total_days <= 0:
        return {}
    
    # 按比例划分阶段（时间蛋糕）
    phases = {
        "需求评审": 0.1,
        "用例设计": 0.2,
        "用例执行": 0.3,
        "回归测试": 0.2,
        "报告编写": 0.2
    }
    time_blocks = {}
    current_date = start_date
    for phase, ratio in phases.items():
        days = int(total_days * ratio)
        end_phase = current_date + timedelta(days=days)
        time_blocks[phase] = f"{current_date.strftime('%Y-%m-%d')} - {end_phase.strftime('%Y-%m-%d')}"
        current_date = end_phase
    return time_blocks

def generate_docx(data):
    # 读取模板文档
    doc = Document('template.docx')  # 假设模板文件为 template.docx

    # 替换段落中的占位符
    for paragraph in doc.paragraphs:
        for key, value in data.items():
            if key in paragraph.text:
                paragraph.text = paragraph.text.replace(f'{{{key}}}', value)  # 使用双大括号包裹

    # 替换表格中的占位符
    for table in doc.tables:
        for row in table.rows:
            for cell in row.cells:
                for key, value in data.items():
                    if key in cell.text:
                        cell.text = cell.text.replace(f'{{{key}}}', value)

    # 保存文档
    doc.save("TestReport.docx")