将模块添加到 sys.modules 以处理非相对子模块导入

Question

我有一个我正在销售的子模块，其中写得不好（它不使用相对导入，但有两个包。需要说明的是，我没有安装这些包（因为它们没有维护良好的包） pypi，只包括源代码。）

所以布局如下所示：

root
  |
  |-code
  |   |-file.py
  |
  |-vendor
      |-submodule
           |-package_1
           |    |-alpha.py
           |-package_2
                |-beta.py

不幸的是，beta.py 尝试使用 import package_1 不起作用，因为没有 __init__.py。因为我在 CI/CD 期间更新了所有子模块，所以缺少相关导入会破坏我的测试。

如果所有内容都在根目录中，这将起作用，但我无法控制子模块。我也不想更改 alpha.py 或 beta.py，因为我不想处理分叉。

有没有什么办法可以拥有一个通用的__init__.py 或类似的东西，这样当 beta.py 导入时，它会看到 package_1 和 alpha.py？

Answer 1

__init__.py 或缺少 PEP 420's Implicit Namespace Packages 与现代 (3.3+) Python 上的某个包是否算作包无关。

某事告诉我供应商包希望包含他们两个包的目录（在这种情况下为root/vendor/submodule）通过某种机制添加到sys.path（可能只是安装在site-packages的顶层，而不是在您设置的某个嵌套目录中）；如果您的重新打包未能将它们放在sys.path 中some 条目的根目录中，也没有更新sys.path 以包含您放置它们的位置，那么它们的包将不起作用。

解决方案是将他们的包安装到sys.path 位置或更新sys.path 以包含他们的位置，因此package_1 是package_2 所期望的顶级包。

假设这些供应商模块仅在file.py 中使用，您可以使用以下代码手动更新file.py 的导入sys.path：

import pathlib
import sys

# Get the path to file.py, go up to the parent directory, then down to 
# vendor/submodule
vendor_dir = pathlib.Path(__file__).parent.parent / 'vendor' / 'submodule'
sys.path.append(str(vendor_dir))

# import package_1.alpha or package_2.beta will now work