oh-my-batch 0.5.7__tar.gz → 0.6.1__tar.gz

{oh_my_batch-0.5.7 → oh_my_batch-0.6.1}/PKG-INFO

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

{oh_my_batch-0.5.7 → oh_my_batch-0.6.1}/oh_my_batch/batch.py

@@ -1,3 +1,4 @@
+from typing import Optional
 import shlex
 import os
 
@@ -25,6 +26,24 @@ class BatchMaker:
         self._work_dirs.extend(paths)
         return self
 
+    def filter(self, expr: str):
+        """
+        Filter working directories with a Python expression
+
+        :param expr: Python expression, the variable `{workdir}, {work_dir} or {w}` is the directory path,
+                     `{index} or {i}` is the index of the directory
+
+        Example: if expr is 'os.path.exits("{workdir}/input.json")',
+        then only the directories containing 'input.json' will be kept.
+        """
+        filtered = []
+        for i, workdir in enumerate(self._work_dirs):
+            expr_eval = expr.format(workdir=workdir, work_dir=workdir, w=workdir, index=i, i=i)
+            if eval(expr_eval):
+                filtered.append(workdir)
+        self._work_dirs = filtered
+        return self
+
     def add_header_files(self, *file: str, encoding='utf-8'):
         """
         Add script header from files
@@ -130,4 +149,4 @@ def load_files(*file, encoding='utf-8', raise_invalid=False):
     for file in expand_globs(file, raise_invalid=raise_invalid):
         with open(file, 'r', encoding=encoding) as f:
             result.append(f.read())
-    return result
+    return result

{oh_my_batch-0.5.7 → oh_my_batch-0.6.1}/pyproject.toml

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