"""add_status_and_training_to_rfp_profile

Revision ID: ddcfbfa665b5
Revises: a4714688877a
Create Date: 2026-02-25 03:34:40.620557

"""
from typing import Sequence, Union

from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects import postgresql

# revision identifiers, used by Alembic.
revision: str = 'ddcfbfa665b5'
down_revision: Union[str, Sequence[str], None] = 'a4714688877a'
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.add_column('rfp_profile', sa.Column('status', sa.Enum('DRAFT', 'IN_PROGRESS', 'COMPLETED', name='rfpprofilestatus', native_enum=False), nullable=True))
    op.add_column('rfp_profile', sa.Column('is_training_data', sa.Boolean(), nullable=True))
    op.create_index(op.f('rfp_profile_status_idx'), 'rfp_profile', ['status'], unique=False)
    # ### end Alembic commands ###


def downgrade() -> None:
    """Downgrade schema."""
    # ### commands auto generated by Alembic - please adjust! ###
    op.drop_index(op.f('rfp_profile_status_idx'), table_name='rfp_profile')
    op.drop_column('rfp_profile', 'is_training_data')
    op.drop_column('rfp_profile', 'status')
    # ### end Alembic commands ###
