feat: accept same cli args as squeue

kabouzeid · Mar 28, 2023 · 1f1a5ac · 1f1a5ac
1 parent 7e6678d
commit 1f1a5ac
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/job_watcher.rs b/src/job_watcher.rs
@@ -1,3 +1,4 @@
+use std::env;
 use std::path::PathBuf;
 use std::{io::BufRead, process::Command, thread, time::Duration};
 
@@ -40,8 +41,11 @@ impl JobWatcher {
             .map(|s| s.to_owned() + ":" + output_separator)
             .join(",");
 
+        let cli_args = env::args().skip(1).collect::<Vec<_>>();
+
         loop {
             let jobs: Vec<Job> = Command::new("squeue")
+                .args(&cli_args)
                 .arg("-h")
                 .arg("-O")
                 .arg(&output_format)