"""Add application fields to trainer_tickets

Revision ID: 003_add_application_fields
Revises: 001_add_video_courses
Create Date: 2025-03-07

"""
from alembic import op
import sqlalchemy as sa

# revision identifiers, used by Alembic.
revision = '003_add_application_fields'
down_revision = '001_add_video_courses'
branch_labels = None
depends_on = None


def upgrade():
    # Add application-specific columns to trainer_tickets
    op.add_column('trainer_tickets', sa.Column('application_budget', sa.Float(), nullable=True))
    op.add_column('trainer_tickets', sa.Column('application_city', sa.String(255), nullable=True))
    op.add_column('trainer_tickets', sa.Column('application_timeline', sa.Text(), nullable=True))
    op.add_column('trainer_tickets', sa.Column('application_team_size', sa.Integer(), nullable=True))


def downgrade():
    op.drop_column('trainer_tickets', 'application_team_size')
    op.drop_column('trainer_tickets', 'application_timeline')
    op.drop_column('trainer_tickets', 'application_city')
    op.drop_column('trainer_tickets', 'application_budget')