87 lines
4.2 KiB
Python
87 lines
4.2 KiB
Python
"""add_payment_methods
|
|
|
|
Revision ID: a1b2c3d4e5f6
|
|
Revises: 956ea1628264
|
|
Create Date: 2026-01-30 10:00:00.000000
|
|
|
|
"""
|
|
from typing import Sequence, Union
|
|
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
from sqlalchemy.dialects import postgresql
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision: str = 'a1b2c3d4e5f6'
|
|
down_revision: Union[str, None] = '956ea1628264'
|
|
branch_labels: Union[str, Sequence[str], None] = None
|
|
depends_on: Union[str, Sequence[str], None] = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
# Create PaymentMethodType enum
|
|
paymentmethodtype = postgresql.ENUM(
|
|
'card', 'cash', 'bank_transfer', 'check',
|
|
name='paymentmethodtype',
|
|
create_type=False
|
|
)
|
|
paymentmethodtype.create(op.get_bind(), checkfirst=True)
|
|
|
|
# Add stripe_customer_id to users table
|
|
op.add_column('users', sa.Column(
|
|
'stripe_customer_id',
|
|
sa.String(),
|
|
nullable=True,
|
|
comment='Stripe Customer ID for payment method management'
|
|
))
|
|
op.create_index('ix_users_stripe_customer_id', 'users', ['stripe_customer_id'])
|
|
|
|
# Create payment_methods table
|
|
op.create_table(
|
|
'payment_methods',
|
|
sa.Column('id', postgresql.UUID(as_uuid=True), primary_key=True),
|
|
sa.Column('user_id', postgresql.UUID(as_uuid=True), sa.ForeignKey('users.id', ondelete='CASCADE'), nullable=False),
|
|
sa.Column('stripe_payment_method_id', sa.String(), nullable=True, unique=True, comment='Stripe pm_xxx reference'),
|
|
sa.Column('card_brand', sa.String(20), nullable=True, comment='Card brand: visa, mastercard, amex, etc.'),
|
|
sa.Column('card_last4', sa.String(4), nullable=True, comment='Last 4 digits of card'),
|
|
sa.Column('card_exp_month', sa.Integer(), nullable=True, comment='Card expiration month'),
|
|
sa.Column('card_exp_year', sa.Integer(), nullable=True, comment='Card expiration year'),
|
|
sa.Column('card_funding', sa.String(20), nullable=True, comment='Card funding type: credit, debit, prepaid'),
|
|
sa.Column('payment_type', paymentmethodtype, nullable=False, server_default='card'),
|
|
sa.Column('is_default', sa.Boolean(), nullable=False, server_default='false', comment='Whether this is the default payment method for auto-renewals'),
|
|
sa.Column('is_active', sa.Boolean(), nullable=False, server_default='true', comment='Soft delete flag - False means removed'),
|
|
sa.Column('is_manual', sa.Boolean(), nullable=False, server_default='false', comment='True for manually recorded methods (cash/check)'),
|
|
sa.Column('manual_notes', sa.Text(), nullable=True, comment='Admin notes for manual payment methods'),
|
|
sa.Column('created_by', postgresql.UUID(as_uuid=True), sa.ForeignKey('users.id', ondelete='SET NULL'), nullable=True, comment='Admin who added this on behalf of user'),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), nullable=False, server_default=sa.func.now()),
|
|
sa.Column('updated_at', sa.DateTime(timezone=True), nullable=False, server_default=sa.func.now(), onupdate=sa.func.now()),
|
|
)
|
|
|
|
# Create indexes
|
|
op.create_index('ix_payment_methods_user_id', 'payment_methods', ['user_id'])
|
|
op.create_index('ix_payment_methods_stripe_pm_id', 'payment_methods', ['stripe_payment_method_id'])
|
|
op.create_index('idx_payment_method_user_default', 'payment_methods', ['user_id', 'is_default'])
|
|
op.create_index('idx_payment_method_active', 'payment_methods', ['user_id', 'is_active'])
|
|
|
|
|
|
def downgrade() -> None:
|
|
# Drop indexes
|
|
op.drop_index('idx_payment_method_active', table_name='payment_methods')
|
|
op.drop_index('idx_payment_method_user_default', table_name='payment_methods')
|
|
op.drop_index('ix_payment_methods_stripe_pm_id', table_name='payment_methods')
|
|
op.drop_index('ix_payment_methods_user_id', table_name='payment_methods')
|
|
|
|
# Drop payment_methods table
|
|
op.drop_table('payment_methods')
|
|
|
|
# Drop stripe_customer_id from users
|
|
op.drop_index('ix_users_stripe_customer_id', table_name='users')
|
|
op.drop_column('users', 'stripe_customer_id')
|
|
|
|
# Drop PaymentMethodType enum
|
|
paymentmethodtype = postgresql.ENUM(
|
|
'card', 'cash', 'bank_transfer', 'check',
|
|
name='paymentmethodtype'
|
|
)
|
|
paymentmethodtype.drop(op.get_bind(), checkfirst=True)
|