alembic/versions/add_payment_methods.py

@@ -19,14 +19,22 @@ depends_on: Union[str, Sequence[str], None] = None
 
 
 def upgrade() -> None:
+    conn = op.get_bind()
+
     # Create PaymentMethodType enum
     paymentmethodtype = postgresql.ENUM(
         'card', 'cash', 'bank_transfer', 'check',
         name='paymentmethodtype',
         create_type=False
     )
-    paymentmethodtype.create(op.get_bind(), checkfirst=True)
+    paymentmethodtype.create(conn, checkfirst=True)
 
+    # Check if stripe_customer_id column exists on users table
+    result = conn.execute(sa.text("""
+        SELECT column_name FROM information_schema.columns
+        WHERE table_name = 'users' AND column_name = 'stripe_customer_id'
+    """))
+    if result.fetchone() is None:
         # Add stripe_customer_id to users table
         op.add_column('users', sa.Column(
             'stripe_customer_id',
@@ -36,6 +44,12 @@ def upgrade() -> None:
         ))
         op.create_index('ix_users_stripe_customer_id', 'users', ['stripe_customer_id'])
 
+    # Check if payment_methods table exists
+    result = conn.execute(sa.text("""
+        SELECT table_name FROM information_schema.tables
+        WHERE table_name = 'payment_methods'
+    """))
+    if result.fetchone() is None:
         # Create payment_methods table
         op.create_table(
             'payment_methods',