"""Add distributor role and pending_validation field

Revision ID: 84ea2c08cb65
Revises: d770f939dc13
Create Date: 2026-01-26 19:50:43.141577

"""
from alembic import op
import sqlalchemy as sa


# revision identifiers, used by Alembic.
revision = '84ea2c08cb65'
down_revision = 'd770f939dc13'
branch_labels = None
depends_on = None


def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    with op.batch_alter_table('users', schema=None) as batch_op:
        batch_op.add_column(sa.Column('distributor_id', sa.String(length=36), nullable=True))
        batch_op.add_column(sa.Column('pending_validation', sa.Boolean(), server_default='false', nullable=False))
        batch_op.create_index(batch_op.f('ix_users_distributor_id'), ['distributor_id'], unique=False)
        batch_op.create_foreign_key(None, 'users', ['distributor_id'], ['id'])

    # ### end Alembic commands ###


def downgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    with op.batch_alter_table('users', schema=None) as batch_op:
        batch_op.drop_constraint(None, type_='foreignkey')
        batch_op.drop_index(batch_op.f('ix_users_distributor_id'))
        batch_op.drop_column('pending_validation')
        batch_op.drop_column('distributor_id')

    # ### end Alembic commands ###