File size: 1,273 Bytes
80feb1b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 |
from pydantic import BaseModel
from typing import List, Optional
from datetime import datetime
from .dish import Dish
class OrderItemBase(BaseModel):
dish_id: int
quantity: int = 1
remarks: Optional[str] = None
class OrderItemCreate(OrderItemBase):
pass
class OrderItem(OrderItemBase):
id: int
order_id: int
created_at: datetime
dish: Optional[Dish] = None
# Add dish_name property to ensure it's always available
@property
def dish_name(self) -> str:
if self.dish:
return self.dish.name
return "Unknown Dish"
class Config:
from_attributes = True # Updated from orm_mode for Pydantic V2
class OrderBase(BaseModel):
table_number: int
unique_id: str
class OrderCreate(OrderBase):
items: List[OrderItemCreate]
username: Optional[str] = None
password: Optional[str] = None
class OrderUpdate(BaseModel):
status: str
class Order(OrderBase):
id: int
status: str
created_at: datetime
updated_at: datetime
items: List[OrderItem] = []
person_id: Optional[int] = None
person_name: Optional[str] = None
visit_count: Optional[int] = None
class Config:
from_attributes = True # Updated from orm_mode for Pydantic V2
|