43 lines
1.3 KiB
Python
43 lines
1.3 KiB
Python
"""add stripe payment links to pricing plans
|
|
|
|
Revision ID: add_stripe_payment_links
|
|
Revises: 9206bf87bb8e
|
|
Create Date: 2024-12-19 13:00:00.000000
|
|
|
|
"""
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
from sqlalchemy import text
|
|
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision = 'add_stripe_payment_links'
|
|
down_revision = '9206bf87bb8e'
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade():
|
|
conn = op.get_bind()
|
|
|
|
# Check if columns already exist
|
|
result = conn.execute(text("""
|
|
SELECT column_name
|
|
FROM information_schema.columns
|
|
WHERE table_name = 'pricing_plans'
|
|
AND column_name IN ('monthly_stripe_link', 'annual_stripe_link')
|
|
"""))
|
|
existing_columns = [row[0] for row in result.fetchall()]
|
|
|
|
# Add Stripe payment link columns if they don't exist
|
|
if 'monthly_stripe_link' not in existing_columns:
|
|
op.add_column('pricing_plans', sa.Column('monthly_stripe_link', sa.String(length=500), nullable=True))
|
|
|
|
if 'annual_stripe_link' not in existing_columns:
|
|
op.add_column('pricing_plans', sa.Column('annual_stripe_link', sa.String(length=500), nullable=True))
|
|
|
|
|
|
def downgrade():
|
|
# Remove Stripe payment link columns
|
|
op.drop_column('pricing_plans', 'annual_stripe_link')
|
|
op.drop_column('pricing_plans', 'monthly_stripe_link') |