"""Added MegaMillions model

Revision ID: f7ca6be06288
Revises: 677501c83fe5
Create Date: 2026-03-04 13:42:14.232761

"""
from typing import Sequence, Union

from alembic import op
import sqlalchemy as sa


# revision identifiers, used by Alembic.
revision: str = 'f7ca6be06288'
down_revision: Union[str, Sequence[str], None] = '677501c83fe5'
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    """Upgrade schema."""
    # ### commands auto generated by Alembic - please adjust! ###
    op.create_table('mega_millions_draws',
    sa.Column('draw_date', sa.Date(), nullable=False),
    sa.Column('main_ball1', sa.Integer(), nullable=False),
    sa.Column('main_ball2', sa.Integer(), nullable=False),
    sa.Column('main_ball3', sa.Integer(), nullable=False),
    sa.Column('main_ball4', sa.Integer(), nullable=False),
    sa.Column('main_ball5', sa.Integer(), nullable=False),
    sa.Column('mega_ball', sa.Integer(), nullable=False),
    sa.Column('megaplier', sa.Integer(), nullable=False),
    sa.PrimaryKeyConstraint('draw_date', name=op.f('pk_mega_millions_draws'))
    )
    with op.batch_alter_table('powerball_draws', schema=None) as batch_op:
        batch_op.alter_column('draw_date',
               existing_type=sa.DATETIME(),
               type_=sa.Date(),
               existing_nullable=False)
        batch_op.alter_column('main_ball1',
               existing_type=sa.INTEGER(),
               nullable=False)
        batch_op.alter_column('main_ball2',
               existing_type=sa.INTEGER(),
               nullable=False)
        batch_op.alter_column('main_ball3',
               existing_type=sa.INTEGER(),
               nullable=False)
        batch_op.alter_column('main_ball4',
               existing_type=sa.INTEGER(),
               nullable=False)
        batch_op.alter_column('main_ball5',
               existing_type=sa.INTEGER(),
               nullable=False)
        batch_op.alter_column('powerball',
               existing_type=sa.INTEGER(),
               nullable=False)
        batch_op.alter_column('power_play',
               existing_type=sa.INTEGER(),
               nullable=False)

    # ### end Alembic commands ###


def downgrade() -> None:
    """Downgrade schema."""
    # ### commands auto generated by Alembic - please adjust! ###
    with op.batch_alter_table('powerball_draws', schema=None) as batch_op:
        batch_op.alter_column('power_play',
               existing_type=sa.INTEGER(),
               nullable=True)
        batch_op.alter_column('powerball',
               existing_type=sa.INTEGER(),
               nullable=True)
        batch_op.alter_column('main_ball5',
               existing_type=sa.INTEGER(),
               nullable=True)
        batch_op.alter_column('main_ball4',
               existing_type=sa.INTEGER(),
               nullable=True)
        batch_op.alter_column('main_ball3',
               existing_type=sa.INTEGER(),
               nullable=True)
        batch_op.alter_column('main_ball2',
               existing_type=sa.INTEGER(),
               nullable=True)
        batch_op.alter_column('main_ball1',
               existing_type=sa.INTEGER(),
               nullable=True)
        batch_op.alter_column('draw_date',
               existing_type=sa.Date(),
               type_=sa.DATETIME(),
               existing_nullable=False)

    op.drop_table('mega_millions_draws')
    # ### end Alembic commands ###