File size: 2,663 Bytes
8fb6433 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 | #!/usr/bin/env python3
"""Convert H3 V1/V2 reference bundles without rendering or deleting them."""
import argparse
import json
from pathlib import Path
import sys
sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
from reference_cache_migration import LEGACY_FORMATS, ReferenceCacheMigrator
def main(argv=None):
parser = argparse.ArgumentParser(description=__doc__)
parser.add_argument("--output-root", required=True, type=Path,
help="ComfyUI output directory containing h3_reference_cache/h3_chains")
scope = parser.add_mutually_exclusive_group(required=True)
scope.add_argument("--metadata", action="append", help="Exact scene cache JSON (repeatable)")
scope.add_argument("--run", help="Convert only this project's reference_cache directory")
scope.add_argument("--all", action="store_true", help="Scan shared and project-local reference caches")
mode = parser.add_mutually_exclusive_group()
mode.add_argument("--apply", action="store_true", help="Write resumable V3 conversions; bundles retire only after a successful H3 render save")
mode.add_argument("--dry-run", action="store_true", help="Read-only report (the default)")
args = parser.parse_args(argv)
migrator = ReferenceCacheMigrator(args.output_root)
if not migrator.root.is_dir():
parser.error("output root does not exist")
if args.run and (Path(args.run).name != args.run or args.run in (".", "..")):
parser.error("--run must be one project directory name")
if args.run:
parent = migrator.absolute(Path("h3_chains") / args.run / "reference_cache")
if not parent.is_dir():
parser.error("project reference_cache directory does not exist")
paths = [path for path in parent.glob("scene_*.json")
if not path.name.endswith((".converted.json", ".retired.json"))]
else:
paths = [migrator.absolute(path) for path in args.metadata] if args.metadata else list(migrator.legacy_manifests())
failures = 0
for path in sorted(set(paths)):
try:
if migrator.read(path).get("format") not in LEGACY_FORMATS:
continue
result = migrator.convert(path, apply=args.apply)
print(json.dumps(result))
except Exception as exc:
failures += 1
print(json.dumps({"status": "error", "metadata": str(path), "error": str(exc)}))
print(json.dumps({"mode": "apply" if args.apply else "dry_run", "errors": failures,
"legacy_bundles_deleted": 0}))
return 1 if failures else 0
if __name__ == "__main__":
raise SystemExit(main())
|