[英]python - list all inner functions of a function?
In python you can do fname.__code__.co_names
to retrieve a list of functions and global things that a function references.在 python 中,您可以执行fname.__code__.co_names
来检索函数列表和函数引用的全局事物。 If I do fname.__code__.co_varnames
, this includes inner functions, I believe.如果我做fname.__code__.co_varnames
,我相信这包括内部函数。
Is there a way to essentially do inner.__code__.co_names
?有没有办法基本上做inner.__code__.co_names
? by starting with a string that looks like 'inner'
, as is returned by co_varnames
?从一个看起来像'inner'
的字符串开始,正如co_varnames
返回的那样?
In Python 3.4+ you can get the names using dis.get_instructions
.在 Python 3.4+ 中,您可以使用dis.get_instructions
获取名称。 To support nested functions as well you need to recursively loop over each code object you encounter:要同时支持嵌套函数,您需要递归循环您遇到的每个代码对象:
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
Demo:演示:
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)))
Outputs:输出:
['f1', 'f2', 'f3', 'f4']
I don't think you can inspect the code object because inner functions are lazy, and their code-objects are only created just in time.我认为您不能检查代码对象,因为内部函数是惰性的,并且它们的代码对象只是及时创建的。 What you probably want to look at instead is the ast module.您可能想要查看的是 ast 模块。 Here's a quick example:这是一个简单的例子:
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
The results are: function1, function2, f1_var1, function3, f2_var1, f2_var2, f3_var1.结果为:function1、function2、f1_var1、function3、f2_var1、f2_var2、f3_var1。 Obligatory disclaimer: there's probably not a good reason for doing this type of thing.. but have fun :)强制性免责声明:做这种事情可能没有充分的理由......但玩得开心:)
Oh and if you only want the names of the inner functions?哦,如果你只想要内部函数的名称?
print dict([[x.name,x] for x in ast.walk(ast.parse(inspect.getsource(some_function))) if type(x).__name__=='FunctionDef'])
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.