mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2025-12-17 14:04:27 +01:00
71 lines
1.8 KiB
Python
71 lines
1.8 KiB
Python
import docker
|
|
import os
|
|
import subprocess
|
|
|
|
|
|
WORKSPACE_FOLDER = "auto_gpt_workspace"
|
|
|
|
|
|
def execute_python_file(file):
|
|
"""Execute a Python file in a Docker container and return the output"""
|
|
|
|
print (f"Executing file '{file}' in workspace '{WORKSPACE_FOLDER}'")
|
|
|
|
if not file.endswith(".py"):
|
|
return "Error: Invalid file type. Only .py files are allowed."
|
|
|
|
file_path = os.path.join(WORKSPACE_FOLDER, file)
|
|
|
|
if not os.path.isfile(file_path):
|
|
return f"Error: File '{file}' does not exist."
|
|
|
|
try:
|
|
client = docker.from_env()
|
|
|
|
# You can replace 'python:3.8' with the desired Python image/version
|
|
# You can find available Python images on Docker Hub:
|
|
# https://hub.docker.com/_/python
|
|
container = client.containers.run(
|
|
'python:3.10',
|
|
f'python {file}',
|
|
volumes={
|
|
os.path.abspath(WORKSPACE_FOLDER): {
|
|
'bind': '/workspace',
|
|
'mode': 'ro'}},
|
|
working_dir='/workspace',
|
|
stderr=True,
|
|
stdout=True,
|
|
detach=True,
|
|
)
|
|
|
|
output = container.wait()
|
|
logs = container.logs().decode('utf-8')
|
|
container.remove()
|
|
|
|
# print(f"Execution complete. Output: {output}")
|
|
# print(f"Logs: {logs}")
|
|
|
|
return logs
|
|
|
|
except Exception as e:
|
|
return f"Error: {str(e)}"
|
|
|
|
|
|
|
|
def execute_shell(command_line):
|
|
|
|
old_dir = os.getcwd()
|
|
|
|
workdir = os.path.abspath(os.path.join(os.path.dirname( __file__ ), '..', WORKSPACE_FOLDER))
|
|
|
|
print (f"Executing command '{command_line}' in working directory '{workdir}'")
|
|
|
|
os.chdir(workdir)
|
|
|
|
result = subprocess.run(command_line, capture_output=True, shell=True)
|
|
output = f"STDOUT:\n{result.stdout}\nSTDERR:\n{result.stderr}";
|
|
|
|
os.chdir(old_dir)
|
|
|
|
return output
|