U n5{i!@sdZddlZddlZddlmZddlmZddlmZm Z ddl m Z ddd Z Gd d d e ZGd d d e jZdidddfddZGddde jZdiddfddZGddde jZdidddfddZddZddZddZdddddgZdS)z Utility methods. Nwraps)BytesIO)deferprotocol)failurec CsJ|dkrddlm}t}||f|}||||ft||||S)Nr)reactor)twisted.internetr rDeferredZ spawnProcesstuple) r executableargsenvpathr protoArgsdprr:/tmp/pip-unpacked-wheel-mnhg7r0q/twisted/internet/utils.py_callProtocolWithDeferreds  rc@seZdZdZddZdS)_UnexpectedErrorOutputay Standard error data was received where it was not expected. This is a subclass of L{IOError} to preserve backward compatibility with the previous error behavior of L{getProcessOutput}. @ivar processEnded: A L{Deferred} which will fire when the process which produced the data on stderr has ended (exited and all file descriptors closed). cCst|d|||_dS)Nz got stderr: )IOError__init__ processEnded)selftextrrrrr*sz_UnexpectedErrorOutput.__init__N)__name__ __module__ __qualname____doc__rrrrrrs rc@s:eZdZdZdddZddZddZd d Zd d Zd S) _BackRelaya Trivial protocol for communicating with a process and turning its output into the result of a L{Deferred}. @ivar deferred: A L{Deferred} which will be called back with all of stdout and, if C{errortoo} is true, all of stderr as well (mixed together in one string). If C{errortoo} is false and any bytes are received over stderr, this will fire with an L{_UnexpectedErrorOutput} instance and the attribute will be set to L{None}. @ivar onProcessEnded: If C{errortoo} is false and bytes are received over stderr, this attribute will refer to a L{Deferred} which will be called back when the process ends. This C{Deferred} is also associated with the L{_UnexpectedErrorOutput} which C{deferred} fires with earlier in this case so that users can determine when the process has actually ended, in addition to knowing when bytes have been received via stderr. rcCs(||_t|_|r|j|_n|j|_dSN)deferredrserrReceivedIsGood errReceivederrReceivedIsBad)rr"errortoorrrrBs  z_BackRelay.__init__cCsF|jdk rBt|_t||j}|jt|d|_|j dSr!) r"rr onProcessEndedrerrbackrZFailure transportZloseConnection)rrerrrrrr&Js    z_BackRelay.errReceivedIsBadcCs|j|dSr!r#writerrrrrr$Rsz_BackRelay.errReceivedIsGoodcCs|j|dSr!r,r.rrr outReceivedUsz_BackRelay.outReceivedcCs8|jdk r|j|jn|jdk r4|j|dSr!)r"callbackr#getvaluer(r)rreasonrrrrXs  z_BackRelay.processEndedN)r) rrrrrr&r$r/rrrrrr /s  r cstfdd|||||S)a Spawn a process and return its output as a deferred returning a L{bytes}. @param executable: The file name to run and get the output of - the full path should be used. @param args: the command line arguments to pass to the process; a sequence of strings. The first string should B{NOT} be the executable's name. @param env: the environment variables to pass to the process; a dictionary of strings. @param path: the path to run the subprocess in - defaults to the current directory. @param reactor: the reactor to use - defaults to the default reactor @param errortoo: If true, include stderr in the result. If false, if stderr is received the returned L{Deferred} will errback with an L{IOError} instance with a C{processEnded} attribute. The C{processEnded} attribute refers to a L{Deferred} which fires when the executed process ends. cs t|dS)Nr')r )rr4rryz"getProcessOutput..)r)r rrrr r'rr4rgetProcessOutput_s r7c@seZdZddZddZdS) _ValueGettercCs ||_dSr!)r")rr"rrrr~sz_ValueGetter.__init__cCs|j|jjdSr!)r"r0valueexitCoder2rrrrsz_ValueGetter.processEndedN)rrrrrrrrrr8}sr8cCstt|||||S)z7Spawn a process and return its exit code as a Deferred.)rr8)r rrrr rrrgetProcessValuesr;c@s&eZdZdddZddZddZdS) _EverythingGetterNcCs4||_t|_t|_|jj|_|jj|_||_dSr!)r"routBuferrBufr-r/r% stdinBytes)rr"r?rrrrs   z_EverythingGetter.__init__cCs(|jdk r$|jd|j|jdS)Nr)r?r*Z writeToChildZ closeStdin)rrrrconnectionMades z _EverythingGetter.connectionMadecCsR|j}|j}|j}|j}|jr<|j|||jfn|j|||fdSr!) r=r1r>r9r:signalr"r)r0)rr3outr+ecoderrrrs  z_EverythingGetter.processEnded)N)rrrrr@rrrrrr<s r<c Cstt||||||fdS)a.Spawn a process and returns a Deferred that will be called back with its output (from stdout and stderr) and it's exit code as (out, err, code) If a signal is raised, the Deferred will errback with the stdout and stderr up to that point, along with the signal, as (out, err, signalNum) )r)rr<)r rrrr r?rrrgetProcessOutputAndValuesrEc Cs4|D]*}ztj|Wqtk r,YqXq|Sr!)warningsfiltersremove ValueError)Z passthrough addedFiltersfrrr_resetWarningFilterss rLc Os|D]\}}tj||qtjdt|}z|||}Wn8tk rrt}td||d|dYn*Xt |t j r| t|n td||SdS)a} Run the function I{f}, but with some warnings suppressed. This calls L{warnings.filterwarnings} to add warning filters before invoking I{f}. If I{f} returns a L{Deferred} then the added filters are removed once the deferred fires. Otherwise they are removed immediately. Note that the list of warning filters is a process-wide resource, so calling this function will affect all threads. @param suppressedWarnings: A list of arguments to pass to L{warnings.filterwarnings}, a sequence of (args, kwargs) 2-tuples. @param f: A callable, which may return a L{Deferred}. @param a: Positional arguments passed to I{f} @param kw: Keyword arguments passed to I{f} @return: The result of C{f(*a, **kw)} @seealso: L{twisted.python.util.runWithWarningsSuppressed} functions similarly, but doesn't handled L{Deferred}s. N) rFfilterwarningsrGlen BaseExceptionsysexc_inforLwith_traceback isinstancerr ZaddBoth) suppressedWarningsrKakwrkwargsrJresultrSrrrrunWithWarningsSuppresseds    r[cstfdd}|S)z Wrap C{f} in a callable which suppresses the indicated warnings before invoking C{f} and unsuppresses them afterwards. If f returns a Deferred, warnings will remain suppressed until the Deferred fires. cstf||Sr!)r[)rWrXrKrVrrwarningSuppressingWrappersz3suppressWarnings..warningSuppressingWrapperr)rKrVr]rr\rsuppressWarningssr^)Nr)rrRrF functoolsriorr rrZtwisted.pythonrrrrZProcessProtocolr r7r8r;r<rErLr[r^__all__rrrrs<    0  +