import contextlib
import os
import traceback
from io import StringIO


def python_exec(code, python_globals, cwd=None):
    try:
        original_cwd = None
        if cwd:
            original_cwd = os.getcwd()
            os.chdir(cwd)

        output = StringIO()
        with contextlib.redirect_stdout(output):
            exec(code, python_globals)

        if original_cwd:
            os.chdir(original_cwd)

        return {"status": "success", "output": output.getvalue()}
    except Exception as e:
        if original_cwd:
            os.chdir(original_cwd)
        return {"status": "error", "error": str(e), "traceback": traceback.format_exc()}