Function execute_python_code

src/sandbox_vm/tools/code/python.py:41–143

Execute Python code. Saves the script to a file if save_to_file is True and a parent and filename are provided. Returns: { stdout, stderr, return_code, execution_time, file_path?, error? }

(
    ctx: Context,
    code: str,
    timeout: int = 30,
    *,
    parent: Optional[str] = None,
    filename: Optional[str] = None,
    save_to_file: bool = False,
)

async def execute_python_code(
    ctx: Context,
    code: str,
    timeout: int = 30,
    *,
    parent: Optional[str] = None,
    filename: Optional[str] = None,
    save_to_file: bool = False,
) -> dict:
    """
    Execute Python code. Saves the script to a file if save_to_file is True and a parent and filename are provided.

    Returns: { stdout, stderr, return_code, execution_time, file_path?, error? }
    """
    sid = getattr(ctx, "session_id", None)
    if not sid:
        return _err("session_id_REQUIRED", message="ctx.session_id is missing")

    rt = MANAGER.get(sid)
    env = rt.minimal_env()
    tout = _clamp_timeout(timeout)

    try:
        base_dir = (
            (rt.root / parent.lstrip("/")) if parent else (rt.root / "tmp" / "code")
        ).resolve()
        base_dir.mkdir(parents=True, exist_ok=True)
    except Exception as e:
        return _err(type(e).__name__, message=str(e))

    clean_code = textwrap.dedent(code or "").strip() + "\n"

    temp_file_path: Optional[str] = None
    persisted = False
    result = {
        "stdout": "",
        "stderr": "",
        "return_code": -1,
        "execution_time": 0.0,
    }

    try:
        with tempfile.NamedTemporaryFile(
            mode="w", suffix=".py", dir=base_dir, delete=False
        ) as tf:
            tf.write(clean_code)
            temp_file_path = tf.name
        # tempfile.NamedTemporaryFile creates files with 0600 (mkstemp);
        # relax to 0644 so the host process can read them for backups.
        os.chmod(temp_file_path, 0o644)

        if save_to_file:
            final_name = _sanitize_filename(
                filename or f"code_{Path(temp_file_path).name}"
            )
            final_path = Path(base_dir) / final_name
            os.replace(temp_file_path, final_path)
            temp_file_path = str(final_path)