Files
ca_auto_table/app/apis/country/info/schema.py

85 lines
2.7 KiB
Python
Raw Normal View History

2025-11-18 16:46:04 +08:00
from datetime import datetime, timezone, timedelta
from pydantic import BaseModel, Field, computed_field
from typing import List
from uuid import UUID
from utils.time_tool import TimestampModel
CHINA_TZ = timezone(timedelta(hours=8))
class Base(BaseModel):
"""
基础地址信息模型
包含地址相关的通用字段供创建与输出模型复用
"""
firstname: str = Field(..., description='')
lastname: str = Field(..., description='')
full_name: str = Field(..., description='全名')
birthday: str = Field(..., description='生日')
street_address: str = Field(..., description='街道地址')
city: str = Field(..., description='城市')
phone: str = Field(..., description='电话')
zip_code: str = Field(..., description='邮编')
state_fullname: str = Field(..., description='州全称')
status: bool = Field(False, description='状态')
class Create(Base):
"""
创建请求模型
"""
pass
class Update(BaseModel):
"""
更新请求模型支持部分更新
"""
firstname: str | None = Field(None, description='')
lastname: str | None = Field(None, description='')
full_name: str | None = Field(None, description='全名')
birthday: str | None = Field(None, description='生日')
street_address: str | None = Field(None, description='街道地址')
city: str | None = Field(None, description='城市')
phone: str | None = Field(None, description='电话')
zip_code: str | None = Field(None, description='邮编')
state_fullname: str | None = Field(None, description='州全称')
status: bool | None = Field(None, description='状态')
class Out(TimestampModel, Base):
"""
输出模型
"""
code: int = Field(200, description='状态码')
message: str = Field('成功', description='提示信息')
id: UUID = Field(..., description='ID')
create_time: datetime = Field(..., description='创建时间')
update_time: datetime = Field(..., description='更新时间')
@computed_field
@property
def create_time_cn(self) -> str:
return self.create_time.astimezone(CHINA_TZ).strftime("%Y-%m-%d %H:%M:%S")
@computed_field
@property
def update_time_cn(self) -> str:
return self.update_time.astimezone(CHINA_TZ).strftime("%Y-%m-%d %H:%M:%S")
class Config:
from_attributes = True
class OutList(BaseModel):
"""
列表输出模型
"""
code: int = Field(200, description='状态码')
message: str = Field('成功', description='提示信息')
count: int = Field(0, description='总数')
num: int = Field(0, description='当前数量')
items: List[Out] = Field([], description='列表数据')