Create src/macg/tools/python_runner.py

src/macg/tools/python_runner.py

@@ -0,0 +1,42 @@
+from __future__ import annotations
+import os
+import subprocess
+import tempfile
+from dataclasses import dataclass
+
+@dataclass
+class RunResult:
+    ok: bool
+    stdout: str
+    stderr: str
+    returncode: int
+
+def run_pytest(module_name: str, code: str, tests: str, timeout_s: int = 15) -> RunResult:
+    """
+    Minimal safety:
+    - runs in temp dir
+    - timeout
+    NOTE: still executes arbitrary code. For “serious” safety, use containers/sandboxing.
+    """
+    with tempfile.TemporaryDirectory() as d:
+        mod_path = os.path.join(d, f"{module_name}.py")
+        test_path = os.path.join(d, "test_solution.py")
+        with open(mod_path, "w", encoding="utf-8") as f:
+            f.write(code)
+        with open(test_path, "w", encoding="utf-8") as f:
+            f.write(tests)
+
+        cmd = ["python", "-m", "pytest", "-q"]
+        try:
+            p = subprocess.run(
+                cmd,
+                cwd=d,
+                capture_output=True,
+                text=True,
+                timeout=timeout_s,
+                env={**os.environ, "PYTHONPATH": d},
+            )
+            ok = (p.returncode == 0)
+            return RunResult(ok=ok, stdout=p.stdout, stderr=p.stderr, returncode=p.returncode)
+        except subprocess.TimeoutExpired as e:
+            return RunResult(ok=False, stdout=e.stdout or "", stderr="Timeout running pytest.", returncode=124)