"""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")