Create python_runner.py

tools/python_runner.py

@@ -0,0 +1,52 @@
+from typing import Any, Dict
+import io
+import traceback
+import contextlib
+import signal
+
+class TimeoutException(Exception):
+    pass
+
+def timeout_handler(signum, frame):
+    raise TimeoutException("Execution timed out.")
+
+def run_python(code: str, timeout: int = 5) -> Dict[str, Any]:
+    """
+    Safely execute Python code and capture output/errors.
+    """
+
+    safe_globals = {
+        "__builtins__": {
+            "print": print,
+            "range": range,
+            "len": len,
+            "int": int,
+            "float": float,
+            "str": str,
+            "list": list,
+            "dict": dict,
+            "sum": sum,
+            "min": min,
+            "max": max,
+            "abs": abs,
+        }
+    }
+
+    stdout_buffer = io.StringIO()
+
+    try:
+        signal.signal(signal.SIGALRM, timeout_handler)
+        signal.alarm(timeout)
+
+        with contextlib.redirect_stdout(stdout_buffer):
+            exec(code, safe_globals, {})
+
+        signal.alarm(0)
+
+        return {"status": "success", "output": stdout_buffer.getvalue()}
+
+    except TimeoutException as e:
+        return {"status": "error", "output": str(e)}
+
+    except Exception:
+        return {"status": "error", "output": traceback.format_exc()}