"""add_speaks_english_to_driver_profile

Revision ID: 5caf8ba3ed4d
Revises: 9f2fbe81a055
Create Date: 2026-01-31 16:02:13.521306

"""
from typing import Sequence, Union

from alembic import op
import sqlalchemy as sa
import sqlmodel


# revision identifiers, used by Alembic.
revision: str = '5caf8ba3ed4d'
down_revision: Union[str, None] = '9f2fbe81a055'
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    # ### commands auto generated by Alembic - please adjust! ###
    op.add_column('driver_profiles', sa.Column('shift', sqlmodel.sql.sqltypes.AutoString(), nullable=True))
    op.add_column('driver_profiles', sa.Column('payment_methods', sqlmodel.sql.sqltypes.AutoString(), nullable=True))
    op.add_column('driver_profiles', sa.Column('speaks_english', sa.Boolean(), nullable=True))
    op.add_column('taxis', sa.Column('rating', sa.Float(), nullable=False, server_default=sa.text('5.0')))
    op.add_column('taxis', sa.Column('english_speaking', sa.Boolean(), nullable=False, server_default=sa.text('false')))
    # ### end Alembic commands ###


def downgrade() -> None:
    # ### commands auto generated by Alembic - please adjust! ###
    op.drop_column('taxis', 'english_speaking')
    op.drop_column('taxis', 'rating')
    op.drop_column('driver_profiles', 'speaks_english')
    op.drop_column('driver_profiles', 'payment_methods')
    op.drop_column('driver_profiles', 'shift')
    # ### end Alembic commands ###

