"""Add conviction, stop_loss, take_profit columns to signals table.

Revision ID: 004
Revises: 003
"""

import sqlalchemy as sa
from alembic import op

revision = "004"
down_revision = "003"


def upgrade():
    op.add_column(
        "signals", sa.Column("conviction", sa.Float, nullable=False, server_default="1.0")
    )
    op.add_column("signals", sa.Column("stop_loss", sa.Numeric, nullable=True))
    op.add_column("signals", sa.Column("take_profit", sa.Numeric, nullable=True))


def downgrade():
    op.drop_column("signals", "take_profit")
    op.drop_column("signals", "stop_loss")
    op.drop_column("signals", "conviction")