IndentationError 在 `ast.parse` 和 `ast.walk` 的一个函数中，该函数是类中的一个方法

Question

我想我知道IndentationError 中IndentationError描述的类似的常见原因：例如，unindent 不匹配任何外部缩进级别。这不适用于这里。

另外，我知道，textwrap.dedent但感觉这里不是正确的方法？

---

如果我有一个“正规”的功能，我可以做ast.parse和ast.walk这样的：

import ast
import inspect

def a():
    pass

code = inspect.getsource(a)
nodes = ast.walk(ast.parse(code))
for node in nodes:
    ...

但是，如果函数是类中的方法，例如：

class B:
    def c(self):
        pass

code = inspect.getsource(B.c)
nodes = ast.walk(ast.parse(code))

我得到：

IndentationError: unexpected indent

我想这是有道理的，因为B.c 它缩进了一级。那么我ast.parse和ast.walk这里怎么办呢？

Answer 1

这是因为您抓住了方法而不是尝试在不撤消缩进的情况下走它。你的班级是：

class B:
    def c(self):
        pass

code = inspect.getsource(B.c)
nodes = ast.walk(ast.parse(code))

如果您打印，code您会看到：

    def c(self):
        pass

注意：上面的代码有一个缩进。您需要取消缩进：

import inspect
import ast
import textwrap
class B:
    def c(self):
        pass
code = textwrap.dedent(inspect.getsource(B.c))
nodes = ast.walk(ast.parse(code))