|
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()}
|