added payment links to prices

migrations/versions/add_stripe_payment_links_to_pricing_plans.py

@@ -0,0 +1,43 @@
+"""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')