32 lines
1.8 KiB
Python
32 lines
1.8 KiB
Python
"""Original occurrence, plan and reporting time contracts."""
|
|
from datetime import datetime
|
|
from typing import Literal
|
|
from pydantic import BaseModel, Field, model_validator
|
|
|
|
|
|
class PackageTimePoint(BaseModel):
|
|
occurred_at: datetime | None = Field(description="发生时间；未知必须为null且time_precision=unknown，已知必须带时区，不用报道时间代替")
|
|
time_precision: Literal['minute', 'day', 'month', 'unknown'] = Field(description='occurred_at为null时必须unknown；非null时按原文精度选择minute/day/month')
|
|
planned_at: datetime | None = Field(default=None, description='计划发生时间，不证明已发生')
|
|
planned_precision: Literal['minute', 'day', 'month', 'unknown'] = 'unknown'
|
|
period_text: str = Field(default='', description='数据统计期间原文')
|
|
time_text: str
|
|
reported_at: datetime | None
|
|
description: str = Field(min_length=1)
|
|
verification: Literal['supported', 'partial', 'conflicting', 'insufficient', 'refuted']
|
|
evidence_refs: list[dict] = Field(min_length=1)
|
|
|
|
@model_validator(mode='after')
|
|
def time_consistency(self):
|
|
if (self.planned_at is None) != (self.planned_precision == 'unknown'):
|
|
raise ValueError('计划时间与精度不一致')
|
|
if self.planned_at and self.planned_at.tzinfo is None:
|
|
raise ValueError('计划时间必须有时区')
|
|
if (self.occurred_at is None) != (self.time_precision == 'unknown'):
|
|
raise ValueError('发生时间与精度不一致')
|
|
if self.occurred_at and self.occurred_at.tzinfo is None:
|
|
raise ValueError('发生时间必须有时区')
|
|
if self.reported_at and self.reported_at.tzinfo is None:
|
|
raise ValueError('报道时间必须有时区')
|
|
return self
|