import sqlite3, sys, json, datetime
sys.stdout.reconfigure(encoding="utf-8", errors="replace")

DB = r"E:/ProgramData/.workbuddy/workbuddy.db"
TARGET = ("deleted_at IS NULL AND status='ACTIVE' AND schedule_type='once' "
          "AND next_run_at IS NULL")
SELF = "45d4e2d4-a73d-4cdf-ac78-1047326f0f8d"

c = sqlite3.connect(DB, timeout=20)
c.execute("PRAGMA busy_timeout=10000")

before = c.execute(f"select count(*) from automations where {TARGET}").fetchone()[0]
print("BEFORE =", before)

rows = c.execute(
    f"select id, name from automations where {TARGET} order by name").fetchall()
print("TARGET_ROWS =", len(rows))
assert before == len(rows), "盘点与列表不一致，停手"

# 只逐条 UPDATE（不批量），每条改完立刻回读核对
now_ms = int(datetime.datetime.now().timestamp() * 1000)
results = []
for aid, name in rows:
    assert aid != SELF, "拒改：本会话自己的排期"
    cur = c.execute(
        f"update automations set status='PAUSED', updated_at=? where id=? and {TARGET}",
        (now_ms, aid))
    # 立刻回读核对（不把「返回成功」当「已生效」）
    chk = c.execute("select status, schedule_type, next_run_at, deleted_at "
                    "from automations where id=?", (aid,)).fetchone()
    ok = (cur.rowcount == 1 and chk[0] == 'PAUSED' and chk[1] == 'once'
          and chk[2] is None and chk[3] is None)
    results.append((aid, name, cur.rowcount, chk, ok))
    print(f"  [{'OK ' if ok else 'FAIL'}] {name}  rowcount={cur.rowcount}  now={chk}")

c.commit()

after = c.execute(f"select count(*) from automations where {TARGET}").fetchone()[0]
print("AFTER =", after)

# 回归：周期排期一个没少、待跑的一次性排期没被动
print("RECURRING_ACTIVE_AFTER =", c.execute(
    "select count(*) from automations where deleted_at is null and status='ACTIVE' "
    "and schedule_type='recurring'").fetchone()[0], "(before=3)")
print("ONCE_PENDING_STILL_ACTIVE =", c.execute(
    "select count(*) from automations where deleted_at is null and status='ACTIVE' "
    "and schedule_type='once' and next_run_at is not null").fetchone()[0], "(before=2)")
print("TOTAL_ROWS_AFTER =", c.execute("select count(*) from automations").fetchone()[0],
      "(before=141  ⛔不删行)")
print("INTEGRITY_AFTER =", c.execute("PRAGMA integrity_check").fetchone()[0])
c.close()

ok_all = after == 0 and all(r[4] for r in results)
print("VERDICT =", "PASS" if ok_all else "FAIL")