Python代码复杂度分析可视化工具：AST语法树解析函数圈复杂度、四级风险颜色编码、统计面板、HTML报告导出

🖥️ 运行效果

🚀 实例简介

本工具基于AST语法树自动分析Python源代码中每个函数的圈复杂度（Cyclomatic Complexity），精准识别分支语句、循环结构、异常处理等复杂度因子。界面采用表格化展示，四级风险颜色编码（低/中/高/危险）直观标记代码质量，统计面板实时显示函数总数、平均复杂度、最高复杂度、高风险函数数量等关键指标。支持一键导出专业HTML分析报告，帮助开发者快速定位复杂函数、优化代码结构。

💻 核心代码

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Python代码复杂度分析可视化工具
功能：加载Python源码文件，计算每个函数的圈复杂度，
      颜色编码标记风险等级，表格化展示分析结果，支持一键导出HTML报告。
作者：51Python
"""
import tkinter as tk
from tkinter import ttk, filedialog, messagebox
import ast
import os
import platform
import html

CHINESE_FONT = "WenQuanYi Zen Hei" if platform.system() == "Linux" else "Microsoft YaHei"


class ComplexityAnalyzer(ast.NodeVisitor):
    """AST访问器：计算函数圈复杂度"""
    def __init__(self):
        self.functions = []  # 存储所有函数分析结果

    def visit_FunctionDef(self, node):
        """访问函数定义节点，计算圈复杂度"""
        complexity = 1  # 基础复杂度为1
        # 遍历函数体内所有节点，统计分支语句
        for child in ast.walk(node):
            if isinstance(child, (ast.If, ast.While, ast.For)):
                complexity += 1
            elif isinstance(child, ast.BoolOp):
                complexity += len(child.values) - 1
            elif isinstance(child, ast.ExceptHandler):
                complexity += 1
            elif isinstance(child, ast.With):
                complexity += 1
            elif isinstance(child, ast.Assert):
                complexity += 1
        # 计算代码行数
        lines = node.end_lineno - node.lineno + 1 if hasattr(node, 'end_lineno') else 0
        self.functions.append({
            "name": node.name,
            "line": node.lineno,
            "complexity": complexity,
            "lines": lines,
            "args": len(node.args.args)
        })
        self.generic_visit(node)


def analyze_file(filepath):
    """分析单个Python文件的函数复杂度"""
    with open(filepath, "r", encoding="utf-8") as f:
        source = f.read()
    tree = ast.parse(source)
    analyzer = ComplexityAnalyzer()
    analyzer.visit(tree)
    return analyzer.functions


def get_risk_level(complexity):
    """根据复杂度返回风险等级和颜色"""
    if complexity <= 5:
        return "低", "#27ae60"
    elif complexity <= 10:
        return "中", "#f39c12"
    elif complexity <= 20:
        return "高", "#e67e22"
    else:
        return "危险", "#e74c3c"


class App(tk.Tk):
    def __init__(self):
        super().__init__()
        self.title("Python代码复杂度分析可视化工具")
        self.geometry("900x600")
        self.configure(bg="#f0f0f0")
        self.results = []
        self._build_ui()

💡 适用场景

办公自动化、数据分析、快速原型开发