63 lines
2.3 KiB
Python
63 lines
2.3 KiB
Python
|
from datetime import datetime
|
||
|
|
from app import db
|
||
|
|
|
||
|
|
|
||
|
|
class Order(db.Model):
|
||
|
|
"""Order model"""
|
||
|
|
__tablename__ = "orders"
|
||
|
|
|
||
|
|
id = db.Column(db.Integer, primary_key=True)
|
||
|
|
user_id = db.Column(db.Integer, db.ForeignKey("users.id"), nullable=False)
|
||
|
|
status = db.Column(db.String(20), default="pending", index=True)
|
||
|
|
total_amount = db.Column(db.Numeric(10, 2), nullable=False)
|
||
|
|
shipping_address = db.Column(db.Text)
|
||
|
|
created_at = db.Column(db.DateTime, default=datetime.utcnow)
|
||
|
|
updated_at = db.Column(db.DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
|
||
|
|
|
||
|
|
# Relationships
|
||
|
|
user = db.relationship("User", back_populates="orders")
|
||
|
|
items = db.relationship("OrderItem", back_populates="order", lazy="dynamic", cascade="all, delete-orphan")
|
||
|
|
|
||
|
|
def to_dict(self):
|
||
|
|
"""Convert order to dictionary"""
|
||
|
|
return {
|
||
|
|
"id": self.id,
|
||
|
|
"user_id": self.user_id,
|
||
|
|
"status": self.status,
|
||
|
|
"total_amount": float(self.total_amount) if self.total_amount else None,
|
||
|
|
"shipping_address": self.shipping_address,
|
||
|
|
"created_at": self.created_at.isoformat() if self.created_at else None,
|
||
|
|
"updated_at": self.updated_at.isoformat() if self.updated_at else None,
|
||
|
|
"items": [item.to_dict() for item in self.items]
|
||
|
|
}
|
||
|
|
|
||
|
|
def __repr__(self):
|
||
|
|
return f"<Order {self.id}>"
|
||
|
|
|
||
|
|
|
||
|
|
class OrderItem(db.Model):
|
||
|
|
"""Order Item model"""
|
||
|
|
__tablename__ = "order_items"
|
||
|
|
|
||
|
|
id = db.Column(db.Integer, primary_key=True)
|
||
|
|
order_id = db.Column(db.Integer, db.ForeignKey("orders.id"), nullable=False)
|
||
|
|
product_id = db.Column(db.Integer, db.ForeignKey("products.id"), nullable=False)
|
||
|
|
quantity = db.Column(db.Integer, nullable=False)
|
||
|
|
price = db.Column(db.Numeric(10, 2), nullable=False)
|
||
|
|
|
||
|
|
# Relationships
|
||
|
|
order = db.relationship("Order", back_populates="items")
|
||
|
|
product = db.relationship("Product", back_populates="order_items")
|
||
|
|
|
||
|
|
def to_dict(self):
|
||
|
|
"""Convert order item to dictionary"""
|
||
|
|
return {
|
||
|
|
"id": self.id,
|
||
|
|
"order_id": self.order_id,
|
||
|
|
"product_id": self.product_id,
|
||
|
|
"quantity": self.quantity,
|
||
|
|
"price": float(self.price) if self.price else None
|
||
|
|
}
|
||
|
|
|
||
|
|
def __repr__(self):
|
||
|
|
return f"<OrderItem {self.id}>"
|