如何获取由python模块导入的模块列表

Question

如果有的话，我想在 Python 中访问某种导入依赖跟踪。

我决定在我的模块中添加一个 __dependencies__ 字典，描述该模块导入的所有模块的版本。

我想要一种自动化的方式来获取我的模块导入的模块列表。最好在模块的最后一行。

ModuleFinder（如 How to list imports within a Python module? 所建议的）将不起作用，因为将对已加载的模块执行检查。

ModuleFinder 的另一个问题是它检查 Python 脚本（带有 if __name__ == '__main__' 分支），而不是模块。

如果我们考虑一个玩具脚本 script.py：

if __name__ == '__main__':
    import foo

那么结果是：

>>> mf = ModuleFinder
>>> mf.run_script('script.py')
>>> 'foo' in mf.modules
True

如果脚本作为模块导入，则应为 False。

我不想列出所有导入的模块——只列出我的模块导入的模块——所以sys.modules（What is the best way of listing all imported modules in python? 建议）会返回太多。

我可能会比较模块代码开头和结尾的sys.modules 的快照。但是那样我会错过我的模块使用的所有模块，但之前由任何其他模块导入。

列出模块从中导入对象的模块也很重要。

如果我们考虑一个玩具模块 example.py：

from foo import bar
import baz

那么结果应该是这样的：

>>> import example
>>> moduleImports(example)
{'foo': <module 'foo' from ... >,
 'baz': <module 'baz' from ...>}

（它还可能包含递归导入的模块或foo.bar 给定的 bar 是一个模块）。

使用globls()（根据How to list imported modules?）需要我手动处理非模块导入，例如：

from foo import bar
import bar

我怎样才能避免这种情况？

到目前为止，我的解决方案还有另一个问题。 PyCharm 倾向于清理我在重构时的手动导入，这使得它很难继续工作。

Answer 1

您可以使用inspect 模块

例如：

import inspect
import os
m = inspect.getmembers(os) # get module content
filter(lambda x: inspect.ismodule(x[1]), m) # filter dependant modules

你有一个live example

如果您想导入本地模块，只需使用：

filter(lambda x: inspect.ismodule(x[1]), locals().items()) # filter dependant modules

另一个live example