1 files changed, 23 insertions, 0 deletions
diff --git a/shared/alembic/versions/004_add_signal_detail_columns.py b/shared/alembic/versions/004_add_signal_detail_columns.py
new file mode 100644
index 0000000..7a8a77b
--- /dev/null
+++ b/shared/alembic/versions/004_add_signal_detail_columns.py
@@ -0,0 +1,23 @@
+"""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")