"""add_stripe_payment_method

Revision ID: add_stripe_payment_method
Revises: 96c23dad405d
Create Date: 2025-01-17 12:00:00.000000

"""
from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects import mysql

# revision identifiers, used by Alembic.
revision = 'add_stripe_payment_method'
down_revision = '96c23dad405d'
branch_labels = None
depends_on = None


def upgrade() -> None:
    # Note: MySQL ENUM modifications can be tricky.
    # If payments table already has data with existing enum values,
    # we need to preserve them when adding 'stripe'
    
    # For MySQL, we need to alter the ENUM column to include the new value
    # Check if we're using MySQL
    bind = op.get_bind()
    if bind.dialect.name == 'mysql':
        # Alter the ENUM column to include 'stripe'
        # This preserves existing values and adds 'stripe'
        op.execute(
            "ALTER TABLE payments MODIFY COLUMN payment_method ENUM('cash', 'credit_card', 'debit_card', 'bank_transfer', 'e_wallet', 'stripe') NOT NULL"
        )
    else:
        # For other databases (PostgreSQL, SQLite), enum changes are handled differently
        # For SQLite, this might not be needed as it doesn't enforce enum constraints
        pass
    # ### end Alembic commands ###


def downgrade() -> None:
    # Remove 'stripe' from the ENUM (be careful if there are existing stripe payments)
    bind = op.get_bind()
    if bind.dialect.name == 'mysql':
        # First, check if there are any stripe payments - if so, this will fail
        # In production, you'd want to migrate existing stripe payments first
        op.execute(
            "ALTER TABLE payments MODIFY COLUMN payment_method ENUM('cash', 'credit_card', 'debit_card', 'bank_transfer', 'e_wallet') NOT NULL"
        )
    # ### end Alembic commands ###