42 lines
1.4 KiB
Python
42 lines
1.4 KiB
Python
"""digital ticket qr tokens and used status
|
|
|
|
Revision ID: 20260404_0009
|
|
Revises: 20260404_0008
|
|
"""
|
|
|
|
import secrets
|
|
from typing import Sequence, Union
|
|
|
|
import sqlalchemy as sa
|
|
from alembic import op
|
|
|
|
revision: str = "20260404_0009"
|
|
down_revision: Union[str, None] = "20260404_0008"
|
|
branch_labels: Union[str, Sequence[str], None] = None
|
|
depends_on: Union[str, Sequence[str], None] = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
op.execute("ALTER TYPE ticketstatus RENAME VALUE 'redeemed' TO 'used'")
|
|
|
|
op.add_column("tickets", sa.Column("qr_token", sa.String(length=64), nullable=True))
|
|
op.add_column("tickets", sa.Column("used_at", sa.DateTime(timezone=True), nullable=True))
|
|
|
|
connection = op.get_bind()
|
|
rows = connection.execute(sa.text("SELECT id FROM tickets WHERE qr_token IS NULL")).fetchall()
|
|
for row in rows:
|
|
token = secrets.token_urlsafe(32)
|
|
connection.execute(
|
|
sa.text("UPDATE tickets SET qr_token = :token WHERE id = :id"),
|
|
{"token": token, "id": row.id},
|
|
)
|
|
|
|
op.alter_column("tickets", "qr_token", nullable=False)
|
|
op.create_index(op.f("ix_tickets_qr_token"), "tickets", ["qr_token"], unique=True)
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.drop_index(op.f("ix_tickets_qr_token"), table_name="tickets")
|
|
op.drop_column("tickets", "used_at")
|
|
op.drop_column("tickets", "qr_token")
|
|
op.execute("ALTER TYPE ticketstatus RENAME VALUE 'used' TO 'redeemed'")
|