Desi-Clans/backend/alembic/versions/20260404_0008_payments.py
2026-10-08 02:52:58 +05:30

73 lines
3.1 KiB
Python

"""payment fields and webhook idempotency
Revision ID: 20260404_0008
Revises: 20260404_0007
"""
from typing import Sequence, Union
import sqlalchemy as sa
from alembic import op
revision: str = "20260404_0008"
down_revision: Union[str, None] = "20260404_0007"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None
def upgrade() -> None:
op.alter_column(
"orders",
"payment_reference",
existing_type=sa.String(length=64),
type_=sa.String(length=128),
existing_nullable=True,
)
op.add_column("orders", sa.Column("provider_order_id", sa.String(length=64), nullable=True))
op.add_column("orders", sa.Column("provider_payment_id", sa.String(length=64), nullable=True))
op.add_column("orders", sa.Column("idempotency_key", sa.String(length=64), nullable=True))
op.add_column("orders", sa.Column("expires_at", sa.DateTime(timezone=True), nullable=True))
op.create_index(op.f("ix_orders_provider_order_id"), "orders", ["provider_order_id"], unique=True)
op.create_index(op.f("ix_orders_provider_payment_id"), "orders", ["provider_payment_id"], unique=False)
op.create_index(op.f("ix_orders_idempotency_key"), "orders", ["idempotency_key"], unique=True)
op.create_index(op.f("ix_orders_expires_at"), "orders", ["expires_at"], unique=False)
op.create_table(
"payment_webhook_events",
sa.Column("id", sa.Integer(), nullable=False),
sa.Column("provider", sa.String(length=32), nullable=False),
sa.Column("provider_event_id", sa.String(length=128), nullable=False),
sa.Column("event_type", sa.String(length=64), nullable=False),
sa.Column("payload", sa.Text(), nullable=False),
sa.Column("processed_at", sa.DateTime(timezone=True), server_default=sa.text("now()"), nullable=False),
sa.PrimaryKeyConstraint("id"),
sa.UniqueConstraint("provider_event_id"),
)
op.create_index(op.f("ix_payment_webhook_events_provider"), "payment_webhook_events", ["provider"], unique=False)
op.create_index(
op.f("ix_payment_webhook_events_provider_event_id"),
"payment_webhook_events",
["provider_event_id"],
unique=True,
)
def downgrade() -> None:
op.drop_index(op.f("ix_payment_webhook_events_provider_event_id"), table_name="payment_webhook_events")
op.drop_index(op.f("ix_payment_webhook_events_provider"), table_name="payment_webhook_events")
op.drop_table("payment_webhook_events")
op.drop_index(op.f("ix_orders_expires_at"), table_name="orders")
op.drop_index(op.f("ix_orders_idempotency_key"), table_name="orders")
op.drop_index(op.f("ix_orders_provider_payment_id"), table_name="orders")
op.drop_index(op.f("ix_orders_provider_order_id"), table_name="orders")
op.drop_column("orders", "expires_at")
op.drop_column("orders", "idempotency_key")
op.drop_column("orders", "provider_payment_id")
op.drop_column("orders", "provider_order_id")
op.alter_column(
"orders",
"payment_reference",
existing_type=sa.String(length=128),
type_=sa.String(length=64),
existing_nullable=True,
)