o $i @sddlZddlZddlZddlZddlZddlZddlmZGdddejZ e dddee dej d ej d e fd d ZdS) N)ListcsPeZdZdZdZddfdd Zededed efd d Z fd d Z Z S)SubprocessCalledProcessErrorz7The subprocess.CalledProcessError with stripped stdout.2N) cmd_indexcs||_tj|i|dS)N)rsuper__init__)selfrargskwargs __class__[/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/ray/_private/runtime_env/utils.pyrsz%SubprocessCalledProcessError.__init__str_data last_n_linesreturncCs.|dkr|S|d}d|| dS)Nr )stripsplitjoin)rrlinesr r r_get_last_n_linesz-SubprocessCalledProcessError._get_last_n_linec s|jdurgnd|jdg}|t|j|jd}|D]'\}}|rId|jd|d}|||j }||dt |dq"d |S) NRun cmd[z$] failed with the following details.stdoutstderrzLast z lines of :rz ) rappendr__str__rritems LAST_N_LINESrrtextwrapindentr)rstr_listoutnamessubtitlelast_n_line_strr r rrs"  z$SubprocessCalledProcessError.__str__) __name__ __module__ __qualname____doc__r r staticmethodstrintrr __classcell__r r r rr sr) cmd_index_gencmdloggerr2rc sdt|}|d|t|d}zztj|tjjtjjd|IdH}|IdH\}}Wn$tj j y>}z|d}~wt yR}z t d|d|d}~ww| d}|rb|d||n|d||jd krvt|j|||d |W|durz|Wn tyYnw|IdHSS|durz|Wn tyYnw|IdHww) aRun command with arguments and return its output. If the return code was non-zero it raises a CalledProcessError. The CalledProcessError object will have the return code in the returncode attribute and any output in the output attribute. Args: cmd: The cmdline should be a sequence of program arguments or else a single string or path-like object. The program to execute is the first item in cmd. logger: The logger instance. cmd_index_gen: The cmd index generator, default is itertools.count(1). kwargs: All arguments are passed to the create_subprocess_exec. Returns: The stdout of cmd. Raises: CalledProcessError: If the return code of cmd is not 0. zRun cmd[%s] %sNrrz] got exception.zutf-8zOutput of cmd[%s]: %szNo output for cmd[%s]r)outputr)nextinforeprasynciocreate_subprocess_exec subprocessPIPESTDOUT communicate exceptionsCancelledError BaseException RuntimeErrordecode returncoderkillProcessLookupErrorwait) r3r4r2r rprocr_er r rcheck_output_cmd/sX           rK)r9 itertoolsloggingr;r!typestypingrCalledProcessErrorrcountr.Logger GeneratorTyperKr r r rs$ )