Add more ignoredirs

hoxbro · Jun 6, 2024 · 4b17239 · 4b17239
1 parent 7fb7e4e
commit 4b17239
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/clean_notebook/clean.py b/src/clean_notebook/clean.py
@@ -7,6 +7,8 @@
 
 __all__ = ("clean_notebook", "clean_single_notebook")
 
+IGNOREDIRS = [".venv", ".pixi", "site-packages", ".ipynb_checkpoints"]
+
 
 def clean_notebook(
     paths: list[str | Path],
@@ -86,9 +88,12 @@ def _get_files(paths: list[str | Path]) -> Iterator[Path]:
         if path.is_file() and path.suffix == ".ipynb":
             yield path
         if path.is_dir():
+            if path.name in IGNOREDIRS:
+                continue
             for file in path.rglob("*.ipynb"):
-                if file.parent.name != ".ipynb_checkpoints":
-                    yield file
+                if any(ps in IGNOREDIRS for ps in file.parts):
+                    continue
+                yield file
 
 
 def _ignore(cell: dict[str, Any], ignore: list[str] | None) -> dict[str, Any]: