"""stock_reference：个股参考数据（tushare 参考数据版块，按股按分类存 JSON 快照懒加载）

Revision ID: 20260907_02
Revises: 20260907_01
Create Date: 2026-09-07
"""
from typing import Sequence, Union

import sqlalchemy as sa
from alembic import op

revision: str = "20260907_02"
down_revision: Union[str, Sequence[str], None] = "20260907_01"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    op.create_table(
        "stock_reference",
        sa.Column("id", sa.Integer(), autoincrement=True, nullable=False),
        sa.Column("ts_code", sa.String(length=12), nullable=False),
        sa.Column("kind", sa.String(length=24), nullable=False),
        sa.Column("rows_json", sa.Text(), nullable=True),
        sa.Column("updated_at", sa.DateTime(), nullable=True),
        sa.PrimaryKeyConstraint("id"),
        sa.UniqueConstraint("ts_code", "kind", name="uq_stock_reference_code_kind"),
    )
    op.create_index("ix_stock_reference_ts_code", "stock_reference", ["ts_code"])


def downgrade() -> None:
    op.drop_index("ix_stock_reference_ts_code", table_name="stock_reference")
    op.drop_table("stock_reference")