Add more ignoredirs (#39)

hoxbro · Jun 6, 2024 · 08dea23 · 08dea23
1 parent 9cf8ef2
commit 08dea23
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -11,7 +11,7 @@ repos:
       - id: check-json
       - id: check-executables-have-shebangs
   - repo: https://github.com/astral-sh/ruff-pre-commit
-    rev: v0.4.5
+    rev: v0.4.8
     hooks:
       - id: ruff
       - id: ruff-format
@@ -26,7 +26,7 @@ repos:
         additional_dependencies:
           - tomli
   - repo: https://github.com/hoxbro/prettier-pre-commit
-    rev: v3.2.5
+    rev: v3.3.1
     hooks:
       - id: prettier
         types_or:

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]: