oh-my-batch 0.4.0__tar.gz → 0.4.1__tar.gz

{oh_my_batch-0.4.0 → oh_my_batch-0.4.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: oh-my-batch
-Version: 0.4.0
+Version: 0.4.1
 Summary: 
 License: GPL
 Author: weihong.xu
@@ -86,9 +86,9 @@ omb combo \
     add_files DATA_FILE tmp/*.data - \
     add_var TEMP 300 400 500 - \
     add_randint RANDOM -n 3 -a 1 -b 1000 - \
+    set_broadcast RANDOM - \
     make_files tmp/tasks/{i}-T-{TEMP}/in.lmp --template tmp/in.lmp.tmp - \
     make_files tmp/tasks/{i}-T-{TEMP}/run.sh --template tmp/run.sh.tmp --mode 755 - \
-    set_broadcast RANDOM - \
     done
 ```
 

{oh_my_batch-0.4.0 → oh_my_batch-0.4.1}/oh_my_batch/combo.py

@@ -4,7 +4,7 @@ import random
 import json
 import os
 
-from .util import expand_globs, mode_translate, ensure_dir
+from .util import expand_globs, mode_translate, ensure_dir, shell_run
 
 class ComboMaker:
 
@@ -214,6 +214,28 @@ class ComboMaker:
         else:
             print(out)
         return self
+    
+    def run_cmd(self, cmd: str):
+        """
+        Run command against each combo
+
+        For example, 
+        
+        run_cmd "cp {DATA_FIEL} ./path/to/workdir/{i}/data.txt"
+
+        will copy each file in DATA_FILE to ./path/to/workdir/{i}/data.txt
+        
+        :param cmd: Command to run, can include format style variables, e.g. {i}, {i:03d}, {TEMP}
+        """
+        combos = self._make_combos()
+        for i, combo in enumerate(combos):
+            _cmd = cmd.format(i=i, **combo)
+            cp = shell_run(_cmd)
+            if cp.returncode != 0:
+                print(cp.stdout.decode('utf-8'))
+                print(cp.stderr.decode('utf-8'))
+                raise RuntimeError(f"Failed to run command: {_cmd}")
+        return self
 
     def done(self):
         """

{oh_my_batch-0.4.0 → oh_my_batch-0.4.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "oh-my-batch"
-version = "0.4.0"
+version = "0.4.1"
 description = ""
 authors = ["weihong.xu <xuweihong.cn@gmail.com>"]
 license = "GPL"

{oh_my_batch-0.4.0 → oh_my_batch-0.4.1}/README.md

@@ -67,9 +67,9 @@ omb combo \
     add_files DATA_FILE tmp/*.data - \
     add_var TEMP 300 400 500 - \
     add_randint RANDOM -n 3 -a 1 -b 1000 - \
+    set_broadcast RANDOM - \
     make_files tmp/tasks/{i}-T-{TEMP}/in.lmp --template tmp/in.lmp.tmp - \
     make_files tmp/tasks/{i}-T-{TEMP}/run.sh --template tmp/run.sh.tmp --mode 755 - \
-    set_broadcast RANDOM - \
     done
 ```