python - LibCST:将任意节点转换为代码

Question

是否可以将任意 LibCST 节点转储到 Python 代码中？我的用例是我想提取与特定命名方案匹配的函数的代码。我可以提取我需要的 FunctionDef 节点，但我似乎找不到将它们转换为代码的方法。

Answer 1

可以使用 code_for_node 方法来自 Module 类。

按如下方式使用它:

import libcst

function_def = libcst.parse_statement("def hello_world():\n  print('Hello World')")
print(libcst.Module([]).code_for_node(function_def))

这将生成输出:

def hello_world():
    print('Hello World')