refactor: 重整 schemas 作用域并统一用户上下文模型

2026-03-13 01:01:54 +08:00
parent f201babb48
commit fb3c649db7
42 changed files with 4205 additions and 2013 deletions
@@ -0,0 +1,96 @@
+from __future__ import annotations
+
+import json
+from enum import Enum
+from typing import ClassVar, Literal, Union
+from uuid import UUID
+
+from pydantic import BaseModel, ConfigDict, Field
+
+
+class AttachmentType(str, Enum):
+    DOCUMENT = "document"
+    REMINDER = "reminder"
+
+
+class ScheduleItemMetadataAttachment(BaseModel):
+    model_config: ClassVar[ConfigDict] = ConfigDict(extra="forbid")
+
+    name: str
+    type: AttachmentType
+    visible_to: list[UUID] = Field(default_factory=list)
+    url: str | None = None
+    note: str | None = None
+    content: str | None = None
+
+
+class ScheduleItemMetadata(BaseModel):
+    model_config: ClassVar[ConfigDict] = ConfigDict(extra="forbid")
+
+    color: str | None = Field(default=None, pattern=r"^#[0-9A-Fa-f]{6}$")
+    location: str | None = None
+    notes: str | None = None
+    attachments: list[ScheduleItemMetadataAttachment] = Field(default_factory=list)
+    reminder_minutes: int | None = Field(default=None, ge=0, le=10080)
+    version: Literal[1] = 1
+
+
+class ScheduleItemStatus(str, Enum):
+    ACTIVE = "active"
+    COMPLETED = "completed"
+    CANCELED = "canceled"
+    ARCHIVED = "archived"
+
+
+class ScheduleItemSourceType(str, Enum):
+    MANUAL = "manual"
+    IMPORTED = "imported"
+    AGENT_GENERATED = "agent_generated"
+
+
+class CalendarInviteContent(BaseModel):
+    model_config: ClassVar[ConfigDict] = ConfigDict(extra="forbid")
+
+    type: Literal["invite"]
+    permission: int = Field(..., description="权限: 1=view, 4=edit, 8=invite")
+    action: Literal["pending"] = "pending"
+
+
+class CalendarUpdateContent(BaseModel):
+    model_config: ClassVar[ConfigDict] = ConfigDict(extra="forbid")
+
+    type: Literal["update"]
+    title: str = Field(..., description="事件标题")
+    action: Literal["updated"] = "updated"
+
+
+class CalendarDeleteContent(BaseModel):
+    model_config: ClassVar[ConfigDict] = ConfigDict(extra="forbid")
+
+    type: Literal["delete"]
+    title: str = Field(..., description="事件标题")
+    action: Literal["deleted"] = "deleted"
+
+
+CalendarContent = Union[
+    CalendarInviteContent,
+    CalendarUpdateContent,
+    CalendarDeleteContent,
+]
+
+
+def parse_calendar_content(content: str | None) -> CalendarContent | None:
+    if not content:
+        return None
+    try:
+        data = json.loads(content)
+        content_type = data.get("type")
+        if content_type == "invite":
+            return CalendarInviteContent(**data)
+        if content_type == "update":
+            return CalendarUpdateContent(**data)
+        if content_type == "delete":
+            return CalendarDeleteContent(**data)
+        raise ValueError(f"Unknown calendar content type: {content_type}")
+    except Exception:
+        return None