我想以编程方式编辑python源代码。基本上我想读取一个.py文件，生成AST，然后写回修改后的python源代码(即另一个.py文件)。
有一些方法可以使用标准的python模块(如ast或compiler)来解析/编译python源代码。但是，我认为它们都不支持修改源代码的方法(例如删除这个函数声明)，然后再写回修改的python源代码。
更新:我想这样做的原因是我想为python写一个突变测试库，主要是通过删除语句/表达式，重新运行测试，看看有什么破坏。
当前回答
如果你在2019年看到这个，那么你可以使用这个libcst 包中。它的语法类似ast。这就像一个魅力，并保留了代码结构。这对于你必须保留注释、空格、换行符等的项目是很有帮助的。
如果您不需要关心保留注释、空格和其他内容，那么ast和astor的组合可以很好地工作。
其他回答
我最近创建了相当稳定的(核心是经过良好测试的)和可扩展的代码段，它从ast树生成代码:https://github.com/paluh/code-formatter。
我正在使用我的项目作为一个小vim插件的基础(我每天都在使用)，所以我的目标是生成非常漂亮和可读的python代码。
P.S. I've tried to extend codegen but it's architecture is based on ast.NodeVisitor interface, so formatters (visitor_ methods) are just functions. I've found this structure quite limiting and hard to optimize (in case of long and nested expressions it's easier to keep objects tree and cache some partial results - in other way you can hit exponential complexity if you want to search for best layout). BUT codegen as every piece of mitsuhiko's work (which I've read) is very well written and concise.
如果你在2019年看到这个，那么你可以使用这个libcst 包中。它的语法类似ast。这就像一个魅力，并保留了代码结构。这对于你必须保留注释、空格、换行符等的项目是很有帮助的。
如果您不需要关心保留注释、空格和其他内容，那么ast和astor的组合可以很好地工作。
我们也有类似的需求，这里的其他答案并没有解决这个问题。因此，我们为此创建了一个库ASTTokens，它采用AST或astroid模块生成的AST树，并用原始源代码中的文本范围标记它。
它不直接修改代码，但在上面添加代码并不难，因为它会告诉您需要修改的文本范围。
例如，这将在WRAP(…)中包装一个函数调用，保留注释和其他内容:
example = """
def foo(): # Test
'''My func'''
log("hello world") # Print
"""
import ast, asttokens
atok = asttokens.ASTTokens(example, parse=True)
call = next(n for n in ast.walk(atok.tree) if isinstance(n, ast.Call))
start, end = atok.get_text_range(call)
print(atok.text[:start] + ('WRAP(%s)' % atok.text[start:end]) + atok.text[end:])
生产:
def foo(): # Test
'''My func'''
WRAP(log("hello world")) # Print
希望这能有所帮助!
花了一些时间，但是Python 3.9有这个: https://docs.python.org/3.9/whatsnew/3.9.html#ast https://docs.python.org/3.9/library/ast.html#ast.unparse
ast.unparse(ast_obj)
取消解析ast.AST对象并生成一个字符串，如果使用ast.parse()进行解析，该字符串将生成一个等效的ast.AST对象。
我以前使用baron，但现在已经切换到parso，因为它是现代python的最新版本。效果很好。
我还需要这个做变异测试。用parso做一个真的很简单，请访问https://github.com/boxed/mutmut查看我的代码
