summaryrefslogtreecommitdiff
path: root/shared/alembic/versions/004_add_signal_detail_columns.py
blob: 4009b6ef9daa1384b9c6b33f77ef33f8dcab8891 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
"""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")