如何将Python包特定版本依赖的安装集成至标准安装流程？

你的需求核心是将特定版本的依赖安装到包内的子目录，并确保运行时优先加载这些依赖，要替代shell脚本的话，最标准的做法是通过setuptools自定义安装命令，配合包初始化时的路径注入来实现。下面是分步的具体方案：

一、自定义安装命令，自动安装依赖到包内子目录 #

我们可以在setup.py中继承setuptools的install命令，在完成包本身的安装后，自动调用pip将指定版本的依赖安装到mypackage/site-packages目录下。

示例setup.py代码： #

import os
import subprocess
from setuptools import setup, find_packages
from setuptools.command.install import install

# 定义你需要的依赖及版本
REQUIREMENTS = [
    "requests==2.28.2",
    "numpy==1.21.6"
]

class CustomInstall(install):
    def run(self):
        # 先执行标准的包安装流程
        install.run(self)
        
        # 获取安装后的mypackage目录路径
        # self.install_lib是setuptools提供的安装目标库路径
        package_site_packages = os.path.join(self.install_lib, "mypackage", "site-packages")
        # 创建目录（如果不存在）
        os.makedirs(package_site_packages, exist_ok=True)
        
        # 调用pip将依赖安装到指定目录
        subprocess.check_call([
            self._get_pip_path(),
            "install",
            "-t", package_site_packages,
            *REQUIREMENTS
        ])
    
    def _get_pip_path(self):
        # 确保使用当前环境的pip，避免跨环境问题
        import sys
        return os.path.join(sys.executable, "..", "pip") if sys.platform == "win32" else sys.executable.replace("python", "pip")

setup(
    name="mypackage",
    version="1.0.0",
    packages=find_packages(),
    cmdclass={
        "install": CustomInstall
    },
    # 这里可以保留install_requires作为兜底，但主要依赖安装逻辑在CustomInstall里
    # install_requires=REQUIREMENTS
)

二、运行时注入内部依赖路径到sys.path #

在你的包的__init__.py中，添加路径注入逻辑，确保Python优先加载包内的依赖：

示例mypackage/__init__.py代码： #

import os
import sys
from importlib.util import find_spec

# 获取当前包的安装路径
package_path = os.path.dirname(find_spec(__name__).origin)
# 拼接内部site-packages路径
internal_site_packages = os.path.join(package_path, "site-packages")

# 将内部路径添加到sys.path最前面，确保优先加载
if internal_site_packages not in sys.path:
    sys.path.insert(0, internal_site_packages)

# 这里可以添加你的包初始化逻辑

三、关键注意事项 #

1. 导入顺序说明：你提到的"必须先导入mypackage再导入依赖"的限制依然存在，一定要在文档中明确说明这一点——因为如果用户先导入了系统中的依赖，Python会缓存该模块，后续再导入mypackage也无法覆盖。
2. 跨平台兼容性：上面的代码已经处理了Windows和类Unix系统的路径差异，但建议在不同环境下测试验证。
3. 依赖体积问题：这种方式会让你的包体积变大（因为包含了所有依赖），如果是公开分发的包，需要权衡利弊；如果是内部使用的包，这个问题不大。
4. pip版本兼容性：确保用户环境的pip支持-t参数（pip 6.0+都支持，现在基本都是满足的）。

替代方案参考 #

如果你不想自定义安装命令，也可以：

• 将依赖打包成package_data，但这种方式需要手动处理依赖的安装结构，不如pip install -t灵活。
• 使用setuptools的data_files参数，但同样需要手动管理依赖的放置路径，维护成本更高。

总体来说，自定义安装命令+运行时路径注入是最贴合你需求的标准实现方式，完全可以替代原来的shell脚本。

内容的提问来源于stack exchange，提问作者Jibbity jobby