Pydantic V2 验证器使用中的常见误区与正确实践

在Pydantic V2版本中，验证器的使用方式发生了显著变化，许多从V1迁移过来的开发者容易陷入一些使用误区。本文将通过一个典型案例，深入分析Pydantic V2验证器的正确使用方式。

验证器参数的变化

Pydantic V2对验证器函数参数进行了重大调整。在V1版本中，字段验证器可以接收一个values参数，该参数是一个包含其他字段值的字典。但在V2版本中，这个参数被替换为ValidationInfo对象，它提供了更丰富的验证上下文信息。

典型案例分析

考虑一个促销活动时间验证的场景：我们需要确保促销结束时间不早于开始时间。在V1中可能会这样实现：

@validator('end_date_time')
def check_enddate_after_startdate(cls, end_date_time, values):
    start_date_time = values['start_date_time']
    if end_date_time >= start_date_time:
        return end_date_time
    raise ValueError("结束时间不能早于开始时间")

迁移到V2时，开发者容易犯两个错误：

1. 仍然期望第二个参数是字典形式的字段值
2. 直接调用验证器函数进行测试

V2的正确实现方式

在Pydantic V2中，正确的实现应该使用field_validator装饰器，并正确处理ValidationInfo对象：

from pydantic import field_validator, ValidationInfo

@field_validator('end_date_time')
@classmethod
def check_enddate_after_startdate(cls, end_date_time: datetime, info: ValidationInfo):
    start_date_time = info.data['start_date_time']
    if end_date_time >= start_date_time:
        return end_date_time
    raise ValueError("结束时间不能早于开始时间")

模型级验证的替代方案

对于涉及多个字段的复杂验证逻辑，更推荐使用model_validator：

from pydantic import model_validator

@model_validator(mode='after')
def validate_dates(self):
    if self.end_date_time >= self.start_date_time:
        return self
    raise ValueError("结束时间不能早于开始时间")

这种方式的优势在于：

1. 可以直接访问模型实例的属性
2. 验证逻辑更清晰直观
3. 避免了字段访问的安全问题

测试策略的调整

在V2中，不应该直接调用验证器函数进行测试，而应该通过实例化模型来触发验证：

def test_date_validation():
    # 正确的时间顺序不应引发异常
    ForecastRequestDTO(
        promotion_id="test",
        is_cumulative_promotion=True,
        start_date_time=datetime(2024, 1, 1),
        end_date_time=datetime(2024, 1, 2)
    )
    
    # 错误的时间顺序应引发验证错误
    with pytest.raises(ValueError):
        ForecastRequestDTO(
            promotion_id="test",
            is_cumulative_promotion=True,
            start_date_time=datetime(2024, 1, 2),
            end_date_time=datetime(2024, 1, 1)
        )

迁移建议

从V1迁移到V2时，建议开发者：

1. 全面检查所有验证器函数，更新参数签名
2. 将简单的字段级验证升级为模型级验证
3. 重构测试用例，改为通过模型实例化触发验证
4. 考虑添加validate_assignment=True配置，确保属性修改时也触发验证

通过遵循这些最佳实践，可以确保验证逻辑在Pydantic V2中正确工作，同时提高代码的可维护性和可测试性。