如何遍历模块的功能

时间:2014-02-19 16:01:47

标签: python functional-programming getattr

导入foo.py后我有这个函数调用。 Foo有几种我需要调用的方法,例如foo.paint,foo.draw:

import foo

code

if foo:
    getattr(foo, 'paint')()

我需要使用while循环来调用和遍历所有函数foo.paint,foo.draw等。我该怎么做呢?

1 个答案:

答案 0 :(得分:23)

你可以这样使用foo.__dict__

for name, val in foo.__dict__.iteritems(): # iterate through every module's attributes
    if callable(val):                      # check if callable (normally functions)
        val()                              # call it

但请注意,这将执行模块中的每个函数(可调用)。如果某个特定函数收到任何参数,它将失败。

获得函数的更优雅(功能)方式是:

[f for _, f in foo.__dict__.iteritems() if callable(f)]

例如,这将列出math方法中的所有函数:

import math
[name for name, val in math.__dict__.iteritems() if callable(val)]
['pow',
 'fsum',
 'cosh',
 'ldexp',
 ...]