Giorgio Gilestro
78ce8c8b0d
Five existing migrations used op.alter_column / op.create_unique_constraint / op.drop_constraint / op.create_foreign_key directly on the users + quotes + quotes_daily tables. SQLite has no native support for those operations and requires Alembic's batch_alter_table copy-and-rename workaround. This wasn't noticed until now because the test suite uses Base.metadata.create_all to materialise schema, not the migration chain itself; and prod is MariaDB. But running `alembic upgrade head` against a fresh SQLite database (developer onboarding, CI smoke tests, the test container's own bootstrap) would fail at 0005. Fixes: - alembic/env.py: set render_as_batch=True when the dialect is SQLite. This auto-wraps any future autogenerated migration but doesn't retroactively rewrite existing op.* calls. - 0005 (widen quotes.symbol), 0013 (referrals), 0018 (polar webhook), 0019 (stripe), 0023 (users.lang index + qd_symbol widen) explicitly wrap their problematic ops in `with op.batch_alter_table(...) as bop`. Now `alembic upgrade head` + `alembic downgrade base` round-trip cleanly on a fresh SQLite database. MariaDB prod behaviour unchanged. Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
52 lines
1.8 KiB
Python
52 lines
1.8 KiB
Python
"""stripe integration: users.stripe_customer_id / stripe_subscription_id,
|
|
stripe_events table.
|
|
|
|
Revision ID: 0019
|
|
Revises: 0018
|
|
Create Date: 2026-05-26
|
|
"""
|
|
from typing import Sequence, Union
|
|
|
|
import sqlalchemy as sa
|
|
from alembic import op
|
|
|
|
|
|
revision: str = "0019"
|
|
down_revision: Union[str, None] = "0018"
|
|
branch_labels: Union[str, Sequence[str], None] = None
|
|
depends_on: Union[str, Sequence[str], None] = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
with op.batch_alter_table("users") as bop:
|
|
bop.add_column(sa.Column("stripe_customer_id", sa.String(length=64), nullable=True))
|
|
bop.add_column(sa.Column("stripe_subscription_id", sa.String(length=64), nullable=True))
|
|
bop.create_unique_constraint(
|
|
"uq_users_stripe_customer", ["stripe_customer_id"],
|
|
)
|
|
|
|
op.create_table(
|
|
"stripe_events",
|
|
sa.Column("id", sa.BigInteger(), autoincrement=True, primary_key=True),
|
|
sa.Column("event_id", sa.String(length=128), nullable=False),
|
|
sa.Column("event_type", sa.String(length=64), nullable=False),
|
|
sa.Column("received_at", sa.DateTime(timezone=True), nullable=False),
|
|
sa.Column("processed_at", sa.DateTime(timezone=True), nullable=True),
|
|
sa.Column("error", sa.Text(), nullable=True),
|
|
sa.Column("payload", sa.Text(), nullable=False),
|
|
sa.UniqueConstraint("event_id", name="uq_stripe_events_event_id"),
|
|
)
|
|
op.create_index(
|
|
"ix_stripe_events_type_received",
|
|
"stripe_events",
|
|
["event_type", "received_at"],
|
|
)
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.drop_index("ix_stripe_events_type_received", table_name="stripe_events")
|
|
op.drop_table("stripe_events")
|
|
with op.batch_alter_table("users") as bop:
|
|
bop.drop_constraint("uq_users_stripe_customer", type_="unique")
|
|
bop.drop_column("stripe_subscription_id")
|
|
bop.drop_column("stripe_customer_id")
|