otivm/pipeline/seed_promote.py

"""
seed_promote.py - TESSERA 4.0 seed promotion
Promotes draft rows in staging_otivm.sqlite3 to current.
Run on the Dell (tessera-pipeline CT) after seed_extract.py completes:
    /opt/tessera-pipeline/venv/bin/python3 /opt/tessera-pipeline/seed_promote.py
"""

import sqlite3
import sys
import os
from datetime import datetime, timezone

STAGING_DB  = "/tmp/staging_otivm.sqlite3"
RUN_KEY     = "tessera3-seed-2026-04-26"
H9_EXPECTED = 2401
H5_WAYPOINTS = [
    ("Ostia",      "851e805bfffffff"),
    ("Capua",      "851e8333fffffff"),
    ("Brundisium", "851e8ba3fffffff"),
    ("Carthago",   "85386e23fffffff"),
    ("Alexandria", "853f5ba7fffffff"),
]

try:
    import h3
except ImportError:
    sys.exit("ERROR: h3 library not installed. Activate venv first.")


def now_utc():
    return datetime.now(timezone.utc).strftime("%Y-%m-%dT%H:%M:%SZ")


def promote():
    print(f"[{now_utc()}] seed_promote.py starting")
    print(f"  Staging: {STAGING_DB}")

    if not os.path.exists(STAGING_DB):
        sys.exit(f"ERROR: {STAGING_DB} not found. Run seed_extract.py first.")

    con = sqlite3.connect(STAGING_DB)
    con.execute("PRAGMA foreign_keys = ON")

    run = con.execute(
        "SELECT id, status, row_count FROM pipeline_runs WHERE run_key = ?",
        (RUN_KEY,)
    ).fetchone()

    if not run:
        con.close()
        sys.exit(f"ERROR: pipeline run '{RUN_KEY}' not found in staging db.")

    run_id, run_status, row_count = run
    if run_status != 2:
        con.close()
        sys.exit(
            f"ERROR: pipeline run status is {run_status}, expected 2 (complete).\n"
            f"Re-run seed_extract.py to completion before promoting."
        )

    print(f"  pipeline_run id={run_id}, status=2 (complete), row_count={row_count}")
    print(f"\n  Pre-promotion verification:")
    all_ok = True
    h5_ints = []

    for city, h5_text in H5_WAYPOINTS:
        h5_int = h3.str_to_int(h5_text)
        h5_ints.append(h5_int)

        draft_count = con.execute(
            "SELECT COUNT(*) FROM tessera_cells WHERE h5 = ? AND status = 1 AND run_id = ?",
            (h5_int, run_id)
        ).fetchone()[0]

        current_count = con.execute(
            "SELECT COUNT(*) FROM tessera_cells WHERE h5 = ? AND status = 2",
            (h5_int,)
        ).fetchone()[0]

        cov = con.execute(
            "SELECT status, h9_total, h9_current FROM h5_coverage WHERE h5 = ?",
            (h5_int,)
        ).fetchone()

        ok = draft_count == H9_EXPECTED
        status_str = "OK" if ok else f"ERROR: expected {H9_EXPECTED}"
        print(f"    {city}: draft={draft_count} current={current_count} "
              f"h5_cov_status={cov[0] if cov else 'MISSING'} [{status_str}]")
        if not ok:
            all_ok = False

    if not all_ok:
        con.close()
        sys.exit("ERROR: Pre-promotion verification failed. Do not promote.")

    print(f"\n  All five H5 hexes verified. Promoting...")

    promoted = con.execute(
        "UPDATE tessera_cells SET status = 2 WHERE run_id = ? AND status = 1",
        (run_id,)
    ).rowcount
    print(f"  tessera_cells promoted: {promoted} rows (1->2)")

    for h5_int in h5_ints:
        con.execute(
            "UPDATE h5_coverage SET status = 2, h9_current = ?, last_updated = ? WHERE h5 = ?",
            (H9_EXPECTED, now_utc(), h5_int)
        )

    con.commit()
    print(f"  h5_coverage updated: {len(h5_ints)} rows (1->2)")

    print(f"\n  Post-promotion verification:")
    grand_total = 0
    all_complete = True

    for city, h5_text in H5_WAYPOINTS:
        h5_int = h3.str_to_int(h5_text)

        current_count = con.execute(
            "SELECT COUNT(*) FROM tessera_cells WHERE h5 = ? AND status = 2",
            (h5_int,)
        ).fetchone()[0]

        cov = con.execute(
            "SELECT status, h9_current FROM h5_coverage WHERE h5 = ?",
            (h5_int,)
        ).fetchone()

        complete = current_count == H9_EXPECTED and cov and cov[0] == 2
        label = "COMPLETE" if complete else "ERROR"
        print(f"    {city}: current={current_count} h5_status={cov[0] if cov else 'MISSING'} [{label}]")
        grand_total += current_count
        if not complete:
            all_complete = False

    con.close()
    print(f"\n  Total current rows: {grand_total}")

    if all_complete:
        print(f"\n[{now_utc()}] Promotion complete. staging_otivm.sqlite3 is ready.")
        print(f"\nNext steps (on root@ai):")
        print(f"  lxc file pull tessera-pipeline/tmp/staging_otivm.sqlite3 /tmp/staging_otivm.sqlite3")
        print(f"  scp -i /home/sandor/.ssh/tessera-transfer /tmp/staging_otivm.sqlite3 sandor@10.0.0.23:/tmp/staging_otivm.sqlite3")
        print(f"\nOn OTIVM container:")
        print(f"  mv /tmp/staging_otivm.sqlite3 /home/otivm/OTIVM/data/otivm.sqlite3")
        print(f"  chown otivm:otivm /home/otivm/OTIVM/data/otivm.sqlite3")
        print(f"  sqlite3 data/otivm.sqlite3 'SELECT h5, h9_current, status FROM h5_coverage;'")
    else:
        print(f"\n[{now_utc()}] ERROR: Post-promotion verification failed.")
        sys.exit(1)


if __name__ == "__main__":
    promote()