thailint 0.8.0__py3-none-any.whl → 0.10.0__py3-none-any.whl

src/linter_config/loader.py

@@ -16,10 +16,10 @@ Overview: Loads linter configuration from .thailint.yaml or .thailint.json files
 Dependencies: PyYAML for YAML parsing with safe_load(), json (stdlib) for JSON parsing,
     pathlib for file path handling
 
-Exports: LinterConfigLoader class
+Exports: load_config function, get_defaults function, LinterConfigLoader class (compat)
 
-Interfaces: load(config_path: Path) -> dict[str, Any] for loading config files,
-    defaults property -> dict[str, Any] for default configuration structure
+Interfaces: load_config(config_path: Path) -> dict[str, Any] for loading config files,
+    get_defaults() -> dict[str, Any] for default configuration structure
 
 Implementation: Extension-based format detection (.yaml/.yml vs .json), yaml.safe_load()
     for security, empty dict handling for null YAML, ValueError for unsupported formats
@@ -31,13 +31,51 @@ from typing import Any
 from src.core.config_parser import parse_config_file
 
 
+def get_defaults() -> dict[str, Any]:
+    """Get default configuration.
+
+    Returns:
+        Default configuration with empty rules and ignore lists.
+    """
+    return {
+        "rules": {},
+        "ignore": [],
+    }
+
+
+def load_config(config_path: Path) -> dict[str, Any]:
+    """Load configuration from file.
+
+    Args:
+        config_path: Path to YAML or JSON config file.
+
+    Returns:
+        Configuration dictionary.
+
+    Raises:
+        ConfigParseError: If file format is unsupported or parsing fails.
+    """
+    if not config_path.exists():
+        return get_defaults()
+
+    return parse_config_file(config_path)
+
+
+# Legacy class wrapper for backward compatibility
 class LinterConfigLoader:
     """Load linter configuration from YAML or JSON files.
 
     Supports loading from .thailint.yaml, .thailint.json, or custom paths.
     Provides sensible defaults when config files don't exist.
+
+    Note: This class is a thin wrapper around module-level functions
+    for backward compatibility.
     """
 
+    def __init__(self) -> None:
+        """Initialize the loader."""
+        pass  # No state needed
+
     def load(self, config_path: Path) -> dict[str, Any]:
         """Load configuration from file.
 
@@ -50,10 +88,7 @@ class LinterConfigLoader:
         Raises:
             ConfigParseError: If file format is unsupported or parsing fails.
         """
-        if not config_path.exists():
-            return self.defaults
-
-        return parse_config_file(config_path)
+        return load_config(config_path)
 
     @property
     def defaults(self) -> dict[str, Any]:
@@ -62,7 +97,4 @@ class LinterConfigLoader:
         Returns:
             Default configuration with empty rules and ignore lists.
         """
-        return {
-            "rules": {},
-            "ignore": [],
-        }
+        return get_defaults()

src/linters/collection_pipeline/__init__.py

@@ -0,0 +1,90 @@
+"""
+Purpose: Collection pipeline linter package initialization
+
+Scope: Exports for collection-pipeline linter module
+
+Overview: Initializes the collection-pipeline linter package and exposes the main rule class
+    for external use. Exports CollectionPipelineRule as the primary interface for the linter,
+    allowing the orchestrator to discover and instantiate the rule. Also exports configuration
+    and detector classes for advanced use cases. Provides a convenience lint() function for
+    direct usage without orchestrator setup. This module serves as the entry point for
+    the collection-pipeline linter functionality within the thai-lint framework.
+
+Dependencies: CollectionPipelineRule, CollectionPipelineConfig, PipelinePatternDetector
+
+Exports: CollectionPipelineRule (primary), CollectionPipelineConfig, PipelinePatternDetector, lint
+
+Interfaces: Standard Python package initialization with __all__ for explicit exports
+
+Implementation: Simple re-export pattern for package interface, convenience lint function
+"""
+
+from pathlib import Path
+from typing import Any
+
+from .config import DEFAULT_MIN_CONTINUES, CollectionPipelineConfig
+from .detector import PatternMatch, PipelinePatternDetector
+from .linter import CollectionPipelineRule
+
+__all__ = [
+    "CollectionPipelineRule",
+    "CollectionPipelineConfig",
+    "PipelinePatternDetector",
+    "PatternMatch",
+    "lint",
+]
+
+
+def lint(
+    path: Path | str,
+    config: dict[str, Any] | None = None,
+    min_continues: int = DEFAULT_MIN_CONTINUES,
+) -> list:
+    """Lint a file or directory for collection pipeline violations.
+
+    Args:
+        path: Path to file or directory to lint
+        config: Configuration dict (optional, uses defaults if not provided)
+        min_continues: Minimum if/continue patterns to flag (default: 1)
+
+    Returns:
+        List of violations found
+
+    Example:
+        >>> from src.linters.collection_pipeline import lint
+        >>> violations = lint('src/my_module.py', min_continues=2)
+        >>> for v in violations:
+        ...     print(f"{v.file_path}:{v.line} - {v.message}")
+    """
+    path_obj = Path(path) if isinstance(path, str) else path
+    project_root = path_obj if path_obj.is_dir() else path_obj.parent
+
+    orchestrator = _setup_pipeline_orchestrator(project_root, config, min_continues)
+    violations = _execute_pipeline_lint(orchestrator, path_obj)
+
+    return [v for v in violations if "collection-pipeline" in v.rule_id]
+
+
+def _setup_pipeline_orchestrator(
+    project_root: Path, config: dict[str, Any] | None, min_continues: int
+) -> Any:
+    """Set up orchestrator with collection-pipeline config."""
+    from src.orchestrator.core import Orchestrator
+
+    orchestrator = Orchestrator(project_root=project_root)
+
+    if config:
+        orchestrator.config["collection-pipeline"] = config
+    else:
+        orchestrator.config["collection-pipeline"] = {"min_continues": min_continues}
+
+    return orchestrator
+
+
+def _execute_pipeline_lint(orchestrator: Any, path_obj: Path) -> list:
+    """Execute linting on file or directory."""
+    if path_obj.is_file():
+        return orchestrator.lint_file(path_obj)
+    if path_obj.is_dir():
+        return orchestrator.lint_directory(path_obj)
+    return []

src/linters/collection_pipeline/config.py

@@ -0,0 +1,63 @@
+"""
+Purpose: Configuration dataclass for collection-pipeline linter
+
+Scope: Define configurable options for embedded filtering pattern detection
+
+Overview: Provides CollectionPipelineConfig for customizing linter behavior including
+    minimum number of continue patterns to flag, enable/disable toggle, and ignore
+    patterns. Integrates with the orchestrator's configuration system to allow users
+    to customize collection-pipeline detection via .thailint.yaml configuration files.
+    Follows the same configuration pattern as other thai-lint linters.
+
+Dependencies: dataclasses, typing
+
+Exports: CollectionPipelineConfig dataclass, DEFAULT_MIN_CONTINUES constant
+
+Interfaces: CollectionPipelineConfig.from_dict() class method for configuration loading
+
+Implementation: Dataclass with sensible defaults and config loading from dictionary
+"""
+
+from dataclasses import dataclass, field
+from typing import Any
+
+# Default threshold for minimum continue guards to flag
+DEFAULT_MIN_CONTINUES = 1
+
+
+@dataclass
+class CollectionPipelineConfig:
+    """Configuration for collection-pipeline linter."""
+
+    enabled: bool = True
+    """Whether the linter is enabled."""
+
+    min_continues: int = DEFAULT_MIN_CONTINUES
+    """Minimum number of if/continue patterns required to flag a violation."""
+
+    ignore: list[str] = field(default_factory=list)
+    """File patterns to ignore."""
+
+    def __post_init__(self) -> None:
+        """Validate configuration values."""
+        if self.min_continues < 1:
+            raise ValueError(f"min_continues must be at least 1, got {self.min_continues}")
+
+    @classmethod
+    def from_dict(
+        cls, config: dict[str, Any], language: str | None = None
+    ) -> "CollectionPipelineConfig":
+        """Load configuration from dictionary.
+
+        Args:
+            config: Dictionary containing configuration values
+            language: Programming language (unused, for interface compatibility)
+
+        Returns:
+            CollectionPipelineConfig instance with values from dictionary
+        """
+        return cls(
+            enabled=config.get("enabled", True),
+            min_continues=config.get("min_continues", DEFAULT_MIN_CONTINUES),
+            ignore=config.get("ignore", []),
+        )

src/linters/collection_pipeline/continue_analyzer.py

@@ -0,0 +1,100 @@
+"""
+Purpose: Analyze continue guard patterns in for loops
+
+Scope: Extract and validate if/continue patterns from loop bodies
+
+Overview: Provides helper functions for analyzing continue guard patterns in for loop
+    bodies. Handles extraction of sequential if/continue statements, validation of
+    simple continue-only patterns, and detection of side effects in conditions.
+    Separates pattern analysis logic from main detection for better maintainability.
+
+Dependencies: ast module for Python AST processing
+
+Exports: extract_continue_patterns, is_continue_only, has_side_effects, has_body_after_continues
+
+Interfaces: Functions for analyzing continue patterns in AST structures
+
+Implementation: AST-based pattern matching for continue guard identification
+"""
+
+import ast
+
+
+def extract_continue_patterns(body: list[ast.stmt]) -> list[ast.If]:
+    """Extract leading if statements that only contain continue.
+
+    Args:
+        body: List of statements in for loop body
+
+    Returns:
+        List of ast.If nodes that are continue guards
+    """
+    continues: list[ast.If] = []
+    for stmt in body:
+        if not isinstance(stmt, ast.If):
+            break
+        if not is_continue_only(stmt):
+            break
+        continues.append(stmt)
+    return continues
+
+
+def is_continue_only(if_node: ast.If) -> bool:
+    """Check if an if statement only contains continue.
+
+    Args:
+        if_node: AST If node to check
+
+    Returns:
+        True if the if statement is a simple continue guard
+    """
+    if len(if_node.body) != 1:
+        return False
+    if not isinstance(if_node.body[0], ast.Continue):
+        return False
+    if if_node.orelse:
+        return False
+    return True
+
+
+def has_side_effects(continues: list[ast.If]) -> bool:
+    """Check if any condition has side effects.
+
+    Args:
+        continues: List of continue guard if statements
+
+    Returns:
+        True if any condition has side effects (e.g., walrus operator)
+    """
+    for if_node in continues:
+        if _condition_has_side_effects(if_node.test):
+            return True
+    return False
+
+
+def _condition_has_side_effects(node: ast.expr) -> bool:
+    """Check if expression has side effects.
+
+    Args:
+        node: AST expression node to check
+
+    Returns:
+        True if expression has side effects
+    """
+    for child in ast.walk(node):
+        if isinstance(child, ast.NamedExpr):
+            return True
+    return False
+
+
+def has_body_after_continues(body: list[ast.stmt], num_continues: int) -> bool:
+    """Check if there are statements after continue guards.
+
+    Args:
+        body: List of statements in for loop body
+        num_continues: Number of continue guards detected
+
+    Returns:
+        True if there are statements after the continue guards
+    """
+    return len(body) > num_continues

src/linters/collection_pipeline/detector.py

@@ -0,0 +1,130 @@
+"""
+Purpose: AST-based detection of collection pipeline anti-patterns
+
+Scope: Pattern matching for for loops with embedded filtering via if/continue
+
+Overview: Implements the core detection logic for identifying imperative loop patterns
+    that use if/continue for filtering instead of collection pipelines. Uses Python's
+    AST module to analyze code structure and identify refactoring opportunities. Detects
+    patterns like 'for x in iter: if not cond: continue; action(x)' and suggests
+    refactoring to generator expressions or filter(). Handles edge cases like walrus
+    operators (side effects), else branches, and empty loop bodies.
+
+Dependencies: ast module, continue_analyzer, suggestion_builder
+
+Exports: PipelinePatternDetector class, PatternMatch dataclass
+
+Interfaces: PipelinePatternDetector.detect_patterns() -> list[PatternMatch]
+
+Implementation: AST visitor pattern with delegated pattern matching and suggestion generation
+"""
+
+import ast
+from dataclasses import dataclass
+
+from . import continue_analyzer, suggestion_builder
+
+
+@dataclass
+class PatternMatch:
+    """Represents a detected anti-pattern."""
+
+    line_number: int
+    """Line number where the for loop starts (1-indexed)."""
+
+    loop_var: str
+    """Name of the loop variable."""
+
+    iterable: str
+    """Source representation of the iterable."""
+
+    conditions: list[str]
+    """List of filter conditions (inverted from continue guards)."""
+
+    has_side_effects: bool
+    """Whether any condition has side effects."""
+
+    suggestion: str
+    """Refactoring suggestion as a code snippet."""
+
+
+class PipelinePatternDetector(ast.NodeVisitor):
+    """Detects for loops with embedded filtering via if/continue patterns."""
+
+    def __init__(self, source_code: str) -> None:
+        """Initialize detector with source code.
+
+        Args:
+            source_code: Python source code to analyze
+        """
+        self.source_code = source_code
+        self.matches: list[PatternMatch] = []
+
+    def detect_patterns(self) -> list[PatternMatch]:
+        """Analyze source code and return detected patterns.
+
+        Returns:
+            List of PatternMatch objects for each detected anti-pattern
+        """
+        try:
+            tree = ast.parse(self.source_code)
+            self.visit(tree)
+        except SyntaxError:
+            pass  # Invalid Python, return empty list
+        return self.matches
+
+    def visit_For(self, node: ast.For) -> None:  # pylint: disable=invalid-name
+        """Visit for loop and check for filtering patterns.
+
+        Args:
+            node: AST For node to analyze
+        """
+        match = self._analyze_for_loop(node)
+        if match is not None:
+            self.matches.append(match)
+        self.generic_visit(node)
+
+    def _analyze_for_loop(self, node: ast.For) -> PatternMatch | None:
+        """Analyze a for loop for embedded filtering patterns.
+
+        Args:
+            node: AST For node to analyze
+
+        Returns:
+            PatternMatch if pattern detected, None otherwise
+        """
+        continues = continue_analyzer.extract_continue_patterns(node.body)
+        if not continues:
+            return None
+
+        if continue_analyzer.has_side_effects(continues):
+            return None
+
+        if not continue_analyzer.has_body_after_continues(node.body, len(continues)):
+            return None
+
+        return self._create_match(node, continues)
+
+    def _create_match(self, for_node: ast.For, continues: list[ast.If]) -> PatternMatch:
+        """Create a PatternMatch from detected pattern.
+
+        Args:
+            for_node: AST For node
+            continues: List of continue guard if statements
+
+        Returns:
+            PatternMatch object with detection information
+        """
+        loop_var = suggestion_builder.get_target_name(for_node.target)
+        iterable = ast.unparse(for_node.iter)
+        conditions = [suggestion_builder.invert_condition(c.test) for c in continues]
+        suggestion = suggestion_builder.build_suggestion(loop_var, iterable, conditions)
+
+        return PatternMatch(
+            line_number=for_node.lineno,
+            loop_var=loop_var,
+            iterable=iterable,
+            conditions=conditions,
+            has_side_effects=False,
+            suggestion=suggestion,
+        )