Python - 查看导入的模块的点状导入

时间:2018-03-28 06:51:07

标签: python python-import python-module

我想从任意模块m导入所有模块。 当m以下列方式导入模块时:

from a.b import c

或:

import a.b.c as c

(其中ab是包,c是模块)

dir(m)的结果将包含c(以及其他内容)。

然而,当我使用:

import a.b.c

dir(m)的结果将包含a,而不是c

我还尝试在导入sys._getframe(1).f_locals.items()后迭代m,但得到了相同的结果。 如果m的对象以这种方式导入c,我怎么知道它实际导入了c

1 个答案:

答案 0 :(得分:1)

您可以使用ast module来解析模块的源代码并提取导入:

import sys
import ast
import inspect


def find_imports(module):
    code = inspect.getsource(module)
    tree = ast.parse(code)

    imports = set()
    for node in ast.walk(tree):
        if isinstance(node, ast.Import):
            imports.update(alias.name for alias in node.names)
        elif isinstance(node, ast.ImportFrom):
            imports.add(node.module)
            for alias in node.names:
                module_name = '{}.{}'.format(node.module, alias.name)
                if module_name in sys.modules:
                    imports.add(module_name)

    return imports

请记住,只有在模块的源代码以普通python代码的形式提供时,这才有效。

例如,请考虑以下模块:

from os.path import isfile
from importlib import util
import urllib.request as urlquest
import urllib.parse
import sys

此模块的输出为

{'sys', 'urllib.request', 'os.path', 'importlib', 'importlib.util', 'urllib.parse'}

注意:

  1. 此代码还捕获函数中的导入,而不仅仅是全局范围内的导入。
  2. 如果代码执行from module import submodule之类的导入,"module.submodule"将出现在输出中。但是,如果代码执行from module import non_module之类的导入,则输出中只会显示"module"