在Pydantic中优化Activity子类Lake的实现方式咨询

我定义了如下的Activity Pydantic类：

from datetime import datetime
from typing import Optional

from pydantic import BaseModel, Extra, Field

class Activity(BaseModel, extra=Extra.ignore, frozen=True):
    """Stores ship specific information."""

    id: ActivityId
    owner: Optional[str] = Field(None, min_length=2, max_length=2)
    ship_number: Optional[int] = None
    origin: str = Field(min_length=3, max_length=3)
    origin_sub_location: Optional[str] = None
    destination: str = Field(min_length=3, max_length=3)
    destination_sub_location: Optional[str] = None
    start_time: datetime
    end_time: datetime

现需继承该类创建Lake子类，添加Lake独有的is_overnight字段。当前已实现的Lake类代码如下：

from datetime import datetime
from typing import Any, Optional

from pydantic import PrivateAttr

class Lake(Activity):
    _is_overnight: bool = PrivateAttr(False)

    def __init__(
        self,
        start_time: datetime,
        location: str,
        is_overnight: bool,
        sub_location: Optional[str] = None,
        **kwargs: Any,
    ):
        # Automatically set end_time, origin, and destination to the same as start_time and location
        super().__init__(
            start_time=start_time,
            end_time=start_time,  # same as start_time for lakes
            origin=location,
            destination=location,
            origin_sub_location=sub_location,
            destination_sub_location=sub_location,
            cancelled=False,
            id=f"LAKE_{location}{sub_location}_{start_time.strftime('%Y-%m-%d %H:%M')}",
            **kwargs,
        )
        self._is_overnight = is_overnight

    @property
    def is_overnight(self) -> bool:
        return self._is_overnight

请问是否存在更简洁的实现方式？

有几种方式可以简化Lake类的实现，核心是利用Pydantic的内置特性替代手动编写__init__逻辑，同时优化is_overnight的定义：

方案：使用model_validator自动填充父类字段 #

from datetime import datetime
from typing import Optional
from pydantic import Field, model_validator

class Lake(Activity):
    is_overnight: bool = Field(default=False)
    # Lake专属输入参数，用于自动生成父类所需字段
    location: str = Field(min_length=3, max_length=3)
    sub_location: Optional[str] = None

    @model_validator(mode='before')
    def auto_fill_activity_fields(cls, values):
        loc = values['location']
        sub_loc = values.get('sub_location', '')
        
        # 批量填充父类所需的重复字段
        values.update({
            'origin': loc,
            'destination': loc,
            'origin_sub_location': sub_loc,
            'destination_sub_location': sub_loc,
            'end_time': values['start_time'],
            'id': f"LAKE_{loc}{sub_loc}_{values['start_time'].strftime('%Y-%m-%d %H:%M')}"
        })
        
        # 移除Lake专属输入字段（可选，父类已设置extra=Extra.ignore）
        values.pop('location')
        values.pop('sub_location')
        return values

简化说明 #

• 抛弃手动__init__：用model_validator符合Pydantic声明式设计，减少样板代码，同时保留自动校验能力。
• 简化is_overnight定义：直接将其作为公共字段，无需用PrivateAttr加property包装——父类的frozen=True约束会被Pydantic自动处理，字段仍保持不可变性。
• 减少重复逻辑：通过update批量填充重复字段，代码更紧凑易读。
• 清晰分离职责：字段定义与自动填充逻辑分离，维护成本更低。

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