"""add_discovery_sessions_table

Revision ID: c2a5f0aecfbc
Revises: eeba3dc9a026
Create Date: 2026-02-12 22:24:15.903604

"""
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 = 'c2a5f0aecfbc'
down_revision: Union[str, Sequence[str], None] = 'eeba3dc9a026'
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('discovery_sessions',
    sa.Column('id', sa.UUID(), nullable=False),
    sa.Column('partner_type', sa.Enum('CORPORATE', 'EDUCATIONAL', name='partnertype'), nullable=False),
    sa.Column('corporate_partner_id', sa.UUID(), nullable=True),
    sa.Column('educational_partner_id', sa.UUID(), nullable=True),
    sa.Column('user_id', sa.UUID(), nullable=False),
    sa.Column('scheduled_date', sa.DateTime(), nullable=False),
    sa.Column('scheduled_time', sa.String(length=20), nullable=False),
    sa.Column('status', sa.Enum('SCHEDULED', 'COMPLETED', 'CANCELLED', 'NO_SHOW', name='discoverysessionstatus'), nullable=False),
    sa.Column('google_calendar_event_id', sa.String(length=255), nullable=True),
    sa.Column('google_calendar_link', sa.String(length=500), nullable=True),
    sa.Column('notes', sa.Text(), nullable=True),
    sa.Column('extra_data', postgresql.JSON(astext_type=sa.Text()), nullable=True),
    sa.Column('created_at', sa.DateTime(), nullable=False),
    sa.Column('updated_at', sa.DateTime(), nullable=False),
    sa.ForeignKeyConstraint(['corporate_partner_id'], ['corporate_partners.id'], ),
    sa.ForeignKeyConstraint(['educational_partner_id'], ['educational_partners.id'], ),
    sa.ForeignKeyConstraint(['user_id'], ['users.id'], ),
    sa.PrimaryKeyConstraint('id')
    )
    op.create_index(op.f('ix_discovery_sessions_corporate_partner_id'), 'discovery_sessions', ['corporate_partner_id'], unique=False)
    op.create_index(op.f('ix_discovery_sessions_educational_partner_id'), 'discovery_sessions', ['educational_partner_id'], unique=False)
    op.create_index(op.f('ix_discovery_sessions_id'), 'discovery_sessions', ['id'], unique=False)
    op.create_index(op.f('ix_discovery_sessions_partner_type'), 'discovery_sessions', ['partner_type'], unique=False)
    op.create_index(op.f('ix_discovery_sessions_scheduled_date'), 'discovery_sessions', ['scheduled_date'], unique=False)
    op.create_index(op.f('ix_discovery_sessions_status'), 'discovery_sessions', ['status'], unique=False)
    op.create_index(op.f('ix_discovery_sessions_user_id'), 'discovery_sessions', ['user_id'], unique=False)
    # ### end Alembic commands ###


def downgrade() -> None:
    """Downgrade schema."""
    # ### commands auto generated by Alembic - please adjust! ###
    op.drop_index(op.f('ix_discovery_sessions_user_id'), table_name='discovery_sessions')
    op.drop_index(op.f('ix_discovery_sessions_status'), table_name='discovery_sessions')
    op.drop_index(op.f('ix_discovery_sessions_scheduled_date'), table_name='discovery_sessions')
    op.drop_index(op.f('ix_discovery_sessions_partner_type'), table_name='discovery_sessions')
    op.drop_index(op.f('ix_discovery_sessions_id'), table_name='discovery_sessions')
    op.drop_index(op.f('ix_discovery_sessions_educational_partner_id'), table_name='discovery_sessions')
    op.drop_index(op.f('ix_discovery_sessions_corporate_partner_id'), table_name='discovery_sessions')
    op.drop_table('discovery_sessions')
    # ### end Alembic commands ###