"""Add indexes to active_trades.stock_code and active_trades.status

Revision ID: 0005
Revises: 0004
Create Date: 2026-03-20
"""
from alembic import op
import sqlalchemy as sa

# revision identifiers
revision = "0005"
down_revision = "0004"
branch_labels = None
depends_on = None


def upgrade() -> None:
    conn = op.get_bind()
    result = conn.execute(sa.text(
        "SELECT 1 FROM pg_indexes WHERE tablename='active_trades' AND indexname='ix_active_trades_stock_code'"
    ))
    if not result.fetchone():
        op.create_index("ix_active_trades_stock_code", "active_trades", ["stock_code"])


def downgrade() -> None:
    op.drop_index("ix_active_trades_stock_code", table_name="active_trades")