nex_docus/backend/app/schemas/notification.py

"""
通知相关的 Pydantic Schema
"""
from pydantic import BaseModel, Field
from typing import Optional, List, Union
from datetime import datetime


class NotificationBase(BaseModel):
    type: str = "info"
    category: str = "system"
    title: str
    content: Optional[str] = None
    link: Optional[str] = None


class NotificationCreate(NotificationBase):
    user_id: int


class NotificationUpdate(BaseModel):
    is_read: Optional[bool] = None


class NotificationResponse(NotificationBase):
    id: Union[str, int]
    user_id: int
    is_read: bool
    created_at: Union[datetime, float]  # Redis returns float timestamp
    read_at: Optional[datetime] = None

    class Config:
        from_attributes = True


class UnreadCountResponse(BaseModel):
    unread_count: int