Question

我有很多模块（数百个）。在每个模块中，我至少有一个功能。我的问题是如何只用一行从所有模块导入所有功能？因为我不想这样做：

from tests.test_001 import test_001_func
from tests.test_002 import test_002_func
from tests.test_003 import test_003_func
...
from tests.test_999 import test_999_func

test_xxx是模块，这些模块包含test_xxx_func函数，所有模块都在一个文件夹中。

我想使用这样的东西：

from tests import *
test_001.test_001_func()

但这不起作用

Answer 1

在tests目录中创建一个__init__.py文件并放在其中：

__all__ = [
    "test_001_func", "test_002_func", # etc
    ]

然后你可以：

import tests

tests.test_001_func()

或

from tests import *  # Not the best way to do things.

N.B。 import *不是首选解决方案的原因是您从调用中删除了命名空间，因此您可以a）可能与其他模块中的名称发生冲突，并且b）您的代码不太清楚。

Answer 2

您需要通过在程序包的tests文件中添加__all__列表来提供__init__.py程序包内容的索引。

在您的情况下，__init__.py目录中的tests文件将具有以下内容：

__all__ == ["test_001", "test_002" <<etc...>>]

请参阅importing * from a package上的文档。

Answer 3

您可以使用下面的脚本自动导入。

假设文件夹结构是下一个：

run.py
tests
  -> test_001.py
  -> test_002.py

tests/__init__.py 的代码

import os
import sys

# Append to path "test" folder (also can be any other name)
sys.path.append(os.path.basename(os.path.dirname(__file__)))

__all__ = []
for k in xrange(1, 3):
    name = "test_%03d" % k

    # __import__ can load module by string name
    globals()[name] = __import__(name)

    # set that we export only required names like test_001, test_002
    __all__.append(name)

test_001.py的代码：

def test_001_func():
    print "test1"

test_002.py的代码：

def test_002_func():
    print "test2"

run.py代码： 导入测试

print tests.test_001
print tests.test_001.test_001_func

print tests.test_002

run.py脚本的输出将是下一个：

test@test:~/Desktop/python$ python run.py 
<module 'test_001' from 'tests/test_001.pyc'>
<function test_001_func at 0xb728b09c>
<module 'test_002' from 'tests/test_002.pyc'>

Answer 4

正如其他答案所指出的那样，这一切都归结为填充__all__，但我想你可以自动完成一些事情：

# In your tests/__init__.py do something along the following lines:

import os.path, pkgutil, importlib
__path = os.path.dirname(__file__)
__all__ = [name for _, name, _ in pkgutil.iter_modules([__path])]
for __module in __all__:
    importlib.import_module("." + __module, __name__)

由于importlib，代码是Python 2.7+，但是可以使用裸__import__修改过时的Python版本。

然后像这样使用它：

import tests
tests.test_002.test_002_func()

或者

from tests import *
test_002.test_002_func()

但是，如果这是用于单元测试，我强烈建议您查看unittest或类似Nose的内容，以更加Pythonic的方式处理测试发现

Python，从模块导入函数

4 个答案: