Fd$ dZddlZddlZddlmZddlZddlmZmZddl m Z ddl m Z ddlmZddlmZdd lmZd Zd Zed d ddddddddddedede defdZedddddddidede defdZdededefd Zed!d"d#dd$ddid%d&'d#ede defd(Zed)d"d*dd+ddid,d-de defd.Zdefd/ZdS)0z"Execute code in a Docker containerN)Path)DockerException ImageNotFound) Container)Agent)command)Config)logger allowlistdenylistexecute_python_codez%Creates a Python file and executes itstringzThe Python code to runT)type descriptionrequiredz%A name to be given to the python file)codenamerragentreturnc4|j}|jt|d}t j|d|ds|dz}|j||z }||sdS t|dd5}| |d d d n #1swxYwYtt||S#t$r}d t|cYd }~Sd }~wwxYw) aiCreate and execute a Python file in a Docker container and return the STDOUT of the executed code. If there is any data that needs to be captured use a print statement Args: code (str): The Python code to run name (str): A name to be given to the Python file Returns: str: The STDOUT captured from the code when it ran executed_codeT)exist_ok.pyzDError: 'name' argument resulted in path traversal, operation abortedzw+utf-8)encodingNError: ) ai_name workspaceget_pathrosmakedirsendswithis_relative_toopenwriteexecute_python_filestr Exception)rrrrcode_dir file_pathfes QC:\Users\Administrator\Downloads\Auto-GPT-master\autogpt\commands\execute_code.pyr r si6mG''Wo(F(FGGHK4(((( ==  e|((D99I  # #H - -VUU" )TG 4 4 4  GGDMMM               #3y>>5999 """!Q!!!!!!!!"sBC1'C = C1 C  C1C C11 D;D DDr&z Executes an existing Python filefilenamezThe name of te file to executec tjd|d|jjd|dsdSt |}|sd|dStrMtj dt|gd d |jj }|j d kr|j Sd |j S tj}d} |j|tjd|dn#t&$rtjd|dtj}||d d D]`}|d}|d} |r| rtj|d| J|rtj|aYnwxYw|j |dt||jjg|jjdddidd d d } | | d} | | S#t<$r0} tjdd t| cYd} ~ Sd} ~ wt>$r} d t| cYd} ~ Sd} ~ wwxYw)zExecute a Python file in a Docker container and return the output Args: filename (str): The name of the file to execute Returns: str: The output of the file zExecuting python file '' in working directory ''rz5Error: Invalid file type. Only .py files are allowed.zpython: can't open file 'z&': [Errno 2] No such file or directorypythonTutf8)capture_outputrcwdrrzpython:3-alpinezImage 'z' found locallyz,' not found locally, pulling from Docker Hub)streamdecodestatusprogressz: z /workspacero)bindmode)volumes working_dirstderrstdoutdetachrzzCould not run the script in a container. If you haven't already, please install Docker https://docs.docker.com/get-docker/N) r infoconfigworkspace_pathr"ris_file$we_are_running_in_a_docker_container subprocessrunr' returncoder@r?dockerfrom_envimagesgetwarnr APIClientpull containers relative_torrootwaitlogsr7removerr() r.rr*resultclient image_namelow_level_clientliner8r9 containerrUr,s r-r&r&Dsz( Kb(bbELD_bbb   U # #GFFXI       Y X X X ,-- - s9~~ & +       ! != ,V],, ,7"""'  ( M  j ) ) ) K=*=== > > > > ( ( ( KR*RRR    &/11 (--jd-SS ( ((++88J//(h(K6 7 7X 7 78888(K''' ( ( (&,%6%:%:  s9001EFFGG H +( .. %&; & &  ~~&&w//  """  I   "Q!!!!!!!! """!Q!!!!!!!!"sVI,2D I, B.F<9I,;F<rms 68rezYou are not allowed to run local shell commands. To execute shell commands, EXECUTE_LOCAL_COMMANDS must be set to 'True' in your config file: .env - do not attempt to bypass the restriction.)enableddisabled_reasonct||jstjd|ddSt j}||jjstj |jjtjd|dtj dtj |dd}d |j d |j}tj ||S) zExecute a shell command and return the output Args: command_line (str): The command line to execute Returns: str: The output of the command Command ' ' not allowed)Error: This Shell Command is not allowed.Executing command 'r0r1T)r4shellzSTDOUT: z STDERR: )rdrCr rBrr5r#rDr chdirgetcwdrGrHr@r?)rgr current_dirrWoutputs r-rfrfs0 L%, 7 7; ; ;;;<<<::(**K  % %el&A B B. ,--- KRlRRBIKKRRR^LT J J JF B B B6= B BFH[ Mreexecute_shell_popenqueryzThe search queryc|jSrirjrls r-rmrms 60rezYou are not allowed to run local shell commands. To execute shell commands, EXECUTE_LOCAL_COMMANDS must be set to 'True' in your config. Do not attempt to bypass the restriction.ct||jstjd|ddSt j}|jj|vrt j|jjtjd|dt jdtj }tj |d||}t j|d t|j dS) aExecute a shell command with Popen and returns an english description of the event and the process id Args: command_line (str): The command line to execute Returns: str: Description of the fact that the process started and its id rqrrrsrtr0r1T)rur@r?zSubprocess started with PID:') rdrCr rBr rwrDrvrGDEVNULLPopenr'pid)rgrrxdo_not_show_outputprocesss r-rzrzs2 L%, 7 7; ; ;;;<<<::)++K |"+55 ,--- KRlRRBIKKRRR$+D);DVG H[ >3w{+;+; > > >>rec@tjdS)zCheck if we are running in a Docker container Returns: bool: True if we are running in a Docker container, False otherwise z /.dockerenv)r pathexistsrer-rFrF!s 7>>- ( ((re)__doc__r rGpathlibrrJ docker.errorsrrdocker.models.containersrDockerContainerautogpt.agent.agentrautogpt.command_decoratorrautogpt.configr autogpt.logsr r`DENYLIST_CONTROLr'r r&boolrdrfrzrFrrer-rs((  88888888AAAAAA%%%%%%------!!!!!! +3  B    "c""U"s"""! "B &;    Z"#Z"eZ"Z"Z"Z"  Z"z9c969d9999* =8   9 8LEc> =-  10@ ?U ?s ? ? ? ?F)d))))))re