"""import_batch: загрузки файлов сделок QUIK (D19)

Revision ID: 3b9f1c2a7d41
Revises: ef0ccff2e887
Create Date: 2026-10-15 00:00:00

"""

from typing import Sequence, Union

import sqlalchemy as sa
from sqlalchemy.dialects import postgresql

from alembic import op
from app.core.config import settings

# revision identifiers, used by Alembic.
revision: str = "3b9f1c2a7d41"
down_revision: Union[str, None] = "ef0ccff2e887"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None

_schema = settings.DATABASE_SCHEMA or None

# Типы зафиксированы в ревизии, как и в начальной миграции.
_BIGINT_PK = sa.BigInteger().with_variant(sa.Integer(), "sqlite")
_JSON = sa.JSON().with_variant(postgresql.JSONB(), "postgresql")


def _fk(table_column: str) -> str:
    return f"{_schema}.{table_column}" if _schema else table_column


def upgrade() -> None:
    """Upgrade schema."""
    op.create_table(
        "import_batch",
        sa.Column("id", _BIGINT_PK, sa.Identity(), primary_key=True),
        sa.Column("source", sa.String(10), nullable=False),
        sa.Column(
            "portfolio_id",
            sa.BigInteger(),
            sa.ForeignKey(_fk("portfolio.id")),
            nullable=False,
        ),
        sa.Column("file_name", sa.String(255), nullable=False),
        sa.Column("content_sha256", sa.String(64), nullable=False),
        sa.Column("date_from", sa.Date(), nullable=False),
        sa.Column("date_to", sa.Date(), nullable=False),
        sa.Column("rows_total", sa.Integer(), nullable=False),
        sa.Column("rows_loaded", sa.Integer(), nullable=False),
        sa.Column("rows_excluded", sa.Integer(), nullable=False),
        sa.Column("loaded_rows", _JSON, nullable=False),
        sa.Column(
            "created_at",
            sa.DateTime(timezone=True),
            server_default=sa.func.now(),
            nullable=False,
        ),
        sa.Column("created_by", sa.String(255), nullable=False),
        sa.Column(
            "updated_at",
            sa.DateTime(timezone=True),
            server_default=sa.func.now(),
            nullable=False,
        ),
        sa.Column("updated_by", sa.String(255), nullable=False),
        sa.UniqueConstraint(
            "content_sha256", "portfolio_id", name="uq_import_batch_content_portfolio"
        ),
        schema=_schema,
    )


def downgrade() -> None:
    """Downgrade schema."""
    op.drop_table("import_batch", schema=_schema)