37 lines
784 B
Python
37 lines
784 B
Python
"""
|
|
通知相关的 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 |