Barcode Betty
c68838acf2
CI / lint (pull_request) Successful in 5s
CI / typecheck (pull_request) Failing after 29s
CI / test (pull_request) Failing after 48s
CI / build-and-push (pull_request) Has been skipped
CI / deploy-dev (pull_request) Has been skipped
CI / deploy-uat (pull_request) Has been skipped
- Auto-fix F401 (unused imports) and I001 (unsorted imports) with ruff --fix - Manually fix E501 (line too long) in alembic migrations and src/ models - Run ruff format to ensure consistent formatting Co-Authored-By: Paperclip <noreply@paperclip.ing>
58 lines
1.6 KiB
Python
58 lines
1.6 KiB
Python
"""Add email_inbound_token to users.
|
|
|
|
Revision ID: 005_add_email_inbound_token
|
|
Revises: 004_fix_user_id_text
|
|
Create Date: 2026-04-02
|
|
"""
|
|
|
|
import secrets
|
|
|
|
import sqlalchemy as sa
|
|
|
|
from alembic import op
|
|
|
|
revision = "005_add_email_inbound_token"
|
|
down_revision = "004_fix_user_id_text"
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
conn = op.get_bind()
|
|
inspector = sa.inspect(conn)
|
|
# Guard: on fresh DB, Base.metadata.create_all already has the column
|
|
if not inspector.has_table("users"):
|
|
return
|
|
existing_cols = [c["name"] for c in inspector.get_columns("users")]
|
|
if "email_inbound_token" in existing_cols:
|
|
return
|
|
|
|
# Add column nullable first so existing rows can be backfilled
|
|
op.add_column(
|
|
"users",
|
|
sa.Column("email_inbound_token", sa.String(22), nullable=True),
|
|
)
|
|
|
|
# Backfill existing users with unique tokens
|
|
result = conn.execute(sa.text("SELECT id FROM users WHERE email_inbound_token IS NULL"))
|
|
for (user_id,) in result:
|
|
token = secrets.token_urlsafe(16)
|
|
conn.execute(
|
|
sa.text("UPDATE users SET email_inbound_token = :token WHERE id = :id"),
|
|
{"token": token, "id": user_id},
|
|
)
|
|
|
|
# Now enforce non-null and unique
|
|
op.alter_column("users", "email_inbound_token", nullable=False)
|
|
op.create_index(
|
|
"ix_users_email_inbound_token",
|
|
"users",
|
|
["email_inbound_token"],
|
|
unique=True,
|
|
)
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.drop_index("ix_users_email_inbound_token", table_name="users")
|
|
op.drop_column("users", "email_inbound_token")
|