"""remove mtgjson data columns

Revision ID: 2025_04_14_remove_mtgjson_data_columns
Revises: 2025_04_13_create_mtgjson_tables
Create Date: 2025-04-14 00:00:00.000000

"""
from alembic import op
import sqlalchemy as sa

# revision identifiers, used by Alembic.
revision = '2025_04_14_remove_mtgjson_data_columns'
down_revision = '2025_04_13_create_mtgjson_tables'
branch_labels = None
depends_on = None


def upgrade() -> None:
    # Remove data column from mtgjson_skus table
    op.drop_column('mtgjson_skus', 'data')
    
    # Remove data column from mtgjson_cards table
    op.drop_column('mtgjson_cards', 'data')


def downgrade() -> None:
    # Add data column back to mtgjson_cards table
    op.add_column('mtgjson_cards',
        sa.Column('data', sa.JSON(), nullable=True)
    )
    
    # Add data column back to mtgjson_skus table
    op.add_column('mtgjson_skus',
        sa.Column('data', sa.JSON(), nullable=True)
    ) 