103 lines
2.3 KiB
Python
103 lines
2.3 KiB
Python
from pydantic import BaseModel, Field, EmailStr, validator
|
|
from typing import List, Optional
|
|
from datetime import datetime
|
|
from bson import ObjectId
|
|
|
|
|
|
class PyObjectId(ObjectId):
|
|
@classmethod
|
|
def __get_validators__(cls):
|
|
yield cls.validate
|
|
|
|
@classmethod
|
|
def validate(cls, v):
|
|
if not ObjectId.is_valid(v):
|
|
raise ValueError("Invalid objectid")
|
|
return ObjectId(v)
|
|
|
|
@classmethod
|
|
def __modify_schema__(cls, field_schema):
|
|
field_schema.update(type="string")
|
|
|
|
|
|
class MongoBaseModel(BaseModel):
|
|
id: Optional[PyObjectId] = Field(default_factory=PyObjectId, alias="_id")
|
|
|
|
class Config:
|
|
allow_population_by_field_name = True
|
|
arbitrary_types_allowed = True
|
|
json_encoders = {ObjectId: str}
|
|
|
|
|
|
class Item(MongoBaseModel):
|
|
name: str
|
|
price: float
|
|
quantity: int
|
|
unit: str
|
|
related_items: List[str] = []
|
|
|
|
|
|
class OrderItem(BaseModel):
|
|
item_id: PyObjectId
|
|
quantity: int
|
|
price_at_order: float
|
|
|
|
|
|
class Order(MongoBaseModel):
|
|
user_id: PyObjectId
|
|
items: List[OrderItem]
|
|
total_amount: float
|
|
payment_method: Optional[str] = None
|
|
payment_status: str = "pending"
|
|
order_status: str = "created"
|
|
created_at: datetime = Field(default_factory=datetime.utcnow)
|
|
updated_at: Optional[datetime] = None
|
|
discount_applied: Optional[float] = None
|
|
notes: Optional[str] = None
|
|
|
|
@validator('order_status')
|
|
def valid_order_status(cls, v):
|
|
allowed_statuses = ["created", "processing",
|
|
"shipped", "delivered", "cancelled"]
|
|
if v not in allowed_statuses:
|
|
raise ValueError(f"Invalid order status. Must be one of: {
|
|
', '.join(allowed_statuses)}")
|
|
return v
|
|
|
|
@validator('payment_status')
|
|
def valid_payment_status(cls, v):
|
|
allowed_statuses = ["pending", "paid", "refunded", "failed"]
|
|
if v not in allowed_statuses:
|
|
raise ValueError(f"Invalid payment status. Must be one of: {
|
|
', '.join(allowed_statuses)}")
|
|
return v
|
|
|
|
|
|
class UserBase(BaseModel):
|
|
username: str
|
|
email: EmailStr
|
|
full_name: str
|
|
is_active: bool = True
|
|
is_superuser: bool = False
|
|
|
|
|
|
class UserCreate(UserBase):
|
|
password: str
|
|
|
|
|
|
class UserInDB(UserBase, MongoBaseModel):
|
|
hashed_password: str
|
|
|
|
|
|
class User(UserBase, MongoBaseModel):
|
|
pass
|
|
|
|
|
|
class Token(BaseModel):
|
|
access_token: str
|
|
token_type: str
|
|
|
|
|
|
class TokenData(BaseModel):
|
|
username: Optional[str] = None
|