"""Add production strategy rank fields to trade setups.

Revision ID: 017_add_trade_setup_strategy_rank
Revises: 016
Create Date: 2026-07-03 20:15:00.000000
"""

from __future__ import annotations

from alembic import op
import sqlalchemy as sa


revision = "017_add_trade_setup_strategy_rank"
down_revision = "016"
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.add_column("trade_setups", sa.Column("strategy_rank", sa.Float(), nullable=True))
    op.add_column("trade_setups", sa.Column("volatility_percentile", sa.Float(), nullable=True))


def downgrade() -> None:
    op.drop_column("trade_setups", "volatility_percentile")
    op.drop_column("trade_setups", "strategy_rank")