如果从 B 调用 A.f，模块 A 中定义的 A.f 如何获取 B 中定义的函数列表

Question

假设我有B.py

import A

def F():
    pass

def G():
    pass

if __name__ == '__main__':
    A.f()

当我运行它时，我希望A.f 像这样输出B.py 中定义的函数列表

./B.py
['F', 'G']

问题是A.py应该写什么？

def f():
    ???

这似乎是可能的，因为doctest 做了这样的事情。

更新感谢您的回复，但我忘了再说一个限制：我不想在A.py 中做import B。

让我试着解释一下原因：A.py 是一个用于运行 MapReduce 作业的库。您只需导入A 并在脚本的不同端运行一些A.f。然后A.f 分析您定义的功能并在服务器上执行它们。所以A.f 不能导入每个调用它的模块。

或者可以吗？函数可以知道它是从哪里调用的吗？

更新

def f():
    module = sys.modules['__main__']
    functions = inspect.getmembers(module, inspect.isfunction)
    ...

Answer 1

def f():
    import B
    import inspect

    functions=inspect.getmembers(B,inspect.isfunction)

http://docs.python.org/library/inspect.html#inspect.getmembers

编辑

我没有尝试过，但您似乎可以将模块作为参数传递给 f 并使用它。

def f(mod):
    import inspect
    functions=inspect.getmembers(mod,inspect.isfunction)
    return functions

然后，从 B 开始，您可以执行以下操作：

import sys
import A
myfuncs=A.f(sys.modules[__name__])

Answer 2

这实际上很容易使用the dir() function：

import B

def f():
    print(dir(B))

这给出了：

['A', 'F', 'G', '__builtins__', '__cached__', '__doc__', '__file__', '__name__', '__package__']

如您所见，这确实包含了模块中通常看不到的隐藏值。如果您想更具体地了解您想要的东西，请查看the inspect module。