oh-my-batch 0.4.2__tar.gz → 0.4.4__tar.gz

{oh_my_batch-0.4.2 → oh_my_batch-0.4.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: oh-my-batch
-Version: 0.4.2
+Version: 0.4.4
 Summary: 
 License: GPL
 Author: weihong.xu

{oh_my_batch-0.4.2 → oh_my_batch-0.4.4}/oh_my_batch/combo.py

@@ -33,7 +33,7 @@ class ComboMaker:
         self.add_var(key, *args)
         return self
 
-    def add_randint(self, key: str, n: int, a: int, b: int, seed=None):
+    def add_randint(self, key: str, n: int, a: int, b: int, uniq=False, seed=None):
         """
         Add a variable with random integer values
 
@@ -41,11 +41,17 @@ class ComboMaker:
         :param n: Number of values
         :param a: Lower bound
         :param b: Upper bound
+        :param uniq: If True, values are unique, default is False
         :param seed: Seed for random number generator
         """
         if seed is not None:
             random.seed(seed)
-        args = [random.randint(a, b) for _ in range(n)]
+        if uniq:
+            if b - a + 1 < n:
+                raise ValueError("Not enough unique values")
+            args = random.sample(range(a, b + 1), n)
+        else:
+            args = [random.randint(a, b) for _ in range(n)]
         self.add_var(key, *args)
         return self
 
@@ -221,7 +227,7 @@ class ComboMaker:
 
         For example,
 
-        run_cmd "cp {DATA_FIEL} ./path/to/workdir/{i}/data.txt"
+        run_cmd "cp {DATA_FILE} ./path/to/workdir/{i}/data.txt"
 
         will copy each file in DATA_FILE to ./path/to/workdir/{i}/data.txt
 
@@ -237,6 +243,20 @@ class ComboMaker:
                 raise RuntimeError(f"Failed to run command: {_cmd}")
         return self
 
+    def show_combos(self):
+        combos = self._make_combos()
+        if not combos:
+            print("No combos")
+        keys = combos[0].keys()
+
+        for i, combo in enumerate(combos):
+            print(f"> Combo {i}:")
+            for k in keys:
+                v = str(combo[k])
+                if '\n' in v:
+                    v = f"\n{v}"
+                print(f"@{k}: {v}")
+
     def done(self):
         """
         End of command chain

{oh_my_batch-0.4.2 → oh_my_batch-0.4.4}/pyproject.toml

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