3

我想对任何函数/方法进行一些反省。对于我的所有示例，我都使用 Python 2.7，但是如果使用 3.3 可以使事情变得更容易，那么使用 3.3 不是问题。

假设我在名为 foobar.py 的模块中有以下代码：

def foo():
    bar()

我可以看到 foo 运行的动态代码：

import inspect
import foobar
inspect.getsource(foobar.foo)

我还可以从这个函数的代码对象中获取反汇编的字节码：

import dis
dis.dis(foobar.foo)

有没有一种方法可以检测到该foo方法调用了另一个函数（bar在这种情况下），然后动态地反汇编/检查它？

我知道代码对象本身具有各种属性，如下所示：

>>> dir(foobar.foo.__code__)
['__class__', '__cmp__', '__delattr__', '__doc__', '__eq__', '__format__', '__ge__', '__getattribute__', '__gt__', '__hash__', '__init__', '__le__', '__lt__', '__ne__', '__new__', '__reduce__', '__reduce_ex__', '__repr__', '__setattr__', '__sizeof__', '__str__', '__subclasshook__', 'co_argcount', 'co_cellvars', 'co_code', 'co_consts', 'co_filename', 'co_firstlineno', 'co_flags', 'co_freevars', 'co_lnotab', 'co_name', 'co_names', 'co_nlocals', 'co_stacksize', 'co_varnames']

我检查了他们中的大多数，只是环顾四周，但还没有完全找到我要找的东西。

最终目标只是一个小实验，看看我是否可以在不执行导入以外的代码的情况下递归打印出一个可能的调用堆栈。我知道理论上的调用堆栈不能解释运行时的事情，比如特定变量的状态等。我只想打印出给定某个调用的所有嵌套函数的源代码（即使代码永远不会基于运行时状态）。

此外，我知道一旦我进入 CPython 代码，inspectanddis模块将无济于事。最终，打印出某种映射可能会很有趣，该映射显示它何时到达inspect并dis分解了 CPython 代码。但是，我什至不确定这是否可能。

4

1 回答 1

2

所有编译器/解释器在解析源代码时都会构建一个抽象语法树。这是基于其上下文无关语法的程序的表示，然后可以递归地遍历以生成可以由机器执行的代码。

Python 提供对其 AST 的访问，您可以自己遍历这棵树并ast.Call在ast.FunctionDef. 下面贴一个简单的例子。请注意，这肯定不会捕获所有可能的调用，因为调用可以嵌入到其他表达式中，被eval表达式隐藏等等。这是一个简单的例子：

import ast

source = """
def foo():
    bar()

def bar():
    baz()

def baz():
    print "hello!"
"""

def get_method_name_for_call(call_obj):
    for fieldname, value in ast.iter_fields(call_obj):
        if fieldname == "func":
            return value.id

def walk_method_calls(node, cur_func):
    if not node:
        return

    for cur_node in ast.iter_child_nodes(node):
        if type(cur_node) == ast.Call:
            method_called = get_method_name_for_call(cur_node)
            print "Found a call to %s in body of %s." % (method_called, cur_func)
        walk_method_calls(cur_node, cur_func)


def walk_function_defs(node):
    if not node:
        return

    for cur_node in ast.iter_child_nodes(node):
        if type(cur_node) == ast.FunctionDef:
            walk_method_calls(cur_node, cur_node.name)

# we pass <string> as a recognizable identifier since
# we have no filename
ast_root = compile(source, "<string>", "exec", ast.PyCF_ONLY_AST)
walk_function_defs(ast_root)

和一个执行示例：

$ python abstract_tree.py 
Found a call to bar in body of foo.
Found a call to baz in body of bar.
于 2012-12-08T16:24:42.050 回答