"""Structured correction and time contracts shared by news stages."""
from datetime import datetime
from typing import Literal
from pydantic import BaseModel, Field, model_validator

class PackageInput(BaseModel):
    id: int = Field(gt=0)
    revision: int = Field(ge=0)


class Assignment(BaseModel):
    analysis_id: int = Field(gt=0)
    evidence: str = Field(min_length=1, max_length=500)


class Replacement(BaseModel):
    keep_id: int | None
    title: str = Field(min_length=1, max_length=300)
    fact: str = Field(min_length=1)
    category: str = Field(min_length=1, max_length=64)
    tags: list[str] = Field(min_length=1, max_length=20)
    content_kind: Literal['report', 'statement', 'data', 'opinion', 'explanation', 'other']
    assignments: list[Assignment] = Field(min_length=1)


class Correction(BaseModel):
    packages: list[PackageInput] = Field(min_length=1)
    replacements: list[Replacement] = Field(min_length=1)
    reason: str = Field(min_length=1)


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