"""Add master_taxonomy_job table for durable large-catalog taxonomy runs.

Revision ID: 0052_add_master_taxonomy_job
Revises: 0051_add_collection_icon_compare_price
"""

from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects import postgresql


revision = "0052_add_master_taxonomy_job"
down_revision = "0051_add_collection_icon_compare_price"
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.create_table(
        "master_taxonomy_job",
        sa.Column("id", sa.String(length=36), nullable=False),
        sa.Column("job_type", sa.String(length=32), nullable=False),
        sa.Column("status", sa.String(length=32), nullable=False, server_default="queued"),
        sa.Column("params", postgresql.JSONB(astext_type=sa.Text()), nullable=True),
        sa.Column("progress", postgresql.JSONB(astext_type=sa.Text()), nullable=True),
        sa.Column("result", postgresql.JSONB(astext_type=sa.Text()), nullable=True),
        sa.Column("error", sa.Text(), nullable=True),
        sa.Column("notes", sa.Text(), nullable=True),
        sa.Column("requested_at", sa.DateTime(timezone=True), server_default=sa.text("now()"), nullable=False),
        sa.Column("started_at", sa.DateTime(timezone=True), nullable=True),
        sa.Column("finished_at", sa.DateTime(timezone=True), nullable=True),
        sa.Column("attempts", sa.Integer(), nullable=False, server_default="0"),
        sa.Column("max_attempts", sa.Integer(), nullable=False, server_default="3"),
        sa.Column("locked_at", sa.DateTime(timezone=True), nullable=True),
        sa.Column("locked_by", sa.String(length=128), nullable=True),
        sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.text("now()"), nullable=False),
        sa.Column("updated_at", sa.DateTime(timezone=True), server_default=sa.text("now()"), nullable=False),
        sa.PrimaryKeyConstraint("id"),
    )
    op.create_index("ix_master_taxonomy_job_job_type", "master_taxonomy_job", ["job_type"])
    op.create_index("ix_master_taxonomy_job_status", "master_taxonomy_job", ["status"])
    op.create_index("ix_master_taxonomy_job_requested_at", "master_taxonomy_job", ["requested_at"])


def downgrade() -> None:
    op.drop_index("ix_master_taxonomy_job_requested_at", table_name="master_taxonomy_job")
    op.drop_index("ix_master_taxonomy_job_status", table_name="master_taxonomy_job")
    op.drop_index("ix_master_taxonomy_job_job_type", table_name="master_taxonomy_job")
    op.drop_table("master_taxonomy_job")