我想从任意模块m
导入所有模块。
当m
以下列方式导入模块时:
from a.b import c
或:
import a.b.c as c
(其中a
和b
是包,c
是模块)
dir(m)
的结果将包含c
(以及其他内容)。
然而,当我使用:
import a.b.c
dir(m)
的结果将包含a
,而不是c
。
我还尝试在导入sys._getframe(1).f_locals.items()
后迭代m
,但得到了相同的结果。
如果m
的对象以这种方式导入c
,我怎么知道它实际导入了c
?
答案 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'}
from module import submodule
之类的导入,"module.submodule"
将出现在输出中。但是,如果代码执行from module import non_module
之类的导入,则输出中只会显示"module"
。