"""Added column commission_place

Revision ID: adf3d2991896
Revises: 0ee52ab23e66
Create Date: 2025-10-26 13:37:33.662318

"""
from typing import Sequence, Union

from alembic import op
import sqlalchemy as sa
from sqlalchemy import inspect


# revision identifiers, used by Alembic.
revision: str = 'adf3d2991896'
down_revision: Union[str, Sequence[str], None] = '0ee52ab23e66'
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! ###
    bind = op.get_bind()
    inspector = inspect(bind)
    columns_user_deals = [col['name'] for col in inspector.get_columns('user_deals')]
    if 'commission_fee' not in columns_user_deals:
        op.add_column('user_deals', sa.Column('commission_fee', sa.String(), server_default='', nullable=True))
    if 'commission_place' not in columns_user_deals:
        op.add_column('user_deals', sa.Column('commission_place', sa.String(), server_default='', nullable=True))

    columns_user_risk_mgmt = [col['name'] for col in inspector.get_columns('user_risk_management')]
    if 'commission_place' not in columns_user_risk_mgmt:
        op.add_column('user_risk_management',
                      sa.Column('commission_place', sa.String(), server_default='', nullable=False))


def downgrade() -> None:
    """Downgrade schema."""
    # ### commands auto generated by Alembic - please adjust! ###
    op.drop_column('user_risk_management', 'commission_place')
    op.drop_column('user_deals', 'commission_place')
    op.drop_column('user_deals', 'commission_fee')
    # ### end Alembic commands ###