Question

使用Clang我们可以做到：

clang -cc1 -ast-dump j.c

TranslationUnitDecl 0x7fbcfc00f608 <<invalid sloc>> <invalid sloc>
|-TypedefDecl 0x7fbcfc00fea0 <<invalid sloc>> <invalid sloc> implicit __int128_t '__int128'
| `-BuiltinType 0x7fbcfc00fba0 '__int128'
|-TypedefDecl 0x7fbcfc00ff08 <<invalid sloc>> <invalid sloc> implicit __uint128_t 'unsigned __int128'
| `-BuiltinType 0x7fbcfc00fbc0 'unsigned __int128'
|-TypedefDecl 0x7fbcfc0101b8 <<invalid sloc>> <invalid sloc> implicit __NSConstantString 'struct __NSConstantString_tag'
| `-RecordType 0x7fbcfc00ffd0 'struct __NSConstantString_tag'
|   `-Record 0x7fbcfc00ff58 '__NSConstantString_tag'
|-TypedefDecl 0x7fbcfc010250 <<invalid sloc>> <invalid sloc> implicit __builtin_ms_va_list 'char *'
| `-PointerType 0x7fbcfc010210 'char *'
|   `-BuiltinType 0x7fbcfc00f6a0 'char'
|-TypedefDecl 0x7fbcfc0104f8 <<invalid sloc>> <invalid sloc> implicit __builtin_va_list 'struct __va_list_tag [1]'
| `-ConstantArrayType 0x7fbcfc0104a0 'struct __va_list_tag [1]' 1
|   `-RecordType 0x7fbcfc010320 'struct __va_list_tag'
|     `-Record 0x7fbcfc0102a0 '__va_list_tag'
|-FunctionDecl 0x7fbcfb844200 <j.c:3:1, line:12:1> line:3:5 main 'int ()'
| `-CompoundStmt 0x7fbcfb8447b8 <col:12, line:12:1>
|   |-DeclStmt 0x7fbcfb844350 <line:4:3, col:8>
|   | `-VarDecl 0x7fbcfb8442f0 <col:3, col:7> col:7 used e 'int'
....

是否可以使用Python的AST做到这一点？

我发现astdump：https://pypi.org/project/astdump/

但是它不会打印令牌的文字：

>>> import astdump
>>> astdump.indented('2+3')
Module
  Expr
    BinOp
      Num
      Add
      Num

我需要能够从AST重构整个代码。

Answer 1

astpretty library似乎很适合您的目的。该库具有漂亮的打印功能pprint，该功能以可读格式呈现AST的树结构，包括节点类型和内容。您需要将此与Python标准库中的ast.parse结合使用。

pprint的默认行为更为冗长，包括每个节点的行号和列偏移，但是可以使用参数show_offsets=False禁用它。下面的用法示例来自astpretty库的自述文件。

>>> astpretty.pprint(ast.parse('x += 5').body[0], show_offsets=False)
AugAssign(
    target=Name(id='x', ctx=Store()),
    op=Add(),
    value=Num(n=5),
)

请注意，如果您不需要精美的打印，则标准库的ast.dump将起作用。输出将有些可读，但不太可读，因为它不会缩进以显示树结构：

>>> print(ast.dump(ast.parse('x += 5').body[0]))
AugAssign(target=Name(id='x', ctx=Store()), op=Add(), value=Num(n=5))

生成Python的AST的文本表示形式

1 个答案: