"""etf_basic：场内 ETF 列表（东财快照 + 规模字段）

Revision ID: 20260905_01
Revises: 20260902_01
Create Date: 2026-09-05
"""
from typing import Sequence, Union

import sqlalchemy as sa
from alembic import op

revision: str = "20260905_01"
down_revision: Union[str, Sequence[str], None] = "20260902_01"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    op.create_table(
        "etf_basic",
        sa.Column("id", sa.Integer(), autoincrement=True, nullable=False),
        sa.Column("ts_code", sa.String(length=12), nullable=False),
        sa.Column("symbol", sa.String(length=10), nullable=False),
        sa.Column("name", sa.String(length=64), nullable=False),
        sa.Column("exchange", sa.String(length=8), nullable=False),
        sa.Column("list_date", sa.String(length=8), nullable=True),
        sa.Column("total_mv", sa.Float(), nullable=True),
        sa.Column("circ_mv", sa.Float(), nullable=True),
        sa.Column("turnover_rate", sa.Float(), nullable=True),
        sa.Column("updated_at", sa.DateTime(), nullable=True),
        sa.PrimaryKeyConstraint("id"),
        sa.UniqueConstraint("ts_code", name="uq_etf_basic_ts_code"),
    )
    op.create_index("ix_etf_basic_ts_code", "etf_basic", ["ts_code"])
    op.create_index("ix_etf_basic_symbol", "etf_basic", ["symbol"])


def downgrade() -> None:
    op.drop_index("ix_etf_basic_symbol", table_name="etf_basic")
    op.drop_index("ix_etf_basic_ts_code", table_name="etf_basic")
    op.drop_table("etf_basic")