"""Warm existing cloud subscription baselines once after release.

Revision ID: 504
Revises: 503
"""

import uuid

import sqlalchemy as sa

from alembic import op

revision = "504"
down_revision = "503"
branch_labels = None
depends_on = None

_JOB = "intelligence.cloud_scope_baseline_bootstrap"


def upgrade() -> None:
    bind = op.get_bind()
    if bind.execute(
        sa.text("SELECT 1 FROM scheduled_jobs WHERE name = :name LIMIT 1"),
        {"name": _JOB},
    ).scalar():
        return
    bind.execute(
        sa.text(
            """
            INSERT INTO scheduled_jobs
                (id, name, display_name, description, handler_key,
                 trigger_type, interval_seconds, cron_expr, timezone,
                 enabled, config, run_count, failure_count, created_at, updated_at)
            VALUES
                (:id, :name, :display_name, :description, :name,
                 'interval', 60, NULL, 'UTC',
                 1, :config, 0, 0, UTC_TIMESTAMP(), UTC_TIMESTAMP())
            """
        ),
        {
            "id": str(uuid.uuid4()),
            "name": _JOB,
            "display_name": "AI 大脑云订阅基线首次回填",
            "description": (
                "部署后约一分钟回填所有有效订阅的确定性健康基线；成功后自动停用。"
                "异常订阅最多自动重试六次，日常任务仍会持续重试。"
            ),
            "config": '{"max_attempts": 6, "timeout_secs": 7200}',
        },
    )


def downgrade() -> None:
    op.get_bind().execute(
        sa.text("DELETE FROM scheduled_jobs WHERE name = :name"),
        {"name": _JOB},
    )