python - 包 __init__.py 导入所有子文件，但只从另一个脚本加载一个？

Question

我创建了一个具有以下文件结构的包：

- package
  - __init__.py
  - load.py
  - train.py
  - test.py

我的__init__.py文件只是这些文件的类的导入：

from package.load import Load
from package.train import Train
from package.test import Test

大多数时候，我想加载所有三个，但有时我只想专门加载其中一个类。例如，在一个临时脚本（包外）中，我希望能够只调用 Load 类，如下所示：

from package import Load

Load虽然上述所有方法都适用于这个设计，但我遇到了一个问题，即当我像上面那样导入时，也会加载来自训练/测试的依赖项。如何设置__init__.py文件，以便在import不从训练/测试加载依赖项的情况下进行相同的调用？

补充说明：

我为什么这样做：我有一个问题，我希望某些人能够使用Load仅使用基本 python 的类，但是训练/测试文件包含专门的依赖项，只有 Load 类的用户不想使用或甚至安装。

标签： pythonpython-3.xpackage

Answer 1

这是一种非常接近你想要的东西的方法。您可以在其中定义一个函数来显式导入所需的任何类（或者如果未指定，则全部导入），而不是无条件地import在您的.__init__.py

__init__.py：

from pathlib import Path
import sys

print(f'In {Path(__file__).name}')

package_name = Path(__file__).parent.name
package_prefix = package_name + '.'
class_to_module_map = {'Load': 'load', 'Train': 'train', 'Test': 'test'}


def import_classes(*class_names):
    namespace = sys._getframe(1).f_globals  # Caller's globals.

    if not class_names:
        class_names = class_to_module_map.keys()  # Import them all.

    for class_name in class_names:
        module = class_to_module_map[class_name]
        temp = __import__(package_prefix+module, globals(), locals(), [class_name])
        namespace[class_name] = getattr(temp, class_name)  # Add to caller's namespace.

出于测试目的，这是我在load.py脚本中添加的内容：（
我还在其他两个模块中添加了类似的内容，以验证它们是否被import编辑。）

load.py：

from pathlib import Path

print(f'In {Path(__file__).name}')

class Load: pass

最后，这是一个仅import将其用于Load类的示例：

ad_hoc.py：

from my_package import import_classes

#from my_package import Load
import_classes('Load')

test = Load()
print(test)

随着产生的输出：

In __init__.py
In load.py
<my_package.load.Load object at 0x001FE4A8>