12

目录结构：

main.py
my_modules/
   module1.py
   module2.py

模块1.py：

class fooBar():
    ....
class pew_pew_FooBarr()
    ....
...

如何在没有前缀的情况下将 module* 中的所有类添加到 main（即像 foo = fooBar() 一样使用它们，而不是 foo = my_modules.module1.fooBar()）。

一个明显的决定是在 main.py 中编写如下内容：

from my_modules.module1 import *
from my_modules.module2 import *
from my_modules.module3 import *
...

但是我不想在创建新模块 N 时更改 main.py。有解决方案吗？

我知道导入这样的类不是一个好主意，但我仍然对此感到好奇。

UPD：这个问题与这个Loading all modules in a folder in Python不同，因为我的问题是加载没有命名空间的模块。

4

1 回答 1

33

在my_modules文件夹中，添加一个__init__.py文件以使其成为正确的包。在该文件中，您可以在文件的全局范围内注入每个模块的全局变量__init__.py，这使得它们在您的模块被导入时可用（在您还将全局名称添加到__all__变量之后）：

__all__ = []

import pkgutil
import inspect

for loader, name, is_pkg in pkgutil.walk_packages(__path__):
    module = loader.find_module(name).load_module(name)

    for name, value in inspect.getmembers(module):
        if name.startswith('__'):
            continue

        globals()[name] = value
        __all__.append(name)

现在，而不是这样做：

from my_modules.class1 import Stuff

你可以这样做：

from my_modules import Stuff

或者将所有内容导入全局范围，这似乎是您想要做的：

from my_modules import *

这种方法的问题是类相互覆盖，所以如果两个模块提供Foo，您将只能使用最后导入的一个。

于 2013-05-31T08:56:06.213 回答