129 lines
6.2 KiB
Python
129 lines
6.2 KiB
Python
"""orders order_items tickets
|
|
|
|
Revision ID: 20260404_0007
|
|
Revises: 20260404_0006
|
|
"""
|
|
|
|
from typing import Sequence, Union
|
|
|
|
import sqlalchemy as sa
|
|
from alembic import op
|
|
from sqlalchemy.dialects import postgresql
|
|
|
|
revision: str = "20260404_0007"
|
|
down_revision: Union[str, None] = "20260404_0006"
|
|
branch_labels: Union[str, Sequence[str], None] = None
|
|
depends_on: Union[str, Sequence[str], None] = None
|
|
|
|
order_status = postgresql.ENUM(
|
|
"pending",
|
|
"paid",
|
|
"cancelled",
|
|
"failed",
|
|
name="orderstatus",
|
|
create_type=False,
|
|
)
|
|
ticket_status = postgresql.ENUM(
|
|
"valid",
|
|
"cancelled",
|
|
"redeemed",
|
|
name="ticketstatus",
|
|
create_type=False,
|
|
)
|
|
|
|
|
|
def upgrade() -> None:
|
|
order_status.create(op.get_bind(), checkfirst=True)
|
|
ticket_status.create(op.get_bind(), checkfirst=True)
|
|
|
|
op.create_table(
|
|
"orders",
|
|
sa.Column("id", sa.Integer(), nullable=False),
|
|
sa.Column("user_id", sa.Integer(), nullable=False),
|
|
sa.Column("event_id", sa.Integer(), nullable=False),
|
|
sa.Column("status", order_status, nullable=False, server_default="pending"),
|
|
sa.Column("total_paise", sa.Integer(), nullable=False),
|
|
sa.Column("currency", sa.String(length=3), nullable=False, server_default="INR"),
|
|
sa.Column("payment_reference", sa.String(length=64), nullable=True),
|
|
sa.Column("paid_at", sa.DateTime(timezone=True), nullable=True),
|
|
sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.text("now()"), nullable=False),
|
|
sa.Column("updated_at", sa.DateTime(timezone=True), server_default=sa.text("now()"), nullable=False),
|
|
sa.ForeignKeyConstraint(["event_id"], ["events.id"], ondelete="RESTRICT"),
|
|
sa.ForeignKeyConstraint(["user_id"], ["users.id"], ondelete="RESTRICT"),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
)
|
|
op.create_index(op.f("ix_orders_event_id"), "orders", ["event_id"], unique=False)
|
|
op.create_index(op.f("ix_orders_status"), "orders", ["status"], unique=False)
|
|
op.create_index(op.f("ix_orders_user_id"), "orders", ["user_id"], unique=False)
|
|
|
|
op.create_table(
|
|
"order_items",
|
|
sa.Column("id", sa.Integer(), nullable=False),
|
|
sa.Column("order_id", sa.Integer(), nullable=False),
|
|
sa.Column("ticket_type_id", sa.Integer(), nullable=False),
|
|
sa.Column("quantity", sa.Integer(), nullable=False),
|
|
sa.Column("unit_price_paise", sa.Integer(), nullable=False),
|
|
sa.Column("line_total_paise", sa.Integer(), nullable=False),
|
|
sa.CheckConstraint("quantity > 0", name="ck_order_items_quantity_positive"),
|
|
sa.ForeignKeyConstraint(["order_id"], ["orders.id"], ondelete="CASCADE"),
|
|
sa.ForeignKeyConstraint(["ticket_type_id"], ["ticket_types.id"], ondelete="RESTRICT"),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
)
|
|
op.create_index(op.f("ix_order_items_order_id"), "order_items", ["order_id"], unique=False)
|
|
op.create_index(op.f("ix_order_items_ticket_type_id"), "order_items", ["ticket_type_id"], unique=False)
|
|
|
|
op.create_table(
|
|
"tickets",
|
|
sa.Column("id", sa.Integer(), nullable=False),
|
|
sa.Column("code", sa.String(length=32), nullable=False),
|
|
sa.Column("order_id", sa.Integer(), nullable=False),
|
|
sa.Column("order_item_id", sa.Integer(), nullable=False),
|
|
sa.Column("ticket_type_id", sa.Integer(), nullable=False),
|
|
sa.Column("event_id", sa.Integer(), nullable=False),
|
|
sa.Column("user_id", sa.Integer(), nullable=False),
|
|
sa.Column("status", ticket_status, nullable=False, server_default="valid"),
|
|
sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.text("now()"), nullable=False),
|
|
sa.ForeignKeyConstraint(["event_id"], ["events.id"], ondelete="RESTRICT"),
|
|
sa.ForeignKeyConstraint(["order_id"], ["orders.id"], ondelete="CASCADE"),
|
|
sa.ForeignKeyConstraint(["order_item_id"], ["order_items.id"], ondelete="CASCADE"),
|
|
sa.ForeignKeyConstraint(["ticket_type_id"], ["ticket_types.id"], ondelete="RESTRICT"),
|
|
sa.ForeignKeyConstraint(["user_id"], ["users.id"], ondelete="RESTRICT"),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
sa.UniqueConstraint("code"),
|
|
)
|
|
op.create_index(op.f("ix_tickets_code"), "tickets", ["code"], unique=True)
|
|
op.create_index(op.f("ix_tickets_event_id"), "tickets", ["event_id"], unique=False)
|
|
op.create_index(op.f("ix_tickets_order_id"), "tickets", ["order_id"], unique=False)
|
|
op.create_index(op.f("ix_tickets_order_item_id"), "tickets", ["order_item_id"], unique=False)
|
|
op.create_index(op.f("ix_tickets_status"), "tickets", ["status"], unique=False)
|
|
op.create_index(op.f("ix_tickets_ticket_type_id"), "tickets", ["ticket_type_id"], unique=False)
|
|
op.create_index(op.f("ix_tickets_user_id"), "tickets", ["user_id"], unique=False)
|
|
op.create_index("ix_tickets_user_status", "tickets", ["user_id", "status"], unique=False)
|
|
|
|
op.create_check_constraint(
|
|
"ck_ticket_types_sold_not_exceed_total",
|
|
"ticket_types",
|
|
"quantity_sold <= quantity_total",
|
|
)
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.drop_constraint("ck_ticket_types_sold_not_exceed_total", "ticket_types", type_="check")
|
|
op.drop_index("ix_tickets_user_status", table_name="tickets")
|
|
op.drop_index(op.f("ix_tickets_user_id"), table_name="tickets")
|
|
op.drop_index(op.f("ix_tickets_ticket_type_id"), table_name="tickets")
|
|
op.drop_index(op.f("ix_tickets_status"), table_name="tickets")
|
|
op.drop_index(op.f("ix_tickets_order_item_id"), table_name="tickets")
|
|
op.drop_index(op.f("ix_tickets_order_id"), table_name="tickets")
|
|
op.drop_index(op.f("ix_tickets_event_id"), table_name="tickets")
|
|
op.drop_index(op.f("ix_tickets_code"), table_name="tickets")
|
|
op.drop_table("tickets")
|
|
op.drop_index(op.f("ix_order_items_ticket_type_id"), table_name="order_items")
|
|
op.drop_index(op.f("ix_order_items_order_id"), table_name="order_items")
|
|
op.drop_table("order_items")
|
|
op.drop_index(op.f("ix_orders_user_id"), table_name="orders")
|
|
op.drop_index(op.f("ix_orders_status"), table_name="orders")
|
|
op.drop_index(op.f("ix_orders_event_id"), table_name="orders")
|
|
op.drop_table("orders")
|
|
ticket_status.drop(op.get_bind(), checkfirst=True)
|
|
order_status.drop(op.get_bind(), checkfirst=True)
|