Function execute_matplotlib_code

dataflow_agent/utils_common.py:1228–1353

安全执行 matplotlib 代码并保存图表。 参数 ---- code : str 要执行的 matplotlib Python 代码 output_path : str | Path 图表输出路径（包含文件名，如 /tmp/chart.png） timeout : int, default 30 执行超时时间（秒） allowed_modules : List[str], optional 允许导入的模块列表，默认为 ["matplotlib", "numpy",

(
    code: str,
    output_path: Union[str, Path],
    timeout: int = 30,
    allowed_modules: Optional[List[str]] = None,
)

def execute_matplotlib_code(
    code: str,
    output_path: Union[str, Path],
    timeout: int = 30,
    allowed_modules: Optional[List[str]] = None,
) -> Dict[str, Any]:
    """
    安全执行 matplotlib 代码并保存图表。

    参数
    ----
    code : str
        要执行的 matplotlib Python 代码
    output_path : str | Path
        图表输出路径（包含文件名，如 /tmp/chart.png）
    timeout : int, default 30
        执行超时时间（秒）
    allowed_modules : List[str], optional
        允许导入的模块列表，默认为 ["matplotlib", "numpy", "pandas"]

    返回
    ----
    Dict[str, Any]
        {
            "success": bool,
            "output_path": str,      # 成功时返回图片路径
            "error": str,            # 失败时返回错误信息
        }
    """
    import subprocess
    import tempfile

    if allowed_modules is None:
        allowed_modules = ["matplotlib", "numpy", "pandas", "math"]

    output_path = Path(output_path)
    output_path.parent.mkdir(parents=True, exist_ok=True)

    # 安全性检查：禁止危险操作
    dangerous_patterns = [
        r'\bos\.system\b',
        r'\bsubprocess\b',
        r'\beval\b',
        r'\bexec\b',
        r'\bopen\s*\(',
        r'\b__import__\b',
        r'\bimport\s+os\b',
        r'\bimport\s+sys\b',
        r'\bimport\s+subprocess\b',
        r'\bfrom\s+os\b',
        r'\bfrom\s+sys\b',
    ]

    for pattern in dangerous_patterns:
        if re.search(pattern, code):
            return {
                "success": False,
                "output_path": "",