80 lines
2.0 KiB
Python
80 lines
2.0 KiB
Python
import uuid
|
|
|
|
from datetime import date, datetime
|
|
from typing import List, Optional
|
|
|
|
from pydantic import BaseModel, Field, validator
|
|
|
|
from . import to_camel
|
|
|
|
|
|
class GuestBookIn(BaseModel):
|
|
name: str
|
|
phone: str
|
|
address: Optional[str]
|
|
pax: int = Field(ge=0)
|
|
|
|
class Config:
|
|
fields = {"id_": "id"}
|
|
anystr_strip_whitespace = True
|
|
alias_generator = to_camel
|
|
|
|
|
|
class GuestBook(GuestBookIn):
|
|
id_: uuid.UUID
|
|
date_: datetime
|
|
|
|
@validator("date_", pre=True)
|
|
def parse_date(cls, value):
|
|
if isinstance(value, datetime):
|
|
return value
|
|
return datetime.strptime(value, "%d-%b-%Y %H:%M")
|
|
|
|
class Config:
|
|
fields = {"id_": "id"}
|
|
anystr_strip_whitespace = True
|
|
alias_generator = to_camel
|
|
json_encoders = {datetime: lambda v: v.strftime("%d-%b-%Y %H:%M")}
|
|
|
|
|
|
class GuestBookListItem(BaseModel):
|
|
id_: uuid.UUID
|
|
serial: int
|
|
name: str
|
|
phone: str
|
|
pax: int
|
|
date_: datetime
|
|
status: Optional[str]
|
|
table_id: Optional[uuid.UUID]
|
|
voucher_id: Optional[uuid.UUID]
|
|
table_name: Optional[str]
|
|
|
|
@validator("date_", pre=True)
|
|
def parse_date(cls, value):
|
|
if isinstance(value, datetime):
|
|
return value
|
|
return datetime.strptime(value, "%d-%b-%Y %H:%M")
|
|
|
|
class Config:
|
|
fields = {"id_": "id"}
|
|
anystr_strip_whitespace = True
|
|
alias_generator = to_camel
|
|
json_encoders = {datetime: lambda v: v.strftime("%d-%b-%Y %H:%M"), date: lambda v: v.strftime("%d-%b-%Y")}
|
|
|
|
|
|
class GuestBookList(BaseModel):
|
|
date_: date
|
|
list: List[GuestBookListItem]
|
|
|
|
@validator("date_", pre=True)
|
|
def parse_date(cls, value):
|
|
if isinstance(value, date):
|
|
return value
|
|
return datetime.strptime(value, "%d-%b-%Y").date()
|
|
|
|
class Config:
|
|
fields = {"id_": "id"}
|
|
anystr_strip_whitespace = True
|
|
alias_generator = to_camel
|
|
json_encoders = {datetime: lambda v: v.strftime("%d-%b-%Y %H:%M"), date: lambda v: v.strftime("%d-%b-%Y")}
|