Python中根据环境为枚举定义不同值的方案求助

这确实是mypy处理动态导入时的常见痛点——虽然运行时逻辑没问题，但静态类型检查器没法很好地追踪这种分支导入的类型。下面给你几个更优雅的解决方案，既能满足环境区分需求，又能让mypy乖乖通过检查：

方案一：用单个枚举类结合环境变量分支定义 #

完全不需要维护两个枚举类，直接在一个Ports枚举里根据环境变量设置不同值。这种方式mypy可以正常解析条件分支，不会报错：

from enum import Enum
import os

class Ports(Enum):
    if os.getenv("APPLICATION_CONFIG") == "production":
        service_1 = 3000
        service_2 = 3001
    else:
        service_1 = 5000
        service_2 = 5001

优点： #

• 只需要维护一个枚举类，减少代码冗余
• 类型检查完全友好，mypy能识别所有可能的枚举值
• 启动时就确定枚举值，符合配置初始化的常规逻辑

方案二：用TYPE_CHECKING辅助mypy识别类型 #

如果你必须保留两个分开的枚举类，可以借助typing.TYPE_CHECKING常量——这个常量在运行时是False，但mypy会把它当作True来处理，这样就能给类型检查器提供明确的类型提示：

import os
from typing import TYPE_CHECKING

# 给mypy看的分支，运行时不会执行
if TYPE_CHECKING:
    from .services_ports import Ports
else:
    # 运行时实际导入逻辑
    if os.getenv("APPLICATION_CONFIG") == "production":
        from .services_ports import Ports
    else:
        from .services_ports import PortsDev as Ports

注意事项： #

• 必须保证Ports和PortsDev的结构完全一致（成员名称、类型都相同），否则mypy可能会误报类型不一致
• 这种方式保留了你原有的代码结构，适合已经有两个枚举类且不想改动的场景

方案三：用Literal类型替代枚举（灵活替代方案） #

如果枚举的核心只是固定值约束，也可以用Literal类型来定义端口常量，同样能让mypy进行类型检查：

import os
from typing import Literal

# 根据环境定义字面量类型
if os.getenv("APPLICATION_CONFIG") == "production":
    Service1Port = Literal[3000]
    Service2Port = Literal[3001]
else:
    Service1Port = Literal[5000]
    Service2Port = Literal[5001]

# 定义常量类
class Ports:
    service_1: Service1Port = 3000 if os.getenv("APPLICATION_CONFIG") == "production" else 5000
    service_2: Service2Port = 3001 if os.getenv("APPLICATION_CONFIG") == "production" else 5001

优点： #

• 比枚举更轻量，不需要继承Enum类
• Literal类型能严格约束变量只能取指定值，类型检查同样严格
• 适合不需要枚举额外功能（比如成员遍历、值转名称等）的场景

内容的提问来源于stack exchange，提问作者Maurício Silva