如何在setup.py中通过extras_require实现依赖Extension模块的可选编译安装？

我完全理解你的痛点——不想把可选的Fortran模块拆成单独的包，也不想让用户跑两次setup，只想通过一个简单的pip命令就能选择是否编译那个依赖f2py的扩展模块。其实这个需求是可以实现的，核心思路是在setup.py里动态判断用户是否指定了安装对应extra，再决定要不要添加Extension模块，下面给你详细讲具体实现：

核心实现步骤 #

我们可以利用Python的运行时特性，在执行setup()函数之前，检查用户的安装命令是否包含我们定义的extra名称，动态生成需要编译的扩展模块列表。

1. 编写setup.py的基本结构

首先导入必要的模块，然后写一个函数来动态判断并返回扩展模块：

import sys
from setuptools import setup
from numpy.distutils.core import Extension  # 用numpy的Extension来处理f2py编译

def get_extensions():
    extensions = []
    # 检查命令行参数中是否包含我们的extra名称（比如wind_model）
    # 适配pip install .[wind_model] 或 pip install package_a[wind_model]的情况
    if any('wind_model' in arg for arg in sys.argv):
        # 定义你的f2py扩展模块
        wind_ext = Extension(
            name='package_a.wind_model',  # 模块的完整名称，要和你的包结构对应
            sources=['src/wind_model.f90'],  # Fortran源码路径
            f2py_options=['--quiet'],  # 可选的f2py编译参数
        )
        extensions.append(wind_ext)
    return extensions

# 主setup配置
setup(
    name='package_a',
    version='0.1.0',
    packages=['package_a'],
    # 定义extra依赖：安装wind_model时需要的依赖（比如numpy是f2py必需的）
    extras_require={
        'wind_model': ['numpy>=1.21.0'],
    },
    # 传入动态生成的扩展模块列表
    ext_modules=get_extensions(),
)

2. 用户的使用方式

• 如果用户不需要这个可选模块，直接运行常规安装命令即可，不会编译Fortran代码：

  pip install .
  

• 如果用户需要安装并编译wind_model模块，只需要指定对应的extra：

  pip install .[wind_model]
  

  这时候setup.py会自动检测到extra参数，添加Fortran扩展模块的编译步骤，同时安装extras_require中指定的numpy依赖。

额外注意事项 #

• 编译器依赖：如果用户选择安装wind_model，必须确保系统已经安装了Fortran编译器（比如gfortran），否则编译会失败。你可以在项目文档中明确说明这一点，或者在setup.py里加入简单的编译器检查逻辑，抛出更友好的错误提示。
• 参数兼容性：这种方式适配大多数pip的使用场景，包括从PyPI安装（pip install package_a[wind_model]）和本地安装，只要命令行参数中包含wind_model关键词就能触发编译。
• 灵活性扩展：如果你的可选模块有更多复杂的编译需求，还可以在get_extensions()函数里添加更多逻辑，比如根据操作系统调整编译参数，或者检查依赖是否存在等。

这样就完美解决了你之前的问题——不用拆分包，用户只用一次命令就能选择是否安装可选的扩展模块，没有额外的操作成本。

备注：内容来源于stack exchange，提问作者Jehan Dastoor