o 3g:@sdddlmZddlZddlZddlZddlZddlZddlmZ ddlm Z ddl m Z ddl mZddZd d Zd(d d Zd(d dZerHeneZGdddeZGdddeZGdddeZGdddeZGdddZe ZdaddZe eddZde_ e!dd Z"d!d"Z#e$e#d(d#d$Z%e&Z'e&Z(e'Z)ej*Z+ddd%dddefd&d'Z,dS))) annotationsN)Empty)Queue)Thread)IS_WIN32cCs||||||j||fSN)verify returncodeprocretcodetimeoutstdoutstderrrR/home/garg/my-data/venv/lib/python3.10/site-packages/plumbum/commands/processes.py_check_processs rcs@gfdd}|r|d|j|d|jt|dd}|s S)aGet a list of all valid standard streams for proc that were opened with PIPE option. If proc was started from a Pipeline command, this function assumes it will have a "srcproc" member pointing to the previous command in the pipeline. That link will be used to traverse all started processes started from the pipeline, the list will include stdout/stderr streams opened as PIPE for all commands in the pipeline. If that was not the case, some processes could write to pipes no one reads from which would result in process stalling after the pipe's buffer is filled. Streams that were closed (because they were redirected to the input of a subsequent command) are not included in the result cs$|dus|jr dS||fdSr)closedappend)type_streamstreamsrr add_stream&sz&_get_piped_streams..add_streamrsrcprocN)rrgetattr)r rrrr_get_piped_streamss    rc #s~ddlmmtfdd}|D] }|Vdur)nqD]\}}|D] }||fVq2q,dS)Nr) EVENT_READDefaultSelectorc3s|}D] \}}|||q |}|s*r*tdtddtdd|D]\}}|j|jfVq,q)NTpopen line timeout expiredargvmachine)registerselectProcessLineTimedOutrdatafileobjreadline)sel stream_typerreadykey_maskrrdecode line_timeoutlinesizer rrrselector9s     z#_iter_lines_posix..selector) selectorsrrrpoll) r r/r1r0r2retr*rlinerr.r_iter_lines_posix3s   r7c #s Gfdddt}|durtd}t|d|j|d|jg}t}d} z Vt}d}Wn tyCd}Ynwt||krZtdt |d dt |d d| duran|rht d q+|D]}| qk zVWn tyYdSwqs) Ncs6eZdZfddZfddZfddZZS)z _iter_lines_win32..Pipercs<tjd|dd||_||_d|_d|_tdS)N PlumbumPiperr)nameFT)super__init__pipefdemptydaemonstart)selfr=r< __class__rrr;Ss z)_iter_lines_win32..Piper.__init__cs |jSr)r<r(rA)r1rrread_from_pipe[s z/_iter_lines_win32..Piper.read_from_pipecs*t|jdD] }|j|fqdS)N)iterrEputr=)rAr6)r/queuerrrun^sz$_iter_lines_win32..Piper.run)__name__ __module__ __qualname__r;rErJ __classcell__rr/r1rIrBrPiperRs  rPinfrrTFr r!r"皙?) rfloatrrrtime get_nowait QueueEmptyr%rr4sleepjoin) r r/r1r0rPpipers last_line_tsr>piperrrOr_iter_lines_win32QsH        r\c@s(eZdZdZdddddZddZdS) ProcessExecutionErroraJRepresents the failure of a process. When the exit code of a terminated process does not match the expected result, this exception is raised by :func:`run_proc `. It contains the process' return code, stdout, and stderr, as well as the command line used to create the process (``argv``) N)hostcCs^t|||||||_||_||_||_t|trt|}t|tr't|}||_ ||_ dSr) Exceptionr;messager^r!r isinstancebytesasciirr)rAr!r rrr`r^rrrr;s   zProcessExecutionError.__init__cCsddlm}dt|j}dt|j}d||j}g}|jr1|jdt|j g}ndt|j g}d|}|d|g7}|j rO|d|j g7}|rW|d |g7}|r_|d |g7}d |S) Nr) shquote_listz |  z Return code: | zUnexpected exit code: z Command line: | z Host: | z Stdout: | z Stderr: | ) plumbum.commands.baserdrXstrr splitlinesrr!r`r r^)rArdrrcmdlinesrrr__str__s"     zProcessExecutionError.__str__r)rKrLrM__doc__r;rlrrrrr]s r]c@eZdZdZddZdS)ProcessTimedOutzRaises by :func:`run_proc ` when a ``timeout`` has been specified and it has elapsed before the process terminatedcCst|||||_dSr)r_r;r!)rAmsgr!rrrr;s zProcessTimedOut.__init__NrKrLrMrmr;rrrrro roc@rn)r%zRaises by :func:`iter_lines ` when a ``line_timeout`` has been specified and it has elapsed before the process yielded another linecCs t||||||_||_dSr)r_r;r!r")rArpr!r"rrrr; zProcessLineTimedOut.__init__Nrqrrrrr%rrr%cs eZdZdZfddZZS)CommandNotFoundzRaised by :func:`local.which ` and :func:`RemoteMachine.which ` when a command was not found in the system's ``PATH``cs t|||||_||_dSr)r:r;programpath)rArurvrBrrr;rszCommandNotFound.__init__)rKrLrMrmr;rNrrrBrrtsrtc@s6eZdZd ddZddZddZdd Zd d Zd S)MinHeaprcCst||_t|jdSr)list_itemsheapqheapify)rAitemsrrrr;s zMinHeap.__init__cCs t|jSr)lenryrDrrr__len__ zMinHeap.__len__cCst|j|dSr)rzheappushry)rAitemrrrpushsz MinHeap.pushcCst|jdSr)rzheappopryrDrrrpopsz MinHeap.popcCs |jdS)Nr)ryrDrrrpeekrz MinHeap.peekN)r)rKrLrMr;r~rrrrrrrrws   rwFcCs4t}zts|r|\}}td|t}nd}tt%tj |d\}}|t ur6 WdWdS| ||fWdn1sGwYt}|r|\}}||kr]n(| tt |durt|d|_Wdn1s~wY|sRtrWdSWdStytrYdSw)Nr)r T)rw_shutting_downrmaxrT contextlibsuppressrV_timeout_queueget SystemExitrrOSErrorr4kill _timed_outr_)waitingttk_r r time_to_killnowrrr_timeout_thread_funcsB      rPlumbumTimeoutThread)targetr9TcCs&|durt|t|fdSdSr)rrHrT)r r rrr_register_proc_timeoutsrcCs(datrttdftddSdS)NTrrR)rrrHrbgthdrXrrrr_shutdown_bg_threadss rcCsht|||\}}t|_|sd}|sd}t|ddr,||jd}||jd}t|||||S)aWaits for the given process to terminate, with the expected exit code :param proc: a running Popen-like object, with all the expected methods. :param retcode: the expected return (exit) code of the process. It defaults to 0 (the convention for success). If ``None``, the return code is ignored. It may also be a tuple (or any object that supports ``__contains__``) of expected return codes. :param timeout: the number of seconds (a ``float``) to allow the process to run, before forcefully terminating it. If ``None``, not timeout is imposed; otherwise the process is expected to terminate within that timeout value, or it will be killed and :class:`ProcessTimedOut ` will be raised :returns: A tuple of (return code, stdout, stderr) rFcustom_encodingNignore)r communicaterT _end_timerr/rrr rrrrun_proc-s    rc#s|durt}|dur t}|ttfvsJt|ddpdfdd}t||ggg} |||||D]B\} } |j|dddd| | } |dkrX| | |tj krX| d| =|turjddg} | | | <t | Vq3|turu| d| fVq3t |||gd d | DRdS) aRuns the given process (equivalent to run_proc()) and yields a tuples of (out, err) line pairs. If the exit code of the process does not match the expected one, :class:`ProcessExecutionError ` is raised. :param retcode: The expected return code of this process (defaults to 0). In order to disable exit-code validation, pass ``None``. It may also be a tuple (or any iterable) of expected exit codes. :param timeout: The maximal amount of time (in seconds) to allow the process to run. ``None`` means no timeout is imposed; otherwise, if the process hasn't terminated after that many seconds, the process will be forcefully terminated an exception will be raised :param linesize: Maximum number of characters to read from stdout/stderr at each iteration. ``-1`` (default) reads until a b'\n' is encountered. :param line_timeout: The maximal amount of time (in seconds) to allow between consecutive lines in either stream. Raise an :class:`ProcessLineTimedOut ` if the timeout has been reached. ``None`` means no timeout is imposed. :param buffer_size: Maximum number of lines to keep in the stdout/stderr buffers, in case of a ProcessExecutionError. Default is ``None``, which defaults to DEFAULT_BUFFER_SIZE (which is infinite by default). ``0`` will disable buffering completely. :param mode: Controls what the generator yields. Defaults to DEFAULT_ITER_LINES_MODE (which is BY_POSITION by default) - BY_POSITION (default): yields ``(out, err)`` line tuples, where either item may be ``None`` - BY_TYPE: yields ``(fd, line)`` tuples, where ``fd`` is 1 (stdout) or 2 (stderr) :returns: An iterator of (out, err) line tuples. Nrzutf-8cs|jddS)Nreplace)errors)r/rstrip)sencodingrrsziter_lines..)r r rrrrcss|] }d|dVqdS) N)rX).0rrrr sziter_lines..) DEFAULT_ITER_LINES_MODEDEFAULT_BUFFER_SIZE BY_POSITIONBY_TYPErrrrmathrQtupler)r r r r1r0 buffer_sizemode _iter_linesr/bufferstr6bufferr5rrr iter_linesWs2(      "rr)- __future__ratexitrrzrrTrIrrVr threadingr plumbum.librrrr7r\rrr]r_ror%AttributeErrorrtrwrrrrr?r@rrr#robjectrrrrQrrrrrrsT        4/   ! $