andika/membership-be
1
0
Fork 1
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix migrations 004-006: add column existence checks to prevent duplicate column errors

Browse Source
This commit is contained in:
Koncept Kit
2026-01-05 11:50:26 +07:00
parent 9266521bf1
commit df789612e8
3 changed files with 62 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

52
alembic/versions/005_fix_subscriptions_and_storage.py
View File

@@ -22,26 +22,44 @@ depends_on: Union[str, Sequence[str], None] = None
def upgrade() -> None:
"""Add missing columns and fix naming"""
"""Add missing columns and fix naming (skip if already exists)"""
from sqlalchemy import inspect
# Add missing columns to subscriptions table
op.add_column('subscriptions', sa.Column('start_date', sa.DateTime(timezone=True), nullable=True))
op.add_column('subscriptions', sa.Column('end_date', sa.DateTime(timezone=True), nullable=True))
op.add_column('subscriptions', sa.Column('amount_paid_cents', sa.Integer(), nullable=True))
op.add_column('subscriptions', sa.Column('manual_payment_notes', sa.Text(), nullable=True))
op.add_column('subscriptions', sa.Column('manual_payment_admin_id', UUID(as_uuid=True), nullable=True))
op.add_column('subscriptions', sa.Column('manual_payment_date', sa.DateTime(timezone=True), nullable=True))
op.add_column('subscriptions', sa.Column('payment_method', sa.String(50), nullable=True))
conn = op.get_bind()
inspector = inspect(conn)
# Add foreign key for manual_payment_admin_id
op.create_foreign_key(
'subscriptions_manual_payment_admin_id_fkey',
'subscriptions', 'users',
['manual_payment_admin_id'], ['id']
)
# Check existing columns in subscriptions table
existing_columns = {col['name'] for col in inspector.get_columns('subscriptions')}
# Rename storage_usage.last_calculated_at to last_updated
op.alter_column('storage_usage', 'last_calculated_at', new_column_name='last_updated')
# Add missing columns to subscriptions table only if they don't exist
if 'start_date' not in existing_columns:
op.add_column('subscriptions', sa.Column('start_date', sa.DateTime(timezone=True), nullable=True))
if 'end_date' not in existing_columns:
op.add_column('subscriptions', sa.Column('end_date', sa.DateTime(timezone=True), nullable=True))
if 'amount_paid_cents' not in existing_columns:
op.add_column('subscriptions', sa.Column('amount_paid_cents', sa.Integer(), nullable=True))
if 'manual_payment_notes' not in existing_columns:
op.add_column('subscriptions', sa.Column('manual_payment_notes', sa.Text(), nullable=True))
if 'manual_payment_admin_id' not in existing_columns:
op.add_column('subscriptions', sa.Column('manual_payment_admin_id', UUID(as_uuid=True), nullable=True))
if 'manual_payment_date' not in existing_columns:
op.add_column('subscriptions', sa.Column('manual_payment_date', sa.DateTime(timezone=True), nullable=True))
if 'payment_method' not in existing_columns:
op.add_column('subscriptions', sa.Column('payment_method', sa.String(50), nullable=True))
# Add foreign key for manual_payment_admin_id if it doesn't exist
existing_fks = [fk['name'] for fk in inspector.get_foreign_keys('subscriptions')]
if 'subscriptions_manual_payment_admin_id_fkey' not in existing_fks:
op.create_foreign_key(
'subscriptions_manual_payment_admin_id_fkey',
'subscriptions', 'users',
['manual_payment_admin_id'], ['id']
)
# Rename storage_usage.last_calculated_at to last_updated (only if needed)
storage_columns = {col['name'] for col in inspector.get_columns('storage_usage')}
if 'last_calculated_at' in storage_columns and 'last_updated' not in storage_columns:
op.alter_column('storage_usage', 'last_calculated_at', new_column_name='last_updated')
def downgrade() -> None: