能否在Python代码内调用2to3等工具检测并升级代码兼容Python 3？

2to3对应的核心能力封装在Python标准库lib2to3模块中，可直接在代码内调用，无需启动终端进程。以下是符合需求的实现：

from lib2to3.refactor import RefactoringTool, get_fixers_from_package

def checkCodeCompatibility(functionInString):
    # 加载所有Python2转3的内置修复规则
    fixers = get_fixers_from_package("lib2to3.fixes")
    refactor_tool = RefactoringTool(fixers)
    
    # 执行转换
    try:
        converted_bytes = refactor_tool.refactor_string(functionInString, "<string>")
    except Exception:
        # 语法解析异常可根据业务需求调整处理逻辑，这里默认返回原字符串
        return functionInString
    
    converted_str = str(converted_bytes)
    # 转换后内容和原内容一致说明无Python3兼容问题
    if converted_str.strip() == functionInString.strip():
        return functionInString
    # 存在兼容问题，返回转换后的Python3兼容版本
    return converted_str

测试示例 #

输入Python2语法的函数字符串：

test_py2_code = """
def calculate_sum(start, end):
    total = 0
    for i in xrange(start, end):
        if i % 2 == 0:
            print i
            total += i
    return total
"""

调用函数后返回的转换结果：

def calculate_sum(start, end):
    total = 0
    for i in range(start, end):
        if i % 2 == 0:
            print(i)
            total += i
    return total

可选优化项 #

• 可通过筛选fixers列表指定需要启用的转换规则，减少不必要的检测开销
• 转换后可使用Python3的ast.parse()方法对结果做语法校验，确保转换结果可用
• 如需检测Python3.10+等更高版本的语法兼容性，可搭配pyupgrade等工具的API使用

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