Gstat: adding -d output for efficient piping (#55)

tdegeus · Oct 14, 2022 · 85c94be · 85c94be
1 parent cce9839
commit 85c94be
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/GooseSLURM/cli/Gstat.py b/GooseSLURM/cli/Gstat.py
@@ -127,6 +127,10 @@
     --debug=<FILE>
         Debug: read ``squeue -o "%all"`` from file.
 
+    -d, --print-dependency
+        Print the selected jobs as ``-d <jobid> -d <jobid> ...``.
+        Use to for example ``Gsub *slurm `Gstat -d -U --partition "serial"```.
+
     -h, --help
         Show help.
 
@@ -191,6 +195,7 @@ def print_help(self):
         parser.add_argument("--sep", type=str, default=" ")
         parser.add_argument("--long", action="store_true")
         parser.add_argument("--debug", type=str)
+        parser.add_argument("-d", "--print-dependency", action="store_true")
         parser.add_argument("--version", action="version", version=version)
         parser.add_argument("jobs", type=int, nargs="*")
 
@@ -500,7 +505,9 @@ def print(self):
         Print.
         """
 
-        if not self.args["summary"]:
+        if self.args["print_dependency"]:
+            print("-d " + " -d ".join([str(line["JOBID"]) for line in self.lines]))
+        elif not self.args["summary"]:
             self.print_all()
         else:
             self.print_summary()