oh-my-batch 0.4.2__py3-none-any.whl → 0.4.4__py3-none-any.whl

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.dist-info → oh_my_batch-0.4.4.dist-info}/METADATA

@@ -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.dist-info → oh_my_batch-0.4.4.dist-info}/RECORD

@@ -4,12 +4,12 @@ oh_my_batch/assets/__init__.py,sha256=Exub46UbQaz2V2eXpQeiVfnThQpXaNeuyjlGY6gBSZ
 oh_my_batch/assets/functions.sh,sha256=LaiavZBu84D7C1r-dWhf91vPTuHXCMV1DQZUIPVQnjE,1001
 oh_my_batch/batch.py,sha256=6qnaXEVyA493heGzzbCrdZXCcnYk8zgl7WP0rmo7KlU,3690
 oh_my_batch/cli.py,sha256=Jyz8q2pUYke3mfJS6F_G9S9hApddgXxQw1BsN6Kfkjc,553
-oh_my_batch/combo.py,sha256=XK1t3rOQozf9oBYPr_CER7jhCMYpqJaZo0ke4ctK7t0,9307
+oh_my_batch/combo.py,sha256=A5flYsjugcelj105P6Nq4M3kNGtdqh3c7QaSIL3hUcg,9954
 oh_my_batch/job.py,sha256=Uk0E-WxnexBu9wuUV3uc1aAwVF_5rWdNdLEOB8Y9B-U,6252
 oh_my_batch/misc.py,sha256=G_iOovRCrShBJJCc82QLN0CvMqW4adOefEoY1GedEiw,452
 oh_my_batch/util.py,sha256=5ve2QcviuF0UHFLrsXmjMTj0ogXJ4g05q1y-yWCFuOk,2409
-oh_my_batch-0.4.2.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-oh_my_batch-0.4.2.dist-info/METADATA,sha256=VrGmQGhqGB3RGqy8E5sxS4o288Ddd-uNqjTKnZmtAts,5509
-oh_my_batch-0.4.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-oh_my_batch-0.4.2.dist-info/entry_points.txt,sha256=ZY2GutSoNjjSyJ4qO2pTeseKUFgoTYdvmgkuZZkwi68,77
-oh_my_batch-0.4.2.dist-info/RECORD,,
+oh_my_batch-0.4.4.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+oh_my_batch-0.4.4.dist-info/METADATA,sha256=D9pKRVr4csOtcs05l-hvNku0CL8UQi57ZBScaAMc06M,5509
+oh_my_batch-0.4.4.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+oh_my_batch-0.4.4.dist-info/entry_points.txt,sha256=ZY2GutSoNjjSyJ4qO2pTeseKUFgoTYdvmgkuZZkwi68,77
+oh_my_batch-0.4.4.dist-info/RECORD,,