如何在Pyodide中运行python -m black或调用其API格式化代码？

1. 安装Black包 #

首先通过micropip安装纯Python版本的Black轮子：

import micropip
await micropip.install("black==22.12.0")

（注：Pyodide中需使用异步调用，若在JS环境中执行，可通过pyodide.runPythonAsync运行这段代码）

2. 两种实现格式化的方式 #

方式一：模拟python -m black执行 #

利用Pyodide的虚拟文件系统，将用户代码写入临时文件后调用Black的命令行入口处理：

import black
import tempfile
import sys

def format_with_black_cli(code):
    # 创建临时文件写入用户代码
    with tempfile.NamedTemporaryFile(mode='w', suffix='.py', delete=False) as f:
        f.write(code)
        temp_path = f.name
    
    # 模拟命令行参数，调用Black的main函数
    sys.argv = ['black', temp_path]
    try:
        black.main()
        # 读取格式化后的内容
        with open(temp_path, 'r') as f:
            formatted_code = f.read()
        return formatted_code
    finally:
        # 清理临时文件
        import os
        os.unlink(temp_path)

调用该函数时，直接传入用户输入的代码字符串即可获取格式化结果。

方式二：直接调用Black的API（更高效） #

使用Black提供的format_str函数直接处理字符串，无需文件操作：

import black
from black import Mode

def format_with_black_api(code):
    try:
        # 使用默认格式化模式，可根据需求调整Mode参数（如设置行宽）
        formatted_code = black.format_str(code, mode=Mode())
        return formatted_code
    except black.InvalidInput as e:
        # 处理非法Python代码的情况
        return f"格式化错误：{str(e)}"

这种方式更轻量，适合直接处理用户输入的字符串。

3. 用户输入的安全处理 #

Black的API本身会处理合法的Python代码字符串，在Pyodide环境中传递用户输入时：

• 从JS向Python传递字符串时，使用Pyodide提供的类型转换方法（如pyodide.toPy()）避免编码问题
• 无需手动转义特殊字符，Black会自动处理代码中的引号、语法元素
• 必须添加异常捕获，处理用户输入的非法Python代码（如语法错误），避免程序崩溃

内容的提问来源于stack exchange，提问作者fejyesynb