Skip to content

Add Python type hints to ai_bom.scanner module #66

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
biliyajanardhan wants to merge 1 commit into
base: main
Choose a base branch
from
Open

Add Python type hints to ai_bom.scanner module #66

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 5 additions & 4 deletions src/ai_bom/scanners/base.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@
_scanner_registry: list[type[BaseScanner]] = []

# Cached .ai-bomignore spec (module-level singleton)
_ignore_spec: Any = None
_ignore_spec: Any | None = None
_ignore_spec_loaded: bool = False


Expand Down Expand Up @@ -54,7 +54,7 @@ def _load_ignore_spec(root: Path) -> Any:
return None

try:
text = ignore_file.read_text(encoding="utf-8")
text: str = ignore_file.read_text(encoding="utf-8")
_ignore_spec = pathspec.PathSpec.from_lines("gitwildmatch", text.splitlines())
logger.debug("Loaded .ai-bomignore with %d patterns", len(_ignore_spec.patterns))
except Exception as exc:
Expand Down Expand Up @@ -215,7 +215,7 @@ def iter_files(
root = root.resolve()

# Load .ai-bomignore spec (cached after first call)
ignore_spec = _load_ignore_spec(root if root.is_dir() else root.parent)
ignore_spec: Any | None = _load_ignore_spec(root if root.is_dir() else root.parent)

# Track real paths to detect symlink cycles
seen_real_paths: set[Path] = set()
Expand Down Expand Up @@ -267,7 +267,7 @@ def iter_files(
logger.warning("Cannot read file %s: %s", root, e)
return

matches = False
matches:bool = False
if extensions is not None and root.suffix.lower() in extensions:
matches = True
if filenames is not None and (root.name in filenames or root.name.lower() in filenames):
Expand All @@ -290,6 +290,7 @@ def iter_files(
# Walk the directory tree
try:
for dirpath, dirnames, filenames_list in os.walk(root, topdown=True, followlinks=True):

# Resolve current directory to detect symlink cycles
try:
current_real = Path(os.path.realpath(dirpath))
Expand Down