add get_batch_runs.py

Author: Donaim

micall/utils/analyze_kive_batches/analyze_kive_batches.py

@@ -12,6 +12,7 @@
 
 from .download import download
 from .make_stats_1 import make_stats_1
+from .get_batch_runs import get_batch_runs
 
 
 def dir_path(string: str) -> DirPath:
@@ -37,6 +38,12 @@ def cli_parser() -> argparse.ArgumentParser:
     all.add_argument("--properties", type=Path, required=True,
                      help="Additional properties associated with particular images.")
 
+    get_batch_runs = mode_parsers.add_parser("get_batch_runs", help="The main entry to this script. Runs get_batch_runs other subentries.")
+    get_batch_runs.add_argument("--batch", type=str, required=True,
+                                help="The name of the batch to download the runs info for.")
+    get_batch_runs.add_argument("--target", type=dir_path, required=True,
+                                help="Target file where to put the runs info to.")
+
     download = mode_parsers.add_parser("download")
     download.add_argument("--json-file", type=Path, required=True,
                           help="The big json file with all the run infos.")
@@ -64,8 +71,10 @@ def parse_args(argv: Sequence[str]) -> argparse.Namespace:
 
 
 def main_typed(subcommand: str, args: argparse.Namespace) -> None:
-    if args.subcommand == 'download':
-        download(json_file=args.json_file, root=args.root)
+    if args.subcommand == 'all':
+        raise NotImplementedError()
+    elif args.subcommand == 'get-batch-runs':
+        get_batch_runs(batch=args.batch, target=args.target)
     elif args.subcommand == 'make-stats-1':
         make_stats_1(input=args.input, output=args.output)
     else:

micall/utils/analyze_kive_batches/batch.py

@@ -0,0 +1,4 @@
+
+from typing import NewType
+
+BatchName = NewType("BatchName", str)

micall/utils/analyze_kive_batches/get_batch_runs.py

@@ -0,0 +1,15 @@
+
+from typing import Iterable
+from pathlib import Path
+import subprocess
+import json
+
+from .batch import BatchName
+
+
+def get_batch_runs(batch: BatchName, target: Path) -> None:
+    with target.open("wt") as writer:
+        output = subprocess.check_call(
+            ["kivecli", "findbatches", "--debug", "--json", "--name", str(batch)],
+            stdout=writer,
+        )