ballpython 2.0.3__py3-none-any.whl → 2.0.5__py3-none-any.whl

{ballpython-2.0.3.dist-info → ballpython-2.0.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ballpython
-Version: 2.0.3
+Version: 2.0.5
 Summary: The Ultimate Static Python Intelligence, Healing, Type Verification, and Security Suite
 Author: Developer
 License-Expression: MIT
@@ -78,7 +78,7 @@ Add `ballpython` to your `.pre-commit-config.yaml`:
 ```yaml
 repos:
   - repo: https://github.com/moonrox420/ball-python
-    rev: v2.0.3
+    rev: v2.0.5
     hooks:
       - id: pycleaner
         args: ["check"]

{ballpython-2.0.3.dist-info → ballpython-2.0.5.dist-info}/RECORD

@@ -2,11 +2,11 @@ ballpython/__init__.py,sha256=d8ONxgaz9jml_fHPVvg867Vt5jGRhGTqzWPtux3WJOc,206
 ballpython/__main__.py,sha256=aCRr0vnZfKYHDrPHBsYSjX5xTxtFJpTTSPVUSZTBK-I,136
 ballpython/cli.py,sha256=LllCLRAQQ-fRR_lIXwmd2e3Q7DLCIvf-5jHBXsAhJWs,125
 ballpython/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pycleaner/__init__.py,sha256=dK4jaytHQuXqaK3pqum6o8z8JrRLU-XHUfv_NBOtdxs,2894
+pycleaner/__init__.py,sha256=G8pQ2M9iH32OK4bKf5Koao554EdGiiHue5rdZ5AS7sM,2894
 pycleaner/__main__.py,sha256=8QBFrPyxId563F6cipPMNVEBTG8TIZUgxQfp8gk1VSI,135
 pycleaner/baseline.py,sha256=DS32C8_G8khYYNJo42FVv6Cc58lJj7NUmPOzWNfQwlU,3856
 pycleaner/cache.py,sha256=WhCiMkuL6ryWVr3l3XrYLLoI7A3bRF8jNxpOyedNW8M,10761
-pycleaner/cli.py,sha256=NhMMmLcf7RMuKjc0rjCmSGYn5ns_yib2rUbXRJDVaYY,73034
+pycleaner/cli.py,sha256=MMK0U57pam-FuvB1UL7jwVk7m9Z_Cosuwuu5FIfrPp0,73029
 pycleaner/complexity_analyzer.py,sha256=uGC40cHCcr1tshzfIUw56sodEDySMFR3kGmjo7nq_1c,14977
 pycleaner/config.py,sha256=XhloyaY_9pdwbQuc3QaAPCerLeXz-i1da4xT50h-pjY,8514
 pycleaner/dead_code_detector.py,sha256=1DwtBSZL1ug1OI9-6jcJyxkhqHmJ0F8ZR19POI2zO3c,26549
@@ -18,17 +18,17 @@ pycleaner/linter_formatter.py,sha256=LB5j66OCASU8WG8QgRFpSX7C2c7_UNMBgkJh2hSWxAY
 pycleaner/modernizer.py,sha256=L9hxXXfdyTwf8itDO58y6bW5LobGo6dg8q537QdiYbM,12946
 pycleaner/pipeline.py,sha256=ZG7_9PNz0jBadFrZoiI7juLtUJXglGQhLgQhrLJqEac,20612
 pycleaner/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pycleaner/security_scanner.py,sha256=h7WSPYf40X-PBlJVOsJL-KaknDaOcWCMhcPwXkUvlbA,21986
+pycleaner/security_scanner.py,sha256=cWf4cAGQrgwsGslSQDfZv0XUN3sWvlcOIQaogpmv6C0,23422
 pycleaner/syntax_healer.py,sha256=TlIMKIJhaX9GED3e2JNzNMJ5aXmL9eLJMCu7xKSjdv8,26711
 pycleaner/taint_engine.py,sha256=96cZdDpSPbo5b13zIZY18f_4GJm_tw0clqCZMBtjXMQ,26481
 pycleaner/test_generator.py,sha256=TJsNJSojc6eIsaKZkwMlpnb-MKHuDpNLZCPnMlO-a8M,19663
 pycleaner/type_checker.py,sha256=h09QI_8MRHv83s23myjSYFdMTsWoQ900ffBJHtRuhmE,34087
 pycleaner/typeshed_resolver.py,sha256=N-kWU_CI5DG59lQMkt7wMW9sHYjNCBdfFS42b4y2Kr4,12742
 pycleaner/verifier.py,sha256=pOfSLRNHe4qDUYJG8mxDxjY6Ph8QKBdJZmqBtoJX87w,17937
-pycleaner/frameworks/__init__.py,sha256=7MLgg469hCNtQR6dsE5gZcAZkCywJHz9zOPrbH8NmrE,3415
-pycleaner/frameworks/plugins.py,sha256=K3J4mVJw2MR_pa6Mhmo5oZOX8pUesL8lx9LUo4RLrNo,9390
-ballpython-2.0.3.dist-info/METADATA,sha256=K1nMS6fIKnHpTsJ0mRLaKNv_o1TFuzb7SI_KgywGCaw,3466
-ballpython-2.0.3.dist-info/WHEEL,sha256=YVMoNqKzERt-wjUZwJ33xBGAwnFl-4cqbYkTtWa4itE,91
-ballpython-2.0.3.dist-info/entry_points.txt,sha256=6ux-5UxoVSmlK5p9M-GDVt24sCdwTQq9wgRDkpMIYU8,81
-ballpython-2.0.3.dist-info/top_level.txt,sha256=u6lmy8RWZ_saxCZAQTL7EijB4sunegOiVyk5fHjpwLk,21
-ballpython-2.0.3.dist-info/RECORD,,
+pycleaner/frameworks/__init__.py,sha256=-W24zqvug8eUkkwqla5oh7rFOqQR-WI4TohQhbXGYoo,3472
+pycleaner/frameworks/plugins.py,sha256=sMhP9mZ82sx10OI3opgXIEPXNAWFZRe2I__6yh2Uq_E,11453
+ballpython-2.0.5.dist-info/METADATA,sha256=LqjBIVBjanDL5rdXJ7gES5vF4ezSqbG2Y1SKMG6rXx4,3466
+ballpython-2.0.5.dist-info/WHEEL,sha256=YVMoNqKzERt-wjUZwJ33xBGAwnFl-4cqbYkTtWa4itE,91
+ballpython-2.0.5.dist-info/entry_points.txt,sha256=6ux-5UxoVSmlK5p9M-GDVt24sCdwTQq9wgRDkpMIYU8,81
+ballpython-2.0.5.dist-info/top_level.txt,sha256=u6lmy8RWZ_saxCZAQTL7EijB4sunegOiVyk5fHjpwLk,21
+ballpython-2.0.5.dist-info/RECORD,,

pycleaner/__init__.py

@@ -18,7 +18,7 @@ from pycleaner.discovery import (
     is_protected_file,
 )
 
-__version__ = "2.0.3"
+__version__ = "2.0.5"
 __all__ = [
     "BaselineFingerprint",
     "BaselineManager",

pycleaner/cli.py

@@ -567,7 +567,7 @@ def _resolve_cli_command(args: argparse.Namespace) -> str:
         return "all"
     if getattr(args, "check", False):
         return "check"
-    return "ultimate"
+    return "fix"
 
 
 def _run_all_command(

pycleaner/frameworks/__init__.py

@@ -103,6 +103,7 @@ def get_default_registry() -> FrameworkRegistry:
         FastAPIPlugin,
         PydanticPlugin,
         PytestPlugin,
+        PyTorchPlugin,
         SQLAlchemyPlugin,
     )
 
@@ -112,4 +113,5 @@ def get_default_registry() -> FrameworkRegistry:
     reg.register(FastAPIPlugin())
     reg.register(SQLAlchemyPlugin())
     reg.register(DataclassPlugin())
+    reg.register(PyTorchPlugin())
     return reg

pycleaner/frameworks/plugins.py

@@ -2,7 +2,7 @@
 pycleaner.frameworks.plugins
 ============================
 
-Concrete Framework Plugins for Pydantic, Pytest, FastAPI, SQLAlchemy, and Dataclasses.
+Concrete Framework Plugins for Pydantic, Pytest, FastAPI, SQLAlchemy, Dataclasses, and PyTorch.
 """
 
 from __future__ import annotations
@@ -50,15 +50,18 @@ class PydanticPlugin:
         for node in ast.walk(tree):
             if isinstance(node, ast.Import):
                 for alias in node.names:
-                    if alias.name.split(".")[0] == "pydantic":
+                    if alias.name.split(".")[0] in ("pydantic", "pydantic_settings"):
                         return True
             elif isinstance(node, ast.ImportFrom):
-                if node.module and node.module.split(".")[0] == "pydantic":
+                if node.module and node.module.split(".")[0] in (
+                    "pydantic",
+                    "pydantic_settings",
+                ):
                     return True
         return False
 
     def get_protected_names(self, tree: ast.AST) -> set[str]:
-        return {"model_config", "Config", "ConfigDict"}
+        return {"model_config", "Config", "ConfigDict", "model_post_init"}
 
     def get_protected_decorators(self) -> set[str]:
         return self._VALIDATOR_DECORATORS
@@ -288,3 +291,74 @@ class DataclassPlugin:
         tree: ast.AST,
     ) -> bool:
         return name in self.get_protected_names(tree)
+
+
+class PyTorchPlugin:
+    """Understands PyTorch nn.Module, LightningModule, HuggingFace models, and datasets."""
+
+    name = "pytorch"
+    _MODULE_BASES = {
+        "Module",
+        "LightningModule",
+        "PreTrainedModel",
+        "Dataset",
+        "IterableDataset",
+        "Sampler",
+    }
+    _PROTECTED_METHODS = {
+        "forward",
+        "backward",
+        "training_step",
+        "validation_step",
+        "test_step",
+        "predict_step",
+        "configure_optimizers",
+        "optimizer_step",
+        "compute_loss",
+        "reset_parameters",
+        "extra_repr",
+        "state_dict",
+        "load_state_dict",
+        "collate_fn",
+    }
+
+    def is_applicable(self, tree: ast.AST, filepath: Path | str) -> bool:
+        torch_pkgs = {
+            "torch",
+            "torchvision",
+            "torchaudio",
+            "pytorch_lightning",
+            "lightning",
+            "transformers",
+            "accelerate",
+            "timm",
+        }
+        for node in ast.walk(tree):
+            if isinstance(node, ast.Import):
+                for alias in node.names:
+                    if alias.name.split(".")[0] in torch_pkgs:
+                        return True
+            elif isinstance(node, ast.ImportFrom):
+                if node.module and node.module.split(".")[0] in torch_pkgs:
+                    return True
+        return False
+
+    def get_protected_names(self, tree: ast.AST) -> set[str]:
+        return self._PROTECTED_METHODS
+
+    def get_protected_decorators(self) -> set[str]:
+        return set()
+
+    def is_protected_field(self, node: ast.AST, class_node: ast.ClassDef) -> bool:
+        return _has_base_named(class_node, self._MODULE_BASES)
+
+    def should_ignore_definition(
+        self,
+        name: str,
+        kind: str,
+        node: ast.AST,
+        context: str,
+        tree: ast.AST,
+    ) -> bool:
+        return name in self._PROTECTED_METHODS
+

pycleaner/security_scanner.py

@@ -190,6 +190,24 @@ class _DangerousCallDetector(ast.NodeVisitor):
         self.filepath = filepath
         self.source_lines = source_lines
         self.findings: list[SecurityFinding] = []
+        self._is_test_module: bool = False
+
+    def check_tree(self, tree: ast.AST) -> None:
+        """Inspect imports to determine if module is a test file."""
+        for node in ast.walk(tree):
+            if isinstance(node, ast.Import):
+                for alias in node.names:
+                    if alias.name.split(".")[0] in ("pytest", "unittest", "mock"):
+                        self._is_test_module = True
+                        return
+            elif isinstance(node, ast.ImportFrom):
+                if node.module and node.module.split(".")[0] in (
+                    "pytest",
+                    "unittest",
+                    "mock",
+                ):
+                    self._is_test_module = True
+                    return
 
     def _get_snippet(self, lineno: int) -> str:
         if 1 <= lineno <= len(self.source_lines):
@@ -245,8 +263,11 @@ class _DangerousCallDetector(ast.NodeVisitor):
         parts = [p.lower() for p in Path(filepath).parts[:-1]]
         return any(p in ("tests", "test", "testing") for p in parts)
 
+    def _is_test(self) -> bool:
+        return self._is_test_module or self._is_test_path(self.filepath)
+
     def visit_Assert(self, node: ast.Assert) -> None:
-        if self._is_test_path(self.filepath):
+        if self._is_test():
             return
         self._add_finding(
             SecurityFinding(
@@ -429,7 +450,20 @@ class _DangerousCallDetector(ast.NodeVisitor):
         if isinstance(arg, ast.BinOp) and isinstance(arg.op, ast.Add):
             return True
         if isinstance(arg, ast.Call):
-            return isinstance(arg.func, ast.Attribute) and arg.func.attr == "format"
+            if isinstance(arg.func, ast.Attribute) and arg.func.attr == "format":
+                # Check for psycopg safe sql.SQL(...).format(...) query composition
+                val = arg.func.value
+                if isinstance(val, ast.Call):
+                    func = val.func
+                    func_id = ""
+                    if isinstance(func, ast.Name):
+                        func_id = func.id
+                    elif isinstance(func, ast.Attribute):
+                        func_id = func.attr
+                    if func_id == "SQL":
+                        return False
+                return True
+            return False
         if isinstance(arg, ast.BinOp) and isinstance(arg.op, ast.Mod):
             return isinstance(arg.left, ast.Constant) and isinstance(
                 arg.left.value, str
@@ -483,6 +517,7 @@ class SecurityScanner:
             tree = ast.parse(source, filename=filename)
             source_lines = source.splitlines()
             detector = _DangerousCallDetector(filename, source_lines)
+            detector.check_tree(tree)
             detector.visit(tree)
             findings.extend(detector.findings)
         except SyntaxError: