Merge pull request #111 from tybug/perf

Improve `shed` speed in the presence of large dot directories

Author: Zac-HD

CHANGELOG.md

@@ -1,6 +1,9 @@
 # Changelog
 *`shed` uses [calendar versioning](https://calver.org/), with a year.month.patch scheme.*
 
+#### 2025.6.1 - 2025-06-05
+- Improve performance in large repositories, by skipping directories that start with `.` during an internal check.
+
 #### 2024.10.1 - 2024-04-27
 - Disable `PIE790` fix due to [ruff issue #10538](https://github.com/astral-sh/ruff/issues/10538)
 - [Python 3.8 has reached end-of-life](https://devguide.python.org/versions/),

README.md

@@ -59,7 +59,7 @@ adding the following to your `.pre-commit-config.yaml`:
 minimum_pre_commit_version: '2.9.0'
 repos:
 - repo: https://github.com/Zac-HD/shed
-  rev: 2024.10.1
+  rev: 2025.6.1
   hooks:
     - id: shed
       # args: [--refactor, --py311-plus]

src/shed/__init__.py

@@ -19,7 +19,7 @@
 from black.mode import TargetVersion
 from black.parsing import lib2to3_parse
 
-__version__ = "2024.10.1"
+__version__ = "2025.6.1"
 __all__ = ["shed", "docshed"]
 
 # Conditionally imported in refactor mode to reduce startup latency in the common case

src/shed/_cli.py

@@ -47,7 +47,26 @@ def _guess_first_party_modules(cwd: Optional[str] = None) -> FrozenSet[str]:
         base = _get_git_repo_root(cwd)
     except (subprocess.SubprocessError, FileNotFoundError):
         return frozenset()
-    provides = {init.parent.name for init in Path(base).glob("**/src/*/__init__.py")}
+
+    def _walk_path(path: Path) -> set[str]:
+        provides = set()
+        try:
+            for p in path.iterdir():
+                if p.name.startswith(".") or not p.is_dir():
+                    continue
+                if p.name == "src":
+                    provides |= {init.parent.name for init in p.glob("*/__init__.py")}
+                    # in case of nested src/ directories, like
+                    #   src/tools/src/helper/__init__.py
+                    provides |= _walk_path(p)
+                else:
+                    provides |= _walk_path(p)
+        except Exception:  # pragma: no cover
+            pass
+
+        return provides
+
+    provides = _walk_path(Path(base))
     return frozenset(
         p
         for p in {Path(base).name} | provides

src/shed/_is_python_file.py

@@ -72,7 +72,7 @@ def _detect_encoding(
 
 
 def _inner_detect_encoding(
-    readline: Callable[[], bytes | bytearray]
+    readline: Callable[[], bytes | bytearray],
 ) -> str:  # pragma: no cover
     """Return file encoding."""
     try: