85 lines
2.1 KiB
Python
85 lines
2.1 KiB
Python
from datetime import datetime
|
|
|
|
from pydantic import BaseModel, model_validator
|
|
|
|
|
|
# ---------- Profile ----------
|
|
|
|
class ProfileCreate(BaseModel):
|
|
nickname: str
|
|
country: str | None = None
|
|
|
|
|
|
class ProfileUpdate(BaseModel):
|
|
nickname: str | None = None
|
|
country: str | None = None
|
|
|
|
|
|
class ProfileOut(BaseModel):
|
|
id: int
|
|
nickname: str
|
|
country: str | None
|
|
created_at: datetime
|
|
|
|
model_config = {"from_attributes": True}
|
|
|
|
@model_validator(mode="after")
|
|
def _resolve_image_urls(self) -> "PostcardOut":
|
|
from app.storage import resolve_url
|
|
self.image_front = resolve_url(self.image_front) or None
|
|
self.image_back = resolve_url(self.image_back) or None
|
|
return self
|
|
|
|
|
|
# ---------- Postcard ----------
|
|
|
|
class PostcardCreate(BaseModel):
|
|
card_number: str
|
|
status: str
|
|
recipient_name: str | None = None
|
|
country_from: str | None = None
|
|
country_to: str | None = None
|
|
send_time: datetime | None = None
|
|
arrival_time: datetime | None = None
|
|
sender_name: str | None = None
|
|
receive_time: datetime | None = None
|
|
|
|
|
|
class PostcardUpdate(BaseModel):
|
|
card_number: str | None = None
|
|
status: str | None = None
|
|
recipient_name: str | None = None
|
|
country_from: str | None = None
|
|
country_to: str | None = None
|
|
send_time: datetime | None = None
|
|
arrival_time: datetime | None = None
|
|
sender_name: str | None = None
|
|
receive_time: str | None = None
|
|
|
|
|
|
class PostcardOut(BaseModel):
|
|
id: int
|
|
profile_id: int
|
|
card_number: str
|
|
status: str
|
|
recipient_name: str | None
|
|
country_from: str | None
|
|
country_to: str | None
|
|
send_time: datetime | None
|
|
arrival_time: datetime | None
|
|
sender_name: str | None
|
|
receive_time: datetime | None
|
|
image_front: str | None
|
|
image_back: str | None
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
|
|
model_config = {"from_attributes": True}
|
|
|
|
@model_validator(mode="after")
|
|
def _resolve_image_urls(self) -> "PostcardOut":
|
|
from app.storage import resolve_url
|
|
self.image_front = resolve_url(self.image_front) or None
|
|
self.image_back = resolve_url(self.image_back) or None
|
|
return self
|