access print-groups in fs,wt,lt,etc

Author: chapmanjacobd

xklb/play_actions.py

@@ -138,6 +138,7 @@ def parse_args(action, default_chromecast=None) -> argparse.Namespace:
     parser.add_argument("--solo", action="store_true")
 
     parser.add_argument("--sort-groups-by", "--sort-groups")
+    parser.add_argument("--print-groups", "--groups", "-g", action="store_true", help="Print groups")
     parser.add_argument("--depth", "-D", default=0, type=int, help="Depth of folders")
     parser.add_argument("--lower", type=int, help="Number of files per folder lower limit")
     parser.add_argument("--upper", type=int, help="Number of files per folder upper limit")

xklb/scripts/cluster_sort.py

@@ -216,9 +216,21 @@ def cluster_dicts(args, media):
         sorted_paths = iterables.flatten(
             s for d in groups for s in d["grouped_paths"] if not bool(media_keyed[s].get("time_deleted"))
         )
+
+    if args.print_groups:
+        print_groups(groups)
+
     media = [media_keyed[p] for p in sorted_paths]
     return media
 
+def print_groups(groups):
+    for group in groups:
+        group["grouped_paths"] = [s.rstrip("\n") for s in group["grouped_paths"]]
+
+    print(json.dumps(groups, indent=4))
+    raise SystemExit(0)
+
+
 
 def cluster_images(paths, n_clusters=None):
     paths = [s.rstrip("\n") for s in paths]
@@ -340,10 +352,7 @@ def cluster_sort() -> None:
         groups = [d for d in groups if len(d["grouped_paths"]) == 1]
 
     if args.print_groups:
-        for group in groups:
-            group["grouped_paths"] = [s.rstrip("\n") for s in group["grouped_paths"]]
-
-        print(json.dumps(groups))
+        print_groups(groups)
     elif args.move_groups:
         min_len = len(str(len(groups) + 1))