54 lines
1.1 KiB
Python

from datetime import datetime
from typing import List, Optional
from pydantic import BaseModel, Field
from app.models.order import OrderStatus
# Order Item Schemas
class OrderItemBase(BaseModel):
product_id: int
quantity: int = Field(..., gt=0)
unit_price: float = Field(..., gt=0)
class OrderItemCreate(OrderItemBase):
pass
class OrderItemInDBBase(OrderItemBase):
id: int
order_id: int
created_at: datetime
class Config:
from_attributes = True
class OrderItem(OrderItemInDBBase):
pass
# Order Schemas
class OrderBase(BaseModel):
shipping_address: str
status: Optional[OrderStatus] = OrderStatus.PENDING
class OrderCreate(OrderBase):
pass
class OrderUpdate(BaseModel):
status: Optional[OrderStatus] = None
shipping_address: Optional[str] = None
class OrderInDBBase(OrderBase):
id: int
user_id: int
total_amount: float
created_at: datetime
updated_at: Optional[datetime] = None
status: OrderStatus
class Config:
from_attributes = True
class Order(OrderInDBBase):
items: List[OrderItem]