feat(zero-3rdparty): only_files iteration

EspenAlbert · Sep 27, 2023 · 93bff16 · 93bff16
1 parent 2a70962
commit 93bff16
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/zero_3rdparty/src/zero_3rdparty/file_utils.py b/zero_3rdparty/src/zero_3rdparty/file_utils.py
@@ -129,7 +129,9 @@ def iter_paths(base_dir: Path, *globs: str, rglob=True) -> Iterable[Path]:
 
 
 def iter_paths_and_relative(
-    base_dir: Path, *globs: str, rglob=True
+    base_dir: Path, *globs: str, rglob=True, only_files: bool = False
 ) -> Iterable[tuple[Path, str]]:
     for path in iter_paths(base_dir, *globs, rglob=rglob):
+        if only_files and not path.is_file():
+            continue
         yield path, str(path.relative_to(base_dir))