carsten
d2abe70333
Nine SQLModel table models covering the Organization/Route hierarchy (WineCategory, Organization, User, Route, Participant, PurchaseRound, WineOffering, Order, OrderLine), with explicit tablenames, FK cascade behavior, and a CHECK constraint preventing a PurchaseRound from reaching open/closed status without its opens_at/order_deadline_at/ pickup_at dates set. Also fixes the Alembic script template to import sqlmodel (needed for autogenerate's AutoString column type). Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com>
48 lines
1.6 KiB
Python
48 lines
1.6 KiB
Python
from datetime import datetime, timezone
|
|
from enum import Enum
|
|
from typing import TYPE_CHECKING, List, Optional
|
|
|
|
from sqlalchemy import Column
|
|
from sqlalchemy import Enum as SAEnum
|
|
from sqlmodel import Field, Relationship, SQLModel
|
|
|
|
if TYPE_CHECKING:
|
|
from app.models.order_line import OrderLine
|
|
from app.models.participant import Participant
|
|
from app.models.purchase_round import PurchaseRound
|
|
|
|
|
|
class PaymentStatus(str, Enum):
|
|
UNPAID = "unpaid"
|
|
PAID = "paid"
|
|
|
|
|
|
class OrderBase(SQLModel):
|
|
payment_status: PaymentStatus = Field(
|
|
default=PaymentStatus.UNPAID,
|
|
sa_column=Column(
|
|
SAEnum(
|
|
PaymentStatus,
|
|
name="payment_status",
|
|
native_enum=False,
|
|
values_callable=lambda enum_cls: [member.value for member in enum_cls],
|
|
),
|
|
nullable=False,
|
|
),
|
|
)
|
|
paid_at: Optional[datetime] = Field(default=None)
|
|
created_at: datetime = Field(default_factory=lambda: datetime.now(timezone.utc))
|
|
|
|
|
|
class Order(OrderBase, table=True):
|
|
__tablename__ = "orders"
|
|
|
|
id: Optional[int] = Field(default=None, primary_key=True)
|
|
participant_id: int = Field(foreign_key="participant.id", ondelete="CASCADE")
|
|
purchase_round_id: int = Field(foreign_key="purchase_round.id", ondelete="CASCADE")
|
|
|
|
participant: "Participant" = Relationship(back_populates="orders")
|
|
purchase_round: "PurchaseRound" = Relationship(back_populates="orders")
|
|
order_lines: List["OrderLine"] = Relationship(
|
|
back_populates="order", sa_relationship_kwargs={"cascade": "all, delete-orphan"}
|
|
)
|