能否通过环境标记判断Databricks Runtime并实现条件依赖安装？

目前PEP 0508的环境标记确实不支持直接识别Databricks Runtime（DBR）版本，但可以通过以下几种方式实现基于DBR版本的条件依赖安装：

1. 利用DBR暴露的环境变量做动态安装 #

Databricks Runtime会在集群节点中设置DATABRICKS_RUNTIME_VERSION环境变量，你可以通过脚本读取这个值，动态生成对应版本的依赖列表：

• 示例bash脚本：

  # 提取DBR主版本号
  DBR_MAJOR=$(echo "$DATABRICKS_RUNTIME_VERSION" | cut -d'.' -f1)
  # 初始化基础依赖
  echo "your-common-lib==1.0.0" > temp-requirements.txt
  # 根据DBR版本追加对应依赖
  case $DBR_MAJOR in
    13) echo "specific-lib==2.3.0" >> temp-requirements.txt ;;
    14) echo "specific-lib==3.1.0" >> temp-requirements.txt ;;
    15) echo "specific-lib==4.0.0" >> temp-requirements.txt ;;
  esac
  # 执行安装
  pip install -r temp-requirements.txt
  

• 如果用Python脚本实现，逻辑类似：读取环境变量，判断版本后生成requirements内容再调用pip安装。

2. 在库的setup.py中动态判断依赖 #

如果你的Python库是通过setup.py分发的，可以在安装阶段自动检测DBR版本并加载对应依赖：

import os
from setuptools import setup

def get_dbr_dependent_requirements():
    base_deps = ["your-core-lib==1.0.0"]
    dbr_version = os.environ.get("DATABRICKS_RUNTIME_VERSION")
    if not dbr_version:
        return base_deps
    # 提取主版本号
    major_ver = int(dbr_version.split('.')[0])
    if major_ver == 13:
        base_deps.append("compat-lib==1.2.0")
    elif major_ver >=14:
        base_deps.append("compat-lib==2.0.0")
    return base_deps

setup(
    name="your-dbr-universal-lib",
    version="0.1.0",
    install_requires=get_dbr_dependent_requirements(),
    # 其他包配置项...
)

当用户在DBR集群中安装你的库时，setup.py会自动根据当前环境加载对应版本的依赖。

3. 借助DBR与Python版本的绑定关系间接实现 #

每个DBR LTS版本都绑定了固定的Python版本（比如DBR 13.x对应Python 3.9，DBR 14.x对应Python 3.10），可以利用PEP 508的python_version标记间接匹配：

# requirements.txt
your-common-lib==1.0.0
compat-lib==1.2.0; python_version == "3.9"
compat-lib==2.0.0; python_version >= "3.10"

注意：这种方式的前提是DBR和Python版本的对应关系长期稳定，需要提前确认官方版本矩阵，避免DBR版本更新后对应关系变化导致依赖安装错误。

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