"""Reject new dependency errors while recording known official-image warnings."""
import json
import subprocess
from pathlib import Path


def check(python):
    result = subprocess.run(["/usr/bin/python3", "-m", "pip", "--python", python, "check"], capture_output=True, text=True)
    return [] if result.returncode == 0 else result.stdout.strip().splitlines()


baseline = check("/usr/bin/python3")
accepted = {line for line in baseline if line.startswith(("pygobject ", "nixl "))}
report = {"official_image_errors": baseline, "environments": {}}
for environment in ("/opt/speculators_venv", "/opt/vllm_venv"):
    errors = check(environment + "/bin/python")
    new_errors = sorted(set(errors) - accepted)
    report["environments"][environment] = {"errors": errors, "new_errors": new_errors}
    print(environment + ": " + ("no new dependency errors" if not new_errors else "; ".join(new_errors)))
Path("logs/ngram_setup/dependency-checks.json").write_text(json.dumps(report, indent=2) + "\n")
if any(result["new_errors"] for result in report["environments"].values()):
    raise SystemExit(1)