"""add_env_to_trading_tables

Revision ID: b96396dcfaff
Revises: 2360941fa0bd
Create Date: 2025-05-27 11:59:55.405410

"""
from typing import Sequence, Union

from alembic import op
import sqlalchemy as sa


# revision identifiers, used by Alembic.
revision: str = 'b96396dcfaff'
down_revision: Union[str, None] = '2360941fa0bd'
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('operations', sa.Column('env', sa.String(length=50), nullable=False))
    op.add_column('portfolios', sa.Column('env', sa.String(length=50), nullable=False))
    op.add_column('transactions', sa.Column('env', sa.String(length=50), nullable=False))
    # ### end Alembic commands ###


def downgrade() -> None:
    # ### commands auto generated by Alembic - please adjust! ###
    op.drop_column('transactions', 'env')
    op.drop_column('portfolios', 'env')
    op.drop_column('operations', 'env')
    # ### end Alembic commands ###
