Add subcription feature
This commit is contained in:
@@ -0,0 +1,41 @@
|
||||
"""add_payment_columns_to_users
|
||||
|
||||
Revision ID: 20260113_add_payment_columns
|
||||
Revises: 510cf1184e0b
|
||||
Create Date: 2026-01-13 13:30:00.000000+00:00
|
||||
|
||||
Add payment_customer_id and default_payment_method_id columns to users table
|
||||
to support payment integration.
|
||||
"""
|
||||
from typing import Sequence, Union
|
||||
|
||||
from alembic import op
|
||||
import sqlalchemy as sa
|
||||
|
||||
# revision identifiers, used by Alembic.
|
||||
revision: str = '20260113_add_payment_columns'
|
||||
down_revision: Union[str, None] = '510cf1184e0b'
|
||||
branch_labels: Union[str, Sequence[str], None] = None
|
||||
depends_on: Union[str, Sequence[str], None] = None
|
||||
|
||||
|
||||
def upgrade() -> None:
|
||||
# Add payment_customer_id column
|
||||
op.add_column('users',
|
||||
sa.Column('payment_customer_id', sa.String(length=255), nullable=True))
|
||||
|
||||
# Add default_payment_method_id column
|
||||
op.add_column('users',
|
||||
sa.Column('default_payment_method_id', sa.String(length=255), nullable=True))
|
||||
|
||||
# Create index for payment_customer_id
|
||||
op.create_index(op.f('ix_users_payment_customer_id'), 'users', ['payment_customer_id'], unique=False)
|
||||
|
||||
|
||||
def downgrade() -> None:
|
||||
# Drop index first
|
||||
op.drop_index(op.f('ix_users_payment_customer_id'), table_name='users')
|
||||
|
||||
# Drop columns
|
||||
op.drop_column('users', 'default_payment_method_id')
|
||||
op.drop_column('users', 'payment_customer_id')
|
||||
Reference in New Issue
Block a user