"""add accepted workers count

Revision ID: add_accepted_workers_count
Revises: 7839e3474de5
Create Date: 2025-10-15 12:00:00.000000

"""
from alembic import op
import sqlalchemy as sa

# revision identifiers, used by Alembic.
revision = 'add_accepted_workers_count'
down_revision = '7839e3474de5'
branch_labels = None
depends_on = None
from alembic import op
import sqlalchemy as sa

def upgrade():
    # Add accepted_workers_count column with default value 0
    conn = op.get_bind()
    inspector = sa.inspect(conn)
    
    if 'jobs' in inspector.get_table_names():
        columns = [col['name'] for col in inspector.get_columns('jobs')]
        
        if 'accepted_workers_count' not in columns:
            op.add_column('jobs', sa.Column('accepted_workers_count', sa.Integer(), nullable=False, server_default='0'))

def downgrade():
    # Remove the column if needed
    conn = op.get_bind()
    inspector = sa.inspect(conn)
    
    if 'jobs' in inspector.get_table_names():
        columns = [col['name'] for col in inspector.get_columns('jobs')]
        
        if 'accepted_workers_count' in columns:
            op.drop_column('jobs', 'accepted_workers_count')