usecli 0.1.60__tar.gz → 0.1.61__tar.gz

{usecli-0.1.60 → usecli-0.1.61}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: usecli
-Version: 0.1.60
+Version: 0.1.61
 Summary: A powerful Python CLI framework for building beautiful, developer-friendly command-line tools.
 Author: Edward Boswell
 Author-email: Edward Boswell <thememium@gmail.com>

{usecli-0.1.60 → usecli-0.1.61}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "usecli"
-version = "0.1.60"
+version = "0.1.61"
 description = "A powerful Python CLI framework for building beautiful, developer-friendly command-line tools."
 readme = "README.md"
 authors = [{ name = "Edward Boswell", email = "thememium@gmail.com" }]

{usecli-0.1.60 → usecli-0.1.61}/src/usecli/cli/config/colors.py

@@ -39,6 +39,78 @@ _SKIP_DIRS = {
     "pipx",
     "venvs",
 }
+
+_MAX_RGLOB_DEPTH = 6
+
+_WALK_SKIP_ALWAYS: frozenset[str] = frozenset(
+    {
+        ".git",
+        "__pycache__",
+        "node_modules",
+        ".tox",
+        ".nox",
+        ".mypy_cache",
+        ".pytest_cache",
+        ".ruff_cache",
+        ".auto",
+        ".eggs",
+        "*.egg-info",
+    }
+)
+
+_WALK_SKIP_VENV: frozenset[str] = frozenset(
+    {
+        ".venv",
+        "venv",
+        "site-packages",
+        "dist-packages",
+        "__pypackages__",
+        "pipx",
+        "venvs",
+    }
+)
+
+
+def _walk_for_filename(
+    directory: Path,
+    filename: str,
+    depth: int,
+    max_depth: int,
+    skip_dirs: frozenset[str],
+    results: list[Path],
+) -> None:
+    if depth > max_depth:
+        return
+    try:
+        entries = list(directory.iterdir())
+    except (PermissionError, OSError):
+        return
+    for entry in entries:
+        try:
+            if entry.is_file() and entry.name == filename:
+                results.append(entry)
+            elif entry.is_dir() and entry.name not in skip_dirs:
+                _walk_for_filename(
+                    entry, filename, depth + 1, max_depth, skip_dirs, results
+                )
+        except (PermissionError, OSError):
+            continue
+
+
+def _rglob_limited(
+    root_dir: Path,
+    filename: str,
+    *,
+    skip_venv: bool = True,
+    max_depth: int = _MAX_RGLOB_DEPTH,
+) -> list[Path]:
+    """Depth-bounded recursive filename search that prunes dirs during the walk."""
+    skip_dirs = _WALK_SKIP_ALWAYS | _WALK_SKIP_VENV if skip_venv else _WALK_SKIP_ALWAYS
+    results: list[Path] = []
+    _walk_for_filename(root_dir, filename, 0, max_depth, skip_dirs, results)
+    return results
+
+
 DEFAULT_THEME_COLORS: dict[str, str] = {
     "primary": "#60D7FF",
     "secondary": "#5EFF87",
@@ -68,13 +140,7 @@ def _find_usecli_config_path(
     if not root_dir.exists() or not root_dir.is_dir():
         return None
 
-    candidates = [path for path in root_dir.rglob(USECLI_CONFIG_TOML)]
-    if skip_venv:
-        candidates = [
-            path
-            for path in candidates
-            if not any(part in _SKIP_DIRS for part in path.parts)
-        ]
+    candidates = _rglob_limited(root_dir, USECLI_CONFIG_TOML, skip_venv=skip_venv)
     if not candidates:
         return None
 
@@ -109,17 +175,28 @@ def _get_command_name() -> str | None:
     return command if command else None
 
 
+_distributions_cache: list[Any] | None = None
+
+
+def _get_distributions() -> list[Any]:
+    global _distributions_cache
+    if _distributions_cache is not None:
+        return _distributions_cache
+    try:
+        import importlib.metadata
+
+        _distributions_cache = list(importlib.metadata.distributions())
+    except Exception:
+        _distributions_cache = []
+    return _distributions_cache
+
+
 def _get_console_script_aliases(command_name: str | None) -> set[str]:
     """Get all aliases for a console script from package metadata."""
     if not command_name:
         return set()
     aliases: set[str] = {command_name}
-    try:
-        import importlib.metadata
-
-        distributions = importlib.metadata.distributions()
-    except Exception:
-        return aliases
+    distributions = _get_distributions()
     for dist in distributions:
         try:
             entry_points = dist.entry_points
@@ -163,13 +240,7 @@ def _find_usecli_config_path_for_command(
     if not root_dir.exists() or not root_dir.is_dir():
         return None
 
-    candidates = [path for path in root_dir.rglob(USECLI_CONFIG_TOML)]
-    if skip_venv:
-        candidates = [
-            path
-            for path in candidates
-            if not any(part in _SKIP_DIRS for part in path.parts)
-        ]
+    candidates = _rglob_limited(root_dir, USECLI_CONFIG_TOML, skip_venv=skip_venv)
     if not candidates:
         return None
 
@@ -217,9 +288,7 @@ def _find_usecli_config_in_package() -> Path | None:
         package_root = Path(location)
         if not package_root.exists() or not package_root.is_dir():
             continue
-        candidates = [
-            path for path in package_root.rglob(USECLI_CONFIG_TOML) if path.exists()
-        ]
+        candidates = _rglob_limited(package_root, USECLI_CONFIG_TOML, skip_venv=False)
         if candidates:
             candidates.sort(key=lambda path: (len(path.parts), str(path)))
             return candidates[0]
@@ -238,9 +307,7 @@ def _find_usecli_config_in_named_package(package_name: str) -> Path | None:
         package_root = Path(location)
         if not package_root.exists() or not package_root.is_dir():
             continue
-        candidates = [
-            path for path in package_root.rglob(USECLI_CONFIG_TOML) if path.exists()
-        ]
+        candidates = _rglob_limited(package_root, USECLI_CONFIG_TOML, skip_venv=False)
         if candidates:
             candidates.sort(key=lambda path: (len(path.parts), str(path)))
             return candidates[0]
@@ -248,15 +315,10 @@ def _find_usecli_config_in_named_package(package_name: str) -> Path | None:
 
 
 def _find_usecli_config_for_console_script() -> Path | None:
-    import importlib.metadata
-
     command_name = os.path.basename(sys.argv[0]) if sys.argv else ""
     if not command_name:
         return None
-    try:
-        distributions = importlib.metadata.distributions()
-    except Exception:
-        return None
+    distributions = _get_distributions()
     for dist in distributions:
         try:
             entry_points = dist.entry_points
@@ -334,10 +396,8 @@ def _find_project_root(start_dir: Path | None = None) -> Path | None:
         current = parent
 
     search_root = git_root or start_dir.resolve()
-    config_match = _find_usecli_config_path(search_root, start_dir, skip_venv=True)
-    if config_match:
-        return config_match.parent
 
+    # Try fast lookups before expensive rglob (perf: global tools).
     console_match = _find_usecli_config_for_console_script()
     if console_match:
         return console_match.parent
@@ -346,6 +406,10 @@ def _find_project_root(start_dir: Path | None = None) -> Path | None:
     if package_match:
         return package_match.parent
 
+    config_match = _find_usecli_config_path(search_root, start_dir, skip_venv=True)
+    if config_match:
+        return config_match.parent
+
     return git_root
 
 
@@ -357,18 +421,19 @@ def _load_usecli_config(
 
     config_path = project_root / USECLI_CONFIG_TOML
     if not config_path.exists():
-        config_path = _find_usecli_config_path_for_command(
-            project_root,
-            project_root,
-            skip_venv=True,
-        )
-    if not config_path or not config_path.exists():
+        # Try fast lookups before expensive rglob (perf: global tools).
         console_match = _find_usecli_config_for_console_script()
         if console_match:
             return _load_usecli_config_file(console_match), console_match
         package_match = _find_usecli_config_in_package()
         if package_match:
             config_path = package_match
+    if not config_path or not config_path.exists():
+        config_path = _find_usecli_config_path_for_command(
+            project_root,
+            project_root,
+            skip_venv=True,
+        )
     if not config_path or not config_path.exists():
         return {}, None
 

{usecli-0.1.60 → usecli-0.1.61}/src/usecli/shared/config/manager.py

@@ -18,6 +18,77 @@ if sys.version_info >= (3, 11):
 else:
     import tomli as tomllib
 
+# Depth cap for rglob – prevents scanning massive trees like ~/ghq.
+_MAX_RGLOB_DEPTH = 6
+
+_WALK_SKIP_ALWAYS: frozenset[str] = frozenset(
+    {
+        ".git",
+        "__pycache__",
+        "node_modules",
+        ".tox",
+        ".nox",
+        ".mypy_cache",
+        ".pytest_cache",
+        ".ruff_cache",
+        ".auto",
+        ".eggs",
+        "*.egg-info",
+    }
+)
+
+_WALK_SKIP_VENV: frozenset[str] = frozenset(
+    {
+        ".venv",
+        "venv",
+        "site-packages",
+        "dist-packages",
+        "__pypackages__",
+        "pipx",
+        "venvs",
+    }
+)
+
+
+def _walk_for_filename(
+    directory: Path,
+    filename: str,
+    depth: int,
+    max_depth: int,
+    skip_dirs: frozenset[str],
+    results: list[Path],
+) -> None:
+    if depth > max_depth:
+        return
+    try:
+        entries = list(directory.iterdir())
+    except (PermissionError, OSError):
+        return
+    for entry in entries:
+        try:
+            if entry.is_file() and entry.name == filename:
+                results.append(entry)
+            elif entry.is_dir() and entry.name not in skip_dirs:
+                _walk_for_filename(
+                    entry, filename, depth + 1, max_depth, skip_dirs, results
+                )
+        except (PermissionError, OSError):
+            continue
+
+
+def _rglob_limited(
+    root_dir: Path,
+    filename: str,
+    *,
+    skip_venv: bool = True,
+    max_depth: int = _MAX_RGLOB_DEPTH,
+) -> list[Path]:
+    """Depth-bounded recursive filename search that prunes dirs during the walk."""
+    skip_dirs = _WALK_SKIP_ALWAYS | _WALK_SKIP_VENV if skip_venv else _WALK_SKIP_ALWAYS
+    results: list[Path] = []
+    _walk_for_filename(root_dir, filename, 0, max_depth, skip_dirs, results)
+    return results
+
 
 def _get_importlib_metadata():
     import importlib.metadata
@@ -25,6 +96,26 @@ def _get_importlib_metadata():
     return importlib.metadata
 
 
+_distributions_cache: list[Any] | None = None
+
+
+def _get_distributions() -> list[Any]:
+    global _distributions_cache
+    if _distributions_cache is not None:
+        return _distributions_cache
+    try:
+        metadata = _get_importlib_metadata()
+        _distributions_cache = list(metadata.distributions())
+    except Exception:
+        _distributions_cache = []
+    return _distributions_cache
+
+
+def _reset_distributions_cache() -> None:
+    global _distributions_cache
+    _distributions_cache = None
+
+
 def _deep_merge(base: dict[str, Any], override: dict[str, Any]) -> dict[str, Any]:
     result = base.copy()
     for key, value in override.items():
@@ -220,6 +311,20 @@ class ConfigManager:
                 break
             current = parent
 
+        # Try fast lookups before expensive rglob (perf: global tools).
+        console_match = cls._find_usecli_config_for_console_script()
+        if console_match:
+            return console_match
+
+        if cls._is_within_usecli_package(start_dir):
+            package_match = cls._find_usecli_config_in_package()
+            if package_match:
+                return package_match
+
+            sys_match = cls._find_usecli_config_on_sys_path()
+            if sys_match:
+                return sys_match
+
         search_root = find_project_root(start_dir) or start_dir.resolve()
         is_framework = command_name == "usecli" if command_name else True
         recursive_match = cls._find_usecli_config_in_tree(
@@ -228,18 +333,7 @@ class ConfigManager:
         if recursive_match:
             return recursive_match
 
-        console_match = cls._find_usecli_config_for_console_script()
-        if console_match:
-            return console_match
-
-        if not cls._is_within_usecli_package(start_dir):
-            return None
-
-        package_match = cls._find_usecli_config_in_package()
-        if package_match:
-            return package_match
-
-        return cls._find_usecli_config_on_sys_path()
+        return None
 
     @staticmethod
     def _find_usecli_config_in_tree(
@@ -248,13 +342,7 @@ class ConfigManager:
         if not root_dir.exists() or not root_dir.is_dir():
             return None
 
-        candidates = [path for path in root_dir.rglob(USECLI_CONFIG_TOML)]
-        if skip_venv:
-            candidates = [
-                path
-                for path in candidates
-                if not any(part in ConfigManager._SKIP_DIRS for part in path.parts)
-            ]
+        candidates = _rglob_limited(root_dir, USECLI_CONFIG_TOML, skip_venv=skip_venv)
         command_name = ConfigManager._get_command_name()
         if command_name:
             candidates = [
@@ -313,9 +401,9 @@ class ConfigManager:
             package_root = Path(location)
             if not package_root.exists() or not package_root.is_dir():
                 continue
-            candidates = [
-                path for path in package_root.rglob(USECLI_CONFIG_TOML) if path.exists()
-            ]
+            candidates = _rglob_limited(
+                package_root, USECLI_CONFIG_TOML, skip_venv=False
+            )
             if command_name:
                 candidates = [
                     path
@@ -354,9 +442,9 @@ class ConfigManager:
             package_root = Path(location)
             if not package_root.exists() or not package_root.is_dir():
                 continue
-            candidates = [
-                path for path in package_root.rglob(USECLI_CONFIG_TOML) if path.exists()
-            ]
+            candidates = _rglob_limited(
+                package_root, USECLI_CONFIG_TOML, skip_venv=False
+            )
             if command_name:
                 candidates = [
                     path
@@ -373,11 +461,7 @@ class ConfigManager:
         command_name = os.path.basename(sys.argv[0]) if sys.argv else ""
         if not command_name:
             return None
-        try:
-            metadata = _get_importlib_metadata()
-            distributions = metadata.distributions()
-        except Exception:
-            return None
+        distributions = _get_distributions()
         for dist in distributions:
             try:
                 entry_points = dist.entry_points
@@ -478,11 +562,7 @@ class ConfigManager:
         if not command_name:
             return set()
         aliases: set[str] = {command_name}
-        try:
-            metadata = _get_importlib_metadata()
-            distributions = metadata.distributions()
-        except Exception:
-            return aliases
+        distributions = _get_distributions()
         for dist in distributions:
             try:
                 entry_points = dist.entry_points
@@ -571,11 +651,7 @@ class ConfigManager:
         """Search a source tree for a ``usecli.config.toml`` that matches."""
         if not source_root.exists() or not source_root.is_dir():
             return None
-        candidates = [
-            p
-            for p in source_root.rglob(USECLI_CONFIG_TOML)
-            if not any(part in ConfigManager._SKIP_DIRS for part in p.parts)
-        ]
+        candidates = _rglob_limited(source_root, USECLI_CONFIG_TOML)
         if command_name:
             candidates = [
                 p
@@ -685,11 +761,7 @@ class ConfigManager:
         project_root = find_project_root(start_dir)
         if project_root is None:
             return None
-        candidates = [
-            p
-            for p in project_root.rglob(USECLI_CONFIG_TOML)
-            if not any(part in self._SKIP_DIRS for part in p.parts)
-        ]
+        candidates = _rglob_limited(project_root, USECLI_CONFIG_TOML)
         if not candidates:
             return None
         candidates.sort(key=lambda p: (len(p.parts), str(p)))
@@ -821,6 +893,17 @@ def find_project_root(start_dir: Path | None = None) -> Path | None:
         current = parent
 
     search_root = git_root or start_dir.resolve()
+
+    # Try fast lookups before expensive rglob (perf: global tools).
+    console_match = ConfigManager._find_usecli_config_for_console_script()
+    if console_match:
+        return console_match.parent
+
+    if ConfigManager._is_within_usecli_package(start_dir):
+        package_match = ConfigManager._find_usecli_config_in_package()
+        if package_match:
+            return package_match.parent
+
     config_match = ConfigManager._find_usecli_config_in_tree(
         search_root,
         start_dir,
@@ -829,11 +912,6 @@ def find_project_root(start_dir: Path | None = None) -> Path | None:
     if config_match:
         return config_match.parent
 
-    if ConfigManager._is_within_usecli_package(start_dir):
-        package_match = ConfigManager._find_usecli_config_in_package()
-        if package_match:
-            return package_match.parent
-
     return git_root