You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
44 lines
1.9 KiB
Python
44 lines
1.9 KiB
Python
"""add earnings_surprises table
|
|
|
|
Revision ID: d5e6f7a8b9c0
|
|
Revises: c4d5e6f7a8b9
|
|
Create Date: 2026-03-24
|
|
"""
|
|
from typing import Sequence, Union
|
|
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
from sqlalchemy.dialects import postgresql
|
|
|
|
revision: str = "d5e6f7a8b9c0"
|
|
down_revision: Union[str, Sequence[str], None] = "c4d5e6f7a8b9"
|
|
branch_labels: Union[str, Sequence[str], None] = None
|
|
depends_on: Union[str, Sequence[str], None] = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
conn = op.get_bind()
|
|
if not conn.dialect.has_table(conn, "earnings_surprises"):
|
|
op.create_table(
|
|
"earnings_surprises",
|
|
sa.Column("id", postgresql.UUID(as_uuid=True), primary_key=True),
|
|
sa.Column("ticker", sa.String(10), nullable=False),
|
|
sa.Column("fiscal_date_ending", postgresql.TIMESTAMP(timezone=True), nullable=False),
|
|
sa.Column("reported_date", postgresql.TIMESTAMP(timezone=True), nullable=True),
|
|
sa.Column("reported_eps", sa.Float(), nullable=True),
|
|
sa.Column("estimated_eps", sa.Float(), nullable=True),
|
|
sa.Column("surprise", sa.Float(), nullable=True),
|
|
sa.Column("surprise_percentage", sa.Float(), nullable=True),
|
|
sa.Column("data_source", sa.String(50), server_default="ALPHA_VANTAGE"),
|
|
sa.Column("created_at", postgresql.TIMESTAMP(timezone=True)),
|
|
sa.Column("updated_at", postgresql.TIMESTAMP(timezone=True)),
|
|
sa.UniqueConstraint("ticker", "fiscal_date_ending", name="uq_earnings_surprise"),
|
|
)
|
|
op.create_index("ix_earnings_surprises_ticker", "earnings_surprises", ["ticker"])
|
|
op.create_index("idx_earnings_ticker_date", "earnings_surprises", ["ticker", "fiscal_date_ending"])
|
|
op.create_index("idx_earnings_reported_date", "earnings_surprises", ["reported_date"])
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.drop_table("earnings_surprises")
|