yuxuanhui
40 lines
1.8 KiB
Python
40 lines
1.8 KiB
Python
"""SUPER candidates and immutable component evidence; retain all existing Alpha rows."""
|
|
|
|
import sqlalchemy as sa
|
|
from alembic import op
|
|
|
|
revision = "0021"
|
|
down_revision = "0020"
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade():
|
|
op.add_column("backtest_items", sa.Column("alpha_type", sa.String(20), nullable=False, server_default="REGULAR"))
|
|
op.add_column("backtest_items", sa.Column("selection", sa.Text(), nullable=True))
|
|
op.add_column("backtest_items", sa.Column("combo", sa.Text(), nullable=True))
|
|
op.create_index("ix_backtest_items_alpha_type", "backtest_items", ["alpha_type"])
|
|
op.create_table("super_selection_snapshots",
|
|
sa.Column("id", sa.String(36), primary_key=True),
|
|
sa.Column("job_id", sa.String(36), sa.ForeignKey("sync_jobs.id"), unique=True),
|
|
sa.Column("item_id", sa.String(36), sa.ForeignKey("backtest_items.id"), unique=True),
|
|
sa.Column("source", sa.String(20), nullable=False),
|
|
sa.Column("request", sa.JSON(), nullable=False),
|
|
sa.Column("request_hash", sa.String(64), nullable=False),
|
|
sa.Column("component_hash", sa.String(64)),
|
|
sa.Column("components", sa.JSON(), nullable=False),
|
|
sa.Column("raw", sa.JSON(), nullable=False),
|
|
sa.Column("complete", sa.Boolean(), nullable=False),
|
|
sa.Column("total", sa.Integer()),
|
|
sa.Column("warnings", sa.JSON(), nullable=False),
|
|
sa.Column("observed_at", sa.DateTime(timezone=True), nullable=False))
|
|
for key in ("request_hash", "component_hash"):
|
|
op.create_index(f"ix_super_selection_snapshots_{key}", "super_selection_snapshots", [key])
|
|
|
|
|
|
def downgrade():
|
|
op.drop_table("super_selection_snapshots")
|
|
op.drop_index("ix_backtest_items_alpha_type", table_name="backtest_items")
|
|
for key in ("combo", "selection", "alpha_type"):
|
|
op.drop_column("backtest_items", key)
|