如何用Pydantic建模AWS服务及其资源？

Enum方案的合理性 #

Enum完全适合你的场景——AWS服务名称、各服务下的资源类型都是固定可选值，用Enum能确保输入的合法性，避免拼写错误，而且Pydantic原生支持Enum类型的验证，无需额外写复杂逻辑，是最优选择。

资源层级建模实现步骤 #

1. 定义服务与资源的Enum

先把固定的服务和对应资源用Enum明确下来：

from enum import Enum
from pydantic import BaseModel, Field, validator

# AWS服务枚举
class AWSService(str, Enum):
    EC2 = "ec2"
    S3 = "s3"
    RDS = "rds"

# 各服务对应资源枚举
class EC2Resource(str, Enum):
    INSTANCE = "instance"
    SECURITY_GROUP = "security-group"
    VOLUME = "volume"

class S3Resource(str, Enum):
    BUCKET = "bucket"
    OBJECT = "object"
    POLICY = "policy"

class RDSResource(str, Enum):
    DB_INSTANCE = "db-instance"
    DB_SUBNET_GROUP = "db-subnet-group"

2. 用Pydantic模型关联服务与资源

有两种实用的建模方式，根据你的场景选择：

方式一：分服务的专用模型（适合服务结构差异大的场景）

每个服务单独定义模型，强制绑定对应资源类型：

class EC2Config(BaseModel):
    service: AWSService = Field(AWSService.EC2, const=True)
    resources: list[EC2Resource]

class S3Config(BaseModel):
    service: AWSService = Field(AWSService.S3, const=True)
    resources: list[S3Resource]

class RDSConfig(BaseModel):
    service: AWSService = Field(AWSService.RDS, const=True)
    resources: list[RDSResource]

# 总配置模型
class AWSConfig(BaseModel):
    services: list[EC2Config | S3Config | RDSConfig]

加载YAML时，Pydantic会自动根据service字段匹配对应模型，自动验证资源类型是否合法。

方式二：通用模型+验证器（适合服务结构统一的场景）

用一个通用模型，通过验证器动态关联服务与资源：

class ServiceResourceConfig(BaseModel):
    service: AWSService
    resources: list[str]

    @validator("resources")
    def match_service_resources(cls, resources, values):
        service = values.get("service")
        if not service:
            return resources
        
        # 建立服务与资源枚举的映射
        resource_map = {
            AWSService.EC2: EC2Resource,
            AWSService.S3: S3Resource,
            AWSService.RDS: RDSResource
        }
        
        # 验证每个资源是否属于当前服务的合法选项
        for res in resources:
            try:
                resource_map[service](res)
            except ValueError:
                raise ValueError(f"资源'{res}'不属于服务'{service.value}'的合法类型")
        return resources

# 总配置模型
class AWSConfig(BaseModel):
    services: list[ServiceResourceConfig]

Enum vs Set/Frozenset的对比 #

• Enum的核心优势：
  • 自带类型安全验证，自动拦截非法值
  • 代码可读性高，通过EC2Resource.INSTANCE引用，避免硬编码字符串
  • 原生支持Pydantic的序列化/反序列化，YAML字符串可直接转成Enum实例，输出时自动转回字符串
• Set/Frozenset的适用场景：
  • 仅当资源类型是动态可扩展（允许用户自定义非标准资源）时才考虑使用
  • 但Set无法提供Enum的类型安全和可读性，需要手动编写更多验证逻辑

综上，如果你的AWS服务和资源类型是固定的（绝大多数场景都是如此），Enum是绝对更优的选择。

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