"""create schemes table

Revision ID: 20260316_02
Revises: 20260316_01
Create Date: 2026-03-16 15:15:00
"""

from alembic import op
import sqlalchemy as sa


revision = "20260316_02"
down_revision = "20260316_01"
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.create_table(
        "schemes",
        sa.Column("id", sa.BigInteger(), primary_key=True, autoincrement=True),
        sa.Column("scheme_id", sa.String(length=32), nullable=False),
        sa.Column("source_upload_id", sa.String(length=32), nullable=False),
        sa.Column("name", sa.String(length=512), nullable=False),
        sa.Column("status", sa.String(length=32), nullable=False),
        sa.Column("normalized_elements_count", sa.Integer(), nullable=False),
        sa.Column("normalized_seats_count", sa.Integer(), nullable=False),
        sa.Column("normalized_groups_count", sa.Integer(), nullable=False),
        sa.Column("normalized_sectors_count", sa.Integer(), nullable=False),
        sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
        sa.ForeignKeyConstraint(["source_upload_id"], ["uploads.upload_id"], ondelete="RESTRICT"),
    )
    op.create_index("ix_schemes_scheme_id", "schemes", ["scheme_id"], unique=True)
    op.create_index("ix_schemes_source_upload_id", "schemes", ["source_upload_id"], unique=False)


def downgrade() -> None:
    op.drop_index("ix_schemes_source_upload_id", table_name="schemes")
    op.drop_index("ix_schemes_scheme_id", table_name="schemes")
    op.drop_table("schemes")