Function run_pyspector

tests/unit/test_taint_engine_extension.py:24–55

(code: str, *, filename: str = "app.py")

def run_pyspector(code: str, *, filename: str = "app.py") -> list[dict]:
    from pyspector._rust_core import run_scan
    from pyspector.config import get_default_rules

    rules_toml = get_default_rules()

    with tempfile.TemporaryDirectory() as tmpdir:
        file_path = os.path.join(tmpdir, filename)
        Path(file_path).write_text(_wrap_in_function(code))

        import ast as _ast, json as _json
        from pyspector.cli import AstEncoder

        with warnings.catch_warnings():
            warnings.filterwarnings("ignore")
            try:
                tree = _ast.parse(Path(file_path).read_text())
                ast_json = _json.dumps(tree, cls=AstEncoder)
            except Exception:
                ast_json = "{}"

        python_files = [{
            "file_path": filename,
            "content": Path(file_path).read_text(),
            "ast_json": ast_json,
        }]

        results = run_scan(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(code, rule_id, **kw):