Initial public khodo snapshot

generator/__main__.py

@@ -0,0 +1,56 @@
+from __future__ import annotations
+
+import argparse
+import shutil
+from pathlib import Path
+
+from generator.arrange.generate import generate_performance_scores
+from generator.practice.generate import generate_practice_scores
+from generator.score import count_json_files, validate_generated_tree
+from generator.summary import validate_expected_performance_summary, validate_practice_variants
+
+
+def parse_args() -> argparse.Namespace:
+  parser = argparse.ArgumentParser(description="Generate Product Vamp package score assets.")
+  parser.add_argument("--data-dir", type=Path, required=True)
+  parser.add_argument("--output-dir", type=Path, required=True)
+  parser.add_argument("--static-assets-dir", type=Path, default=Path("package/assets"))
+  parser.add_argument("--clean", action="store_true")
+  parser.add_argument("--check", action="store_true")
+  return parser.parse_args()
+
+
+def main() -> int:
+  args = parse_args()
+  data_dir = args.data_dir.resolve()
+  output_dir = args.output_dir.resolve()
+  static_assets_dir = args.static_assets_dir.resolve()
+
+  if args.clean and output_dir.exists():
+    shutil.rmtree(output_dir)
+  output_dir.mkdir(parents=True, exist_ok=True)
+
+  fonts_src = static_assets_dir / "fonts"
+  if fonts_src.exists():
+    shutil.copytree(fonts_src, output_dir / "fonts", dirs_exist_ok=True)
+
+  performance_count = generate_performance_scores(data_dir / "songs", output_dir / "songs")
+  practice_count = generate_practice_scores(data_dir / "practice", output_dir / "practice")
+
+  validate_generated_tree(output_dir)
+  print(
+    "Generated Product Vamp assets: "
+    f"{performance_count} performance scores, "
+    f"{practice_count} practice score variants, "
+    f"{count_json_files(output_dir)} total JSON files."
+  )
+
+  if args.check:
+    validate_expected_performance_summary(data_dir / "expected" / "performance_summary.json", output_dir / "songs")
+    validate_practice_variants(output_dir / "practice")
+    print("Generator check passed.")
+  return 0
+
+
+if __name__ == "__main__":
+  raise SystemExit(main())