Question

在python中，你可以fname.__code__.co_names来检索函数引用的函数和全局事物的列表。如果我做fname.__code__.co_varnames，这包括内在功能，我相信。

有没有办法基本上inner.__code__.co_names？从一个类似'inner'的字符串开始，由co_varnames返回？

Answer 1

在Python 3.4+中，您可以使用dis.get_instructions获取名称。要支持嵌套函数，还需要以递归方式遍历遇到的每个代码对象：

import dis
import types

def get_names(f):
    ins = dis.get_instructions(f)
    for x in ins:
        try:
            if x.opcode == 100 and '<locals>' in next(ins).argval\
                                              and next(ins).opcode == 132:
                yield next(ins).argrepr
                yield from get_names(x.argval)
        except Exception:
            pass

<强>演示：

def func():
    x = 1
    y = 2
    print ('foo')
    class A:
        def method(self):
            pass
    def f1():
        z = 3
        print ('bar')
        def f2():
            a = 4
            def f3():
                b = [1, 2, 3]
    def f4():
        pass

print(list(get_names(func)))

<强>输出：

['f1', 'f2', 'f3', 'f4']

Answer 2

我认为你不能检查代码对象，因为内部函数是惰性的，它们的代码对象只是及时创建的。你可能想要看的是ast模块。这是一个简单的例子：

import ast, inspect

# this is the test scenario
def function1():
    f1_var1 = 42
    def function2():
        f2_var1 = 42
        f2_var2 = 42
        def function3():
            f3_var1 = 42

# derive source code for top-level function
src = inspect.getsource(function1)

# derive abstract syntax tree rooted at top-level function
node = ast.parse(src)

# next, ast's walk method takes all the difficulty out of tree-traversal for us
for x in ast.walk(node):
    # functions have names whereas variables have ids,
    # nested-classes may all use different terminology
    # you'll have to look at the various node-types to
    # get this part exactly right
    name_or_id = getattr(x,'name', getattr(x,'id',None))
    if name_or_id:
        print name_or_id

结果为：function1，function2，f1_var1，function3，f2_var1，f2_var2，f3_var1。强制性免责声明：可能没有充分理由做这类事情......但玩得开心：）

哦，如果你只想要内部函数的名字？

print dict([[x.name,x] for x in ast.walk(ast.parse(inspect.getsource(some_function))) if type(x).__name__=='FunctionDef'])

python - 列出函数的所有内部函数？

2 个答案: