Pydantic继承类验证问题：如何在changes字段验证器中访问effective_date字段

你遇到的问题是因为普通的字段级@validator只能访问当前被验证的字段，没办法直接获取模型里的其他字段（比如effective_date）。要实现同时访问多个字段的验证逻辑，需要使用根验证器（Root Validator），具体写法取决于你使用的Pydantic版本，下面分两种情况给出解决方案：

如果你使用的是Pydantic v1 #

Pydantic v1中可以用@root_validator装饰器来定义根验证器，设置pre=False（默认值，确保在所有字段验证完成后运行），这样就能拿到所有已验证的字段值：

from pydantic import root_validator

class PublicCreateBasketV2ApiModel(BaseModel):
    agency: str
    changes: conlist(ChangeApiModel, min_items=1)
    effective_date: Date
    email: Optional[EmailStr] = None
    
    class Config:
        extra = "forbid"

    @root_validator(pre=False)
    def validate_changes_with_effective_date(cls, values):
        # 从values字典中获取已验证的字段值
        changes = values.get("changes")
        effective_date = values.get("effective_date")
        
        # 避免前面字段验证失败时抛出额外错误，直接返回values让Pydantic处理原有错误
        if not changes or not effective_date:
            return values
        
        # 原有的分类逻辑
        lapse_changes, renewal_changes, cancellation_changes = [], [], []
        for change in changes:
            if change.type == ChangeType.LAPSE:
                lapse_changes.append(change)
            elif change.type == ChangeType.RENEW:
                renewal_changes.append(change)
            elif change.type == ChangeType.CANCEL:
                cancellation_changes.append(change)
        
        # 把effective_date传入验证函数
        validate_lapses(lapse_changes, effective_date)
        validate_renewals(renewal_changes, effective_date)
        validate_cancellations(cancellation_changes, effective_date)
        
        return values

如果你使用的是Pydantic v2 #

Pydantic v2废弃了root_validator，改用@model_validator，选择mode="after"模式（对应v1的pre=False），此时可以直接通过self访问所有已验证的字段：

from pydantic import model_validator

class PublicCreateBasketV2ApiModel(BaseModel):
    agency: str
    changes: conlist(ChangeApiModel, min_items=1)
    effective_date: Date
    email: Optional[EmailStr] = None
    
    class Config:
        extra = "forbid"

    @model_validator(mode="after")
    def validate_changes_with_effective_date(self):
        # 直接通过self访问字段
        changes = self.changes
        effective_date = self.effective_date
        
        # 原有的分类逻辑
        lapse_changes, renewal_changes, cancellation_changes = [], [], []
        for change in changes:
            if change.type == ChangeType.LAPSE:
                lapse_changes.append(change)
            elif change.type == ChangeType.RENEW:
                renewal_changes.append(change)
            elif change.type == ChangeType.CANCEL:
                cancellation_changes.append(change)
        
        # 把effective_date传入验证函数
        validate_lapses(lapse_changes, effective_date)
        validate_renewals(renewal_changes, effective_date)
        validate_cancellations(cancellation_changes, effective_date)
        
        return self

额外注意事项 #

1. 验证器定义在父类PublicCreateBasketV2ApiModel中时，子类AuthenticatedCreateBasketV2ApiModel会自动继承这个验证逻辑，无需重复定义。
2. 记得更新你的validate_lapses、validate_renewals、validate_cancellations函数，让它们接收effective_date参数，这样就能在内部使用这个字段完成验证了。
3. 如果你的验证逻辑只需要作用于子类AuthenticatedCreateBasketV2ApiModel，也可以把根验证器直接定义在子类中，同样能正常访问effective_date字段。

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