73 lines
1.5 KiB
Python
73 lines
1.5 KiB
Python
from datetime import datetime
|
|
|
|
from pydantic import BaseModel
|
|
|
|
|
|
class JokeBase(BaseModel):
|
|
title: str
|
|
content: str
|
|
type_ids: list[int] | None = None
|
|
crowd_ids: list[int] | None = None
|
|
|
|
|
|
class JokeCreate(JokeBase):
|
|
pass
|
|
|
|
|
|
class JokeUpdate(BaseModel):
|
|
title: str | None = None
|
|
content: str | None = None
|
|
polished_content: str | None = None
|
|
type_ids: list[int] | None = None
|
|
crowd_ids: list[int] | None = None
|
|
status: str | None = None
|
|
|
|
|
|
class JokeResponse(JokeBase):
|
|
id: int
|
|
status: str
|
|
polished_content: str | None = None
|
|
view_count: int
|
|
like_count: int
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
type_names: list[str] | None = None
|
|
crowd_names: list[str] | None = None
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class PaginatedJokeResponse(BaseModel):
|
|
items: list[JokeResponse]
|
|
total: int
|
|
page: int
|
|
page_size: int
|
|
|
|
|
|
# ============================================================
|
|
# 笑话生成器 Schema
|
|
# ============================================================
|
|
|
|
class GenerateRequest(BaseModel):
|
|
"""笑话生成请求"""
|
|
keywords: list[str] = []
|
|
scenarios: list[str] = []
|
|
|
|
class Config:
|
|
json_schema_extra = {
|
|
"example": {
|
|
"keywords": ["加班"],
|
|
"scenarios": ["职场"]
|
|
}
|
|
}
|
|
|
|
|
|
class GenerateResponse(BaseModel):
|
|
"""笑话生成响应"""
|
|
title: str
|
|
content: str
|
|
created_at: datetime | None = None
|
|
|
|
class Config:
|
|
from_attributes = True |