使用pip安装Python包时如何自动指向外部库目录？

针对你遇到的问题——wheel分发时post-install脚本失效，无法自动识别路径随环境变化的外部.dll，这里给几个可行的解决思路：

方案1：运行时动态找路径（最靠谱） #

不用在安装阶段做操作，直接在包启动时主动查找.dll的路径，加到Python的搜索路径里。

具体操作： #

1. 通过Windows注册表获取路径
   像NVIDIA这类软件通常会在注册表中留下安装记录，用Python自带的winreg模块即可查询：

   import winreg
   import sys
   import os
   
   def get_nv_gst_path():
       try:
           # 打开对应注册表项（路径需根据软件实际情况调整）
           key = winreg.OpenKey(winreg.HKEY_LOCAL_MACHINE, r"SOFTWARE\NVIDIA Corporation\GSTPlugin")
           # 读取安装路径（键名可能为InstallPath，需自行确认）
           path, _ = winreg.QueryValueEx(key, "InstallPath")
           winreg.CloseKey(key)
           return path
       except FileNotFoundError:
           # 注册表未找到时尝试默认路径
           default_paths = [
               r"C:\Program Files\NVIDIA Corporation\GSTPlugin",
               r"C:\Program Files (x86)\NVIDIA Corporation\GSTPlugin"
           ]
           for p in default_paths:
               if os.path.exists(p) and "_NvGSTPlugin.dll" in os.listdir(p):
                   return p
           return None
   
   # 将这段代码放入包的__init__.py中
   dll_path = get_nv_gst_path()
   if dll_path and dll_path not in sys.path:
       sys.path.append(dll_path)
   

2. 验证.dll可正常加载
   为避免路径错误导致后续异常，可在初始化阶段添加加载验证：

   import ctypes
   try:
       _ = ctypes.CDLL("_NvGSTPlugin.dll")
   except OSError:
       raise ImportError("找不到_NvGSTPlugin.dll！请先安装NVIDIA GSTPlugin")
   

方案2：用entry-point实现安装后配置 #

wheel不支持post-install脚本，但可以通过自定义命令让用户安装后执行一次（或包首次启动时自动执行），生成.pth文件记录路径。

具体操作： #

1. 在setup.py中定义entry-point

   from setuptools import setup
   
   setup(
       name="my_package",
       # 其他配置按需填写
       entry_points={
           'console_scripts': [
               'my-package-config=my_package.config:setup_dll_path',
           ],
       },
   )
   

2. 编写配置脚本
   在my_package/config.py中实现路径查找和.pth文件生成逻辑：

   import os
   import sys
   import winreg
   
   def get_nv_gst_path():
       # 复用方案1中的路径查询代码
       pass
   
   def setup_dll_path():
       dll_path = get_nv_gst_path()
       if not dll_path:
           print("未找到_NvGSTPlugin.dll的路径，请先安装NVIDIA GSTPlugin")
           return
       
       # 定位到Python的site-packages目录
       site_pkgs = next(p for p in sys.path if 'site-packages' in p)
       pth_file = os.path.join(site_pkgs, 'my_package_dll.pth')
       
       # 将路径写入.pth文件
       with open(pth_file, 'w') as f:
           f.write(dll_path + '\n')
       
       print(f"配置完成！路径已写入{pth_file}")
   

3. 引导用户执行配置
   在包的README或首次启动时提示用户执行：

   my-package-config
   

   也可以在包的__init__.py中添加判断，首次运行时自动执行配置函数。

方案3：依赖系统PATH环境变量 #

如果用户已将.dll所在路径添加到系统PATH环境变量中，Python会自动搜索到该路径。可以在包的文档中引导用户手动添加，或在方案1的代码中优先检查PATH：

def get_nv_gst_path():
    # 先检查PATH中是否存在目标.dll
    for path in os.environ['PATH'].split(';'):
        if os.path.exists(os.path.join(path, "_NvGSTPlugin.dll")):
            return path
    # 再尝试注册表查询和默认路径
    # ...

注意事项 #

• 注册表的路径和键名需根据软件实际安装情况确认，不同版本可能存在差异。
• 必须处理路径找不到的情况，给出清晰的错误提示，引导用户排查问题。
• 若包需要跨平台，需添加系统判断逻辑，避免在非Windows系统执行注册表查询代码。

内容的提问来源于stack exchange，提问作者Manuel Sánchęz Garcíą