"""Add rating fields to contracts table

Revision ID: dc3c13ef2107
Revises: c5e9c4fdd34a
Create Date: 2025-08-06 13:44:59.956952

"""
from alembic import op
import sqlalchemy as sa


# revision identifiers, used by Alembic.
revision = 'dc3c13ef2107'
down_revision = 'c5e9c4fdd34a'
branch_labels = None
depends_on = None


def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    # Check if the contracts table exists before trying to inspect it
    conn = op.get_bind()
    inspector = sa.inspect(conn)
    
    # Only proceed if the contracts table exists
    if 'contracts' in inspector.get_table_names():
        columns = [col['name'] for col in inspector.get_columns('contracts')]
        
        columns_to_add = []
        if 'contractor_rated' not in columns:
            columns_to_add.append(sa.Column('contractor_rated', sa.Boolean(), nullable=False, server_default='false'))
        if 'worker_rated' not in columns:
            columns_to_add.append(sa.Column('worker_rated', sa.Boolean(), nullable=False, server_default='false'))
        if 'mutual_rating_completed_date' not in columns:
            columns_to_add.append(sa.Column('mutual_rating_completed_date', sa.DateTime(), nullable=True))
        
        if columns_to_add:
            with op.batch_alter_table('contracts', schema=None) as batch_op:
                for column in columns_to_add:
                    batch_op.add_column(column)
    # If table doesn't exist, skip this migration (fresh database)

    # ### end Alembic commands ###


def downgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    # Check if the contracts table exists before trying to modify it
    conn = op.get_bind()
    inspector = sa.inspect(conn)
    
    if 'contracts' in inspector.get_table_names():
        with op.batch_alter_table('contracts', schema=None) as batch_op:
            batch_op.drop_column('mutual_rating_completed_date')
            batch_op.drop_column('worker_rated')
            batch_op.drop_column('contractor_rated')
    # ### end Alembic commands ###