Function run_pyspector

tests/unit/test_false_positive_reductions.py:23–67

Write code to a temp file, run pyspector, return findings as list of dicts.

(code: str, *, filename: str = "sample_code.py", in_tests_dir: bool = False)

# ---------------------------------------------------------------------------

def run_pyspector(code: str, *, filename: str = "sample_code.py", in_tests_dir: bool = False) -> list[dict]:
    """Write code to a temp file, run pyspector, return findings as list of dicts."""
    from pyspector._rust_core import run_scan
    from pyspector.config import get_default_rules

    rules_toml = get_default_rules()

    with tempfile.TemporaryDirectory() as tmpdir:
        if in_tests_dir:
            subdir = os.path.join(tmpdir, "tests")
            os.makedirs(subdir)
            file_path = os.path.join(subdir, filename)
        else:
            file_path = os.path.join(tmpdir, filename)

        Path(file_path).write_text(textwrap.dedent(code))

        import ast as _ast, json as _json, warnings

        with warnings.catch_warnings():
            warnings.filterwarnings("ignore")
            try:
                tree = _ast.parse(Path(file_path).read_text())
                import sys
                # Use AstEncoder from cli
                sys.path.insert(0, str(Path(__file__).parents[2] / "src"))
                from pyspector.cli import AstEncoder
                ast_json = _json.dumps(tree, cls=AstEncoder)
            except Exception:
                ast_json = "{}"

        rel_path = os.path.basename(file_path) if not in_tests_dir else f"tests/{filename}"
        python_files = [{"file_path": rel_path, "content": Path(file_path).read_text(), "ast_json": ast_json}]

        results = run_scan(
            tmpdir if not in_tests_dir else str(Path(tmpdir)),
            rules_toml,
            {"exclude": []},
            python_files,
        )

    return [
        {"rule_id": r.rule_id, "file_path": r.file_path, "line_number": r.line_number, "code": r.code}
        for r in results
    ]


def findings_for_rule(code: str, rule_id: str, **kwargs) -> list[dict]: