5daae76147
- Implement aider-watch v2 event schema with 7 event types - Enforce timezone-aware timestamps via field_validator - Batch schema supports up to 50 events per request - Frozen + forbid extra fields (defensive engineering) - Fix broken src.* imports in models package (incident.py, __init__.py) Task A3 complete: 7/7 tests passing
36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
"""aider-watch event pydantic schemas。由 Mac client POST 送來。"""
|
|
from __future__ import annotations
|
|
from datetime import datetime
|
|
from typing import Any, Literal
|
|
from pydantic import BaseModel, ConfigDict, Field, field_validator
|
|
|
|
EventType = Literal[
|
|
"session_start", "file_edit", "error", "commit",
|
|
"silent_timeout", "session_end", "raw",
|
|
]
|
|
|
|
|
|
class AiderEventIn(BaseModel):
|
|
"""單筆 aider event 輸入 schema。"""
|
|
model_config = ConfigDict(frozen=True, extra="forbid")
|
|
|
|
ts: datetime
|
|
session_id: str = Field(min_length=1, max_length=64)
|
|
host: str = Field(default="ogt-mac", max_length=64)
|
|
type: EventType
|
|
payload: dict[str, Any] = Field(default_factory=dict)
|
|
|
|
@field_validator("ts")
|
|
@classmethod
|
|
def _ensure_tz(cls, v: datetime) -> datetime:
|
|
if v.tzinfo is None:
|
|
raise ValueError("ts must be timezone-aware")
|
|
return v
|
|
|
|
|
|
class AiderBatchIn(BaseModel):
|
|
"""批次 event 輸入（最多 50 筆/次）。"""
|
|
model_config = ConfigDict(frozen=True, extra="forbid")
|
|
|
|
events: list[AiderEventIn] = Field(min_length=1, max_length=50)
|