如何实现Python代码反向美化（丑化）：合并可合并多行至单行

要实现将Python代码中为可读性添加的非必要换行合并，同时保证代码逻辑不变，可以借助Python内置的抽象语法树（AST）模块来处理——AST会忽略代码的格式细节，只保留核心逻辑，再通过反解析生成紧凑的代码。

实现步骤与代码示例 #

import ast

def get_compact_code_lines(file_path):
    # 读取目标文件中的Python代码
    with open(file_path, 'r', encoding='utf-8') as f:
        raw_code = f.read()
    
    # 将代码解析为抽象语法树，剥离格式类冗余信息
    code_ast = ast.parse(raw_code)
    
    # 反解析AST生成紧凑代码，自动合并非语法必要的换行
    compact_code = ast.unparse(code_ast)
    
    # 将紧凑代码按行分割为列表返回
    return compact_code.splitlines()

# 使用示例
compact_lines = get_compact_code_lines("your_python_file.py")
for line in compact_lines:
    print(line)

原理说明 #

AST（抽象语法树）是代码的结构化逻辑表示，它只关注代码的语法逻辑，不会保留为了可读性添加的换行、缩进等格式。ast.unparse方法会基于AST生成最紧凑的代码输出：

• 自动合并括号内的多行内容（如导入语句的括号、函数参数括号、打印语句的括号）
• 移除逗号、运算符后的非必要换行
• 生成的代码与原代码逻辑完全一致，可正常运行

兼容低版本Python #

如果使用Python 3.8及更早版本，ast.unparse方法不可用，可以使用第三方库astor来实现：

1. 安装库：pip install astor
2. 修改代码中的反解析部分：

import astor
# 替换ast.unparse为以下代码
compact_code = astor.to_source(code_ast, pretty_print=False, indent_with_space=False)

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