用Python打造全能测试报告工具,源码直接用!

上两篇文章《提升测试幸福感：从0到1实现测试报告自动化》《终于有人把测试报告自动化源码，讲得这么干净透彻！》梳理了工具的核心逻辑和设计思路，这一部分将深入源码层面，从「图标生成函数」「报告导出函数」「辅助工具函数」与「源码设计亮点与复用建议」做深度拆解，

def generate_charts(self):    # 校验：必须先生成基础数据    ifnot self.module_list:        self._show_msg("提示", "请先生成基础数据！")        return    # 1. 计算衍生数据（调用辅助计算函数）    module_bug_ratio = self._calc_module_bug_ratio()  # 模块缺陷占比    bug_level_dist = self._calc_bug_level_dist()      # 缺陷等级分布    bug_type_dist = self._calc_bug_type_dist()        # 缺陷类型分布    # 2. 设置图表样式：商务风配色，统一字体    plt.rcParams['font.sans-serif'] = ['SimHei']  # 解决中文显示问题    plt.rcParams['axes.unicode_minus'] = False    # 解决负号显示问题    # 3. 生成模块缺陷分布柱形图    fig, ((ax1, ax2), (ax3, _)) = plt.subplots(2, 2, figsize=(12, 10))  # 2行2列布局，第4个位置空着    # 柱形图：x轴=模块名，y轴=BUG数，系统模块浅灰，业务模块橙色    colors = ['#d3d3d3'if self._is_system_module(module) else'#ff7f50'formodule in self.module_list]    ax1.bar(range(len(self.module_list)), [self.bug_num_dict[module] formodule in self.module_list], color=colors)    ax1.set_title(f"{self.system_name} 模块缺陷分布", fontsize=12, fontweight='bold')    ax1.set_xlabel("测试模块")    ax1.set_ylabel("BUG数")    ax1.set_xticks(range(len(self.module_list)))    ax1.set_xticklabels(self.module_list, rotation=45, ha='right')  # 模块名旋转45度，避免重叠    # 给柱形图添加数值标签    for i, v in enumerate([self.bug_num_dict[module] formodule in self.module_list]):        ax1.text(i, v+0.1, str(v), ha='center', va='bottom', fontsize=9)    # 4. 生成缺陷等级分布饼图    levels = list(bug_level_dist.keys())    level_nums = list(bug_level_dist.values())    colors2 = ['#4169e1', '#ff6347', '#32cd32']  # 深蓝、红色、绿色（高、中、低等级）    ax2.pie(level_nums, labels=levels, colors=colors2, autopct='%1.1f%%', startangle=90)    ax2.set_title(f"{self.system_name} 缺陷等级分布", fontsize=12, fontweight='bold')    # 5. 生成缺陷类型分布饼图    types = list(bug_type_dist.keys())    type_nums = list(bug_type_dist.values())    colors3 = ['#ff6347', '#ffd700', '#9370db', '#20b2aa', '#d3d3d3']  # 红、黄、紫、青、灰    ax3.pie(type_nums, labels=types, colors=colors3, autopct='%1.1f%%', startangle=90)    ax3.set_title(f"{self.system_name} 缺陷类型分布", fontsize=12, fontweight='bold')    # 6. 调整子图间距，保存图表    plt.tight_layout()    self.chart_path = f"{self.system_name}_测试图表.png"  # 图表保存路径    plt.savefig(self.chart_path, dpi=150, bbox_inches='tight')  # dpi=150，保证清晰度    plt.close()  # 关闭图表，避免内存占用    self._show_msg("成功", f"图表生成完成！已保存为：{self.chart_path}")

解析：

●中文适配：通过 plt.rcParams 设置中文字体，解决matplotlib默认不显示中文的问题，这是可视化图表的基础；

●样式统一：采用商务风配色（浅灰、橙色、深蓝等），符合正式报告的视觉需求，避免花哨配色；

●布局合理：2行2列布局，预留一个空白位置，避免图表拥挤；柱形图添加数值标签，饼图添加百分比，直观清晰；

●模块区分：柱形图通过颜色区分系统模块和业务模块，让图表更有辨识度，贴合真实测试场景的分析需求。

def download_wps_doc(self):    # 校验：必须先生成基础数据和图表    ifnot self.module_list:        self._show_msg("提示", "请先生成基础数据！")        return    ifnothasattr(self, 'chart_path'):        self._show_msg("提示", "请先生成可视化图表！")        return    # 1. 选择保存路径    save_path = filedialog.asksaveasfilename(        defaultextension=".docx",        filetypes=[("Word 文档", "*.docx"), ("所有文件", "*.*")],        initialfile=f"{self.system_name}_测试报告"    )    ifnot save_path:  # 用户取消选择        return    # 2. 创建Word文档    doc = Document()    # 3. 添加报告标题    title = doc.add_heading(f"{self.system_name} 测试报告", 0)  # 0级标题，最大    title.alignment = WD_ALIGN_PARAGRAPH.CENTER  # 居中对齐    # 4. 添加测试用例执行表    doc.add_heading("一、测试用例执行统计", level=1)    case_table = doc.add_table(rows=1, cols=3)  # 1行3列（表头）    case_table.style = 'Table Grid'  # 表格样式    self._set_table_style(case_table)  # 应用自定义表格样式（边框、对齐等）    # 表头    hdr_cells = case_table.rows[0].cells    self._add_text_to_cell(hdr_cells[0], "测试模块", is_bold=True)    self._add_text_to_cell(hdr_cells[1], "用例总数", is_bold=True)    self._add_text_to_cell(hdr_cells[2], "执行情况", is_bold=True)    # 填充数据    formodule in self.module_list:        row_cells = case_table.add_row().cells        self._add_text_to_cell(row_cells[0], module)        self._add_text_to_cell(row_cells[1], str(self.case_num_dict[module]))        self._add_text_to_cell(row_cells[2], "全部通过")    # 添加总计行    total_row = case_table.add_row().cells    self._add_text_to_cell(total_row[0], "总计", is_bold=True)    self._add_text_to_cell(total_row[1], str(self.total_case), is_bold=True)    self._add_text_to_cell(total_row[2], "全部通过", is_bold=True)    # 5. 添加缺陷汇总表（类似用例表，省略重复代码，核心逻辑一致）    doc.add_heading("二、缺陷汇总统计", level=1)    bug_summary_table = doc.add_table(rows=1, cols=3)    bug_summary_table.style = 'Table Grid'    self._set_table_style(bug_summary_table)    # ... 缺陷汇总表填充逻辑（省略，与用例表一致）    # 6. 添加模块缺陷分布表+图表    doc.add_heading("三、模块缺陷分布分析", level=1)    # 先添加分布表，再添加图表    module_bug_table = doc.add_table(rows=1, cols=3)    module_bug_table.style = 'Table Grid'    self._set_table_style(module_bug_table)    hdr_cells = module_bug_table.rows[0].cells    self._add_text_to_cell(hdr_cells[0], "测试模块", is_bold=True)    self._add_text_to_cell(hdr_cells[1], "BUG数", is_bold=True)    self._add_text_to_cell(hdr_cells[2], "占比（%）", is_bold=True)    # 填充模块缺陷数据（调用衍生数据计算函数）    module_bug_ratio = self._calc_module_bug_ratio()    formodule in self.module_list:        row_cells = module_bug_table.add_row().cells        self._add_text_to_cell(row_cells[0], module)        self._add_text_to_cell(row_cells[1], str(self.bug_num_dict[module]))        self._add_text_to_cell(row_cells[2], f"{module_bug_ratio[module]:.2f}")    # 添加图表：插入本地生成的图表图片    doc.add_paragraph("模块缺陷分布图表：")    doc.add_picture(self.chart_path, width=Inches(6))  # 固定宽度6英寸，避免过大    # 7. 继续添加缺陷等级分布表、缺陷类型分布表（逻辑同上，省略重复代码）    # ... 缺陷等级、类型分布表和图表添加逻辑    # 8. 保存文档    doc.save(save_path)    self._show_msg("成功", f"测试报告导出完成！\n保存路径：{save_path}")

解析：

●前置校验：确保先生成基础数据和图表，避免导出空报告，提升代码健壮性；

●路径选择：通过 filedialog.asksaveasfilename 让用户自定义保存路径和文件名，默认文件名含系统名称，更人性化；

●样式封装：通过 _set_table_style() 和 _add_text_to_cell() 封装表格样式和单元格赋值逻辑，避免重复代码，后续修改样式只需改这两个函数；

●报告结构：按「用例统计→缺陷汇总→分布分析」的顺序排版，符合测试报告的常规阅读逻辑，直接适配正式汇报场景；

●图表插入：直接读取本地生成的图表图片，固定宽度确保排版美观，无需手动调整。

辅助工具函数是提升代码复用性的核心，本工具封装了5个关键辅助函数，被多个业务函数调用，避免重复编写相同逻辑。以两个核心辅助函数为例：

def _is_system_module(self, module_name):    """判断模块是否为系统管理类模块，通过关键词匹配"""    system_keywords = ['管理', '权限', '配置', '日志', '基础', '系统', '监控']    return any(keyword in module_name for keyword in system_keywords)

解析：通过关键词匹配快速判断模块类型，后续若需调整系统模块的判断规则，只需修改 system_keywords 列表，无需改动所有调用处，维护成本极低。

def _set_table_style(self, table):    """统一设置表格样式：边框、居中对齐、表头背景色"""    # 1. 设置所有单元格居中对齐    for row in table.rows:        for cell in row.cells:            cell.paragraphs[0].alignment = WD_ALIGN_PARAGRAPH.CENTER    # 2. 设置表头背景色为蓝色，文字白色    for cell in table.rows[0].cells:        shading_elm_1 = parse_xml(r'<w:shd w:val="clear" w:color="auto" w:fill="4472C4"/>')        cell._tc.get_or_add_tcPr().append(shading_elm_1)        for run in cell.paragraphs[0].runs:            run.font.color.rgb = RGBColor(255, 255, 255)  # 文字白色    # 3. 设置表格边框（黑色实线）    for row in table.rows:        for cell in row.cells:            for border in ['top', 'bottom', 'left', 'right']:                setattr(cell.border, border, WD_BORDER.SINGLE)                setattr(cell.border, f'{border}_color', RGBColor(0, 0, 0))

解析：统一封装表格样式，所有表格都调用此函数，确保报告中所有表格样式一致，后续若需调整报告风格（如修改表头颜色），只需修改此函数，无需逐个调整表格，复用性极强。

●低耦合高内聚：所有功能按模块拆分，模块间仅通过必要的数据交互，修改任意模块不影响其他功能，易维护、易扩展；

●代码复用性强：通用逻辑封装为辅助函数，一次编写多次调用，减少冗余代码，提升开发效率；

●健壮性高：所有用户输入都有严格校验，避免空值、格式错误等异常情况，工具运行稳定；

●用户体验友好：每步操作都有弹窗提示，界面布局符合操作习惯，默认填充常用数据，降低使用成本；

●轻量化易部署：单文件实现所有功能，无复杂依赖，复制即可运行，适配不同环境。