61 lines
2.1 KiB
Python
61 lines
2.1 KiB
Python
"""Cart models."""
|
|
|
|
import uuid
|
|
from datetime import datetime, timedelta
|
|
|
|
from sqlalchemy import String, Integer, Numeric, DateTime, ForeignKey
|
|
from sqlalchemy.dialects.postgresql import UUID
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
|
|
from app.database import Base
|
|
|
|
|
|
def default_expiry():
|
|
return datetime.utcnow() + timedelta(days=7)
|
|
|
|
|
|
class Cart(Base):
|
|
"""Shopping cart model."""
|
|
|
|
__tablename__ = "carts"
|
|
|
|
id: Mapped[uuid.UUID] = mapped_column(
|
|
UUID(as_uuid=True), primary_key=True, default=uuid.uuid4
|
|
)
|
|
customer_id: Mapped[uuid.UUID | None] = mapped_column(
|
|
UUID(as_uuid=True), ForeignKey("customers.id"), nullable=True
|
|
)
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow)
|
|
updated_at: Mapped[datetime] = mapped_column(
|
|
DateTime, default=datetime.utcnow, onupdate=datetime.utcnow
|
|
)
|
|
expires_at: Mapped[datetime] = mapped_column(DateTime, default=default_expiry)
|
|
|
|
# Relationships
|
|
customer: Mapped["Customer | None"] = relationship("Customer", back_populates="carts")
|
|
items: Mapped[list["CartItem"]] = relationship(
|
|
"CartItem", back_populates="cart", cascade="all, delete-orphan"
|
|
)
|
|
|
|
|
|
class CartItem(Base):
|
|
"""Cart item model."""
|
|
|
|
__tablename__ = "cart_items"
|
|
|
|
id: Mapped[uuid.UUID] = mapped_column(
|
|
UUID(as_uuid=True), primary_key=True, default=uuid.uuid4
|
|
)
|
|
cart_id: Mapped[uuid.UUID] = mapped_column(
|
|
UUID(as_uuid=True), ForeignKey("carts.id", ondelete="CASCADE"), nullable=False
|
|
)
|
|
product_slug: Mapped[str] = mapped_column(String(100), nullable=False)
|
|
product_name: Mapped[str] = mapped_column(String(255), nullable=False)
|
|
variant: Mapped[str | None] = mapped_column(String(50), nullable=True)
|
|
quantity: Mapped[int] = mapped_column(Integer, default=1, nullable=False)
|
|
unit_price: Mapped[float] = mapped_column(Numeric(10, 2), nullable=False)
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow)
|
|
|
|
# Relationships
|
|
cart: Mapped["Cart"] = relationship("Cart", back_populates="items")
|