45 lines
1.2 KiB
Python
45 lines
1.2 KiB
Python
|
"""Add expected_move column to kevin_stock_mentions (prompt v2).
|
||
|
|
|
||
|
|
Adds the KevinExpectedMove enum and a NOT NULL column with default
|
||
|
|
'unknown' so existing rows keep loading. New analyses produced by the
|
||
|
|
v2 prompt will populate it with a real directional view.
|
||
|
|
|
||
|
|
Revision ID: e5f6a7b8c9d0
|
||
|
|
Revises: d4e5f6a7b8c9
|
||
|
|
Create Date: 2026-05-28
|
||
|
|
"""
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import sqlalchemy as sa
|
||
|
|
from alembic import op
|
||
|
|
from sqlalchemy.dialects import postgresql
|
||
|
|
|
||
|
|
revision = "e5f6a7b8c9d0"
|
||
|
|
down_revision = "d4e5f6a7b8c9"
|
||
|
|
branch_labels = None
|
||
|
|
depends_on = None
|
||
|
|
|
||
|
|
_ENUM_NAME = "kevin_expected_move"
|
||
|
|
_VALUES = ("up_strong", "up_mild", "sideways", "down_mild", "down_strong", "unknown")
|
||
|
|
|
||
|
|
|
||
|
|
def upgrade() -> None:
|
||
|
|
enum_type = postgresql.ENUM(*_VALUES, name=_ENUM_NAME, create_type=False)
|
||
|
|
enum_type.create(op.get_bind(), checkfirst=True)
|
||
|
|
|
||
|
|
op.add_column(
|
||
|
|
"kevin_stock_mentions",
|
||
|
|
sa.Column(
|
||
|
|
"expected_move",
|
||
|
|
enum_type,
|
||
|
|
nullable=False,
|
||
|
|
server_default="unknown",
|
||
|
|
),
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def downgrade() -> None:
|
||
|
|
op.drop_column("kevin_stock_mentions", "expected_move")
|
||
|
|
op.execute(f"DROP TYPE IF EXISTS {_ENUM_NAME}")
|