U k&p[7.@stddlZddlZddlZddlZddlZddlZddlZddlZddZddZ ddZ dd Z d d Z d d Z dS)Nc Osd}i}|D]*\}}|||<||kr ||||<||=q t|f||\}}} |dkrdd||| f} |dr|drt| n|drt| t| |S)aXRun external command or pipeline. Example: ``runcmd(['grep', 'foo'], ['wc', '-l'], feed_stdin='foo bar ')`` Return the standard output of the command. Raise ``cliapp.AppException`` if external command returns non-zero exit code. ``*args`` and ``**kwargs`` are passed onto ``subprocess.Popen``. )) ignore_failF) log_errorTrzCommand failed: %s %s %s rr)runcmd_uncheckedjoinlogginginfoerrorcliappZ AppException) argvargskwargsZ our_optionsZoptsnamedefaultZ exit_codeouterrmsgr//usr/lib/python3/dist-packages/cliapp/runcmd.pyruncmds"     rc s|gt|}td|fdd}dd}|dd}|dtj}|d tj}|d tj}|d |} |d |} |d d} |dd} z*t||||} t| ||||| | | | WStk r}z,|jtj kr|j dkr|d|_ |nW5d}~XYnXdS)zRun external command or pipeline. Return the exit code, and contents of standard output and error of the command. See also ``runcmd``. zrun external command: %rcs"|kr|}|=|S|SdSNr)rrvaluer rr pop_kwargQs z#runcmd_unchecked..pop_kwargcSsdSrr)_rrrnoopYszruncmd_unchecked..noop feed_stdinstdinstdoutstderrstdout_callbackstderr_callbackoutput_timeoutNtimeout_callbackr) listrdebug subprocessPIPE_build_pipeline _run_pipelineOSErrorerrnoZENOENTfilename)r argvsr rrr pipe_stdin pipe_stdout pipe_stderrr!r"r#r$ZpipelineerrrrDs@            rcCsg}|tjkr t\}}|}n|}t|D]\} } | dkrV| t|dkrV|} |} nD| dkrj|} tj} n0| t|dkr|dj} |} n|dj} tj} tj| f| | |dd|} | dkr| | | q,|tjkrt|t |d|d_ |S)NrT)rrr Z close_fdsrb) r'r(ospipe enumeratelenrPopencloseappendfdopenr )r.r/r0r1r procsZrpipeZwpiper ir rrprrrr)ws<        r)c  sddg} g} d} d} t} d}d}dd}|rP|tjkrP|djtjkrl|djtjkr|djfdd}|s|rg}s̈tjkr|djstjkr|djg}|tjkr| t|kr|dj|s$|rzt ||g|\}}}WnDt j k r}z"|j dt j krpWY qڂW5d}~XYnXnqt}|| }|dk o||k}dj|kr| t|kr|| | | }dj|| t|7} | t|krdjdj|krbdj| }|rZ||}|dkrN|}| |nd d}dj|krdj| }|r||}|dkr|}| |nd d}|r|r|}|rd }q|rd }q|s |r D]}|jdkr|qqd d Dp dg}djdk r@djdjdk r^djdjdk r|dj|dd | d | fS) NFricSs.t|tjd}|tjB}t|tj|dS)Nr)fcntlZF_GETFLr6 O_NONBLOCKZF_SETFL)fdflagsrrrset_nonblockings z&_run_pipeline..set_nonblockingr4csTD] }|qD]}|jdkrdSqtjkr>s>dStjkrPsPdSdS)NTF)Zpoll returncoder'r()r@r1r0r>Z stderr_eofZ stdout_eofrr still_runnings  z$_run_pipeline..still_runningTcSsg|]}|jdkr|jqS)r)rF).0r@rrr s z!_run_pipeline..)timer'r(rfilenorr r<r9selectr r r,ZEINTRwriter;readrFwaitr)r>rr/r0r1r!r"r#r$rrposZio_sizeZ latest_outputZ timeout_quitZtimeoutrErHZrlistZwlistrwrr2ZnowZtime_since_outputdataZdata_newresultr@Z errorcodesrrGrr*s             r*cCsrd}d}d}d}t||||}g}|D]:}||krD||q,|dkrX|dq,|d|q,d|S) z#Return a shell-quoted version of s.ZabcdefghijklmnopqrstuvwxyzZABCDEFGHIJKLMNOPQRSTUVWXYZ 0123456789z-_/=.,:'z"'"z'%c'r)setr<r)sZ lower_asciiZ upper_asciiZdigitsZ punctuationZsafeZquotedcrrr shell_quotes  r\c Ksdg}|dd}|r"|dn|dkr4|d|dg}|ttt||||d|d d}|r|tttd d d d |g|ttt|}t|f|S)aDRun command in argv on remote host target. This is similar to runcmd, but the command is run on the remote machine. The command is given as an argv array; elements in the array are automatically quoted so they get passed to the other side correctly. An optional ``tty=`` parameter can be passed to ``ssh_runcmd`` in order to force or disable pseudo-tty allocation. This is often required to run ``sudo`` on another machine and might be useful in other situations as well. Supported values are ``tty=True`` for forcing tty allocation, ``tty=False`` for disabling it and ``tty=None`` for not passing anything tty related to ssh. With the ``tty`` option, ``cliapp.runcmd(['ssh', '-tt', 'user@host', '--', 'sudo', 'ls'])`` can be written as ``cliapp.ssh_runcmd('user@host', ['sudo', 'ls'], tty=True)`` which is more intuitive. Arbitrary command line options to the ssh command can be given with the ``ssh_options`` argument. For example: ``cliapp.ssh_runcmd( 'user@host', ['sudo', 'ls'], ssh_options=['-oStrictHostChecking=no'])`` The remote command is run in the user's home directory, by default. The directory can be changed with the keyword argument ``remote_cwd``. The target is given as-is to ssh, and may use any syntax ssh accepts. Environment variables may or may not be passed to the remote machine: this is dependent on the ssh and sshd configurations. Invoke env(1) explicitly to pass in the variables you need to exist on the other end. Pipelines are not supported. ZsshttyNz-ttFz-TZ ssh_optionsz-- remote_cwdZshz-czcd "$1" && shift && exec "$@"-)popr<extendr%mapr\r)targetr r Zssh_argvr]Z more_optionsr^Z local_argvrrr ssh_runcmd0s**         rd)r,rArr6rNr'rLr rrr)r*r\rdrrrrs&3/u