47 lines
1.2 KiB
Python
47 lines
1.2 KiB
Python
from __future__ import annotations
|
|
|
|
from pydantic import BaseModel, Field, field_validator
|
|
|
|
from app.enums.notification import NotificationType
|
|
from app.schemas.common import NaiveDatetime
|
|
|
|
|
|
class NotificationCreditsOut(BaseModel):
|
|
"""通知轮询接口携带的当前用户积分信息。"""
|
|
|
|
balance: float = Field(..., description="当前用户最新积分余额")
|
|
|
|
|
|
class NotificationOut(BaseModel):
|
|
id: str
|
|
title: str
|
|
content: str
|
|
type: NotificationType
|
|
is_read: bool
|
|
created_at: NaiveDatetime
|
|
|
|
@field_validator("type", mode="before")
|
|
@classmethod
|
|
def normalize_type(cls, value):
|
|
"""兼容历史数据或后台自定义通知类型，避免响应模型校验失败。"""
|
|
if isinstance(value, NotificationType):
|
|
return value
|
|
if value is None:
|
|
return NotificationType.UNKNOWN
|
|
try:
|
|
return NotificationType(str(value))
|
|
except ValueError:
|
|
return NotificationType.UNKNOWN
|
|
|
|
model_config = {"from_attributes": True}
|
|
|
|
|
|
class NotificationListOut(BaseModel):
|
|
items: list[NotificationOut]
|
|
total: int
|
|
credits: NotificationCreditsOut
|
|
|
|
|
|
class UnreadCountOut(BaseModel):
|
|
count: int
|