Fix migration 010: add column existence check for email_verification_expires

This commit is contained in:
Koncept Kit
2026-01-05 13:56:37 +07:00
parent df789612e8
commit 669d78beb5

View File

@@ -20,7 +20,15 @@ depends_on: Union[str, Sequence[str], None] = None
def upgrade() -> None:
"""Add email_verification_expires column"""
"""Add email_verification_expires column (skip if already exists)"""
from sqlalchemy import inspect
conn = op.get_bind()
inspector = inspect(conn)
existing_columns = {col['name'] for col in inspector.get_columns('users')}
# Add email_verification_expires if missing
if 'email_verification_expires' not in existing_columns:
op.add_column('users', sa.Column('email_verification_expires', sa.DateTime(), nullable=True))