File: //proc/self/root/lib/python3/dist-packages/landscape/lib/__pycache__/twisted_util.cpython-38.pyc
U
���c| � @ s� d dl Z d dlmZmZ d dlmZ d dlmZmZ d dl m
Z
d dl m Z d dl
mZmZ d dlmZ G d d
� d
e�Zdd d
�ZG dd� de�Zdi dddd dddf dd�ZdS )� N)� DeferredList�Deferred)�ProcessProtocol)�Process�
ProcessReader)�reactor)�Failure)�
itervalues�
networkString)�
encode_valuesc @ s e Zd ZdZdS )� SignalErrorz3An error if the process was terminated by a signal.N)�__name__�
__module__� __qualname__�__doc__� r r �</usr/lib/python3/dist-packages/landscape/lib/twisted_util.pyr
s r Fc C s. t | d|d�}|�dd� � |�dd� � |S )N� )ZfireOnOneErrbackZ
consumeErrorsc S s dd� | D �S )Nc S s g | ] }|d �qS )r r )�.0�xr r r �
<listcomp> s z4gather_results.<locals>.<lambda>.<locals>.<listcomp>r )�rr r r �<lambda> � z gather_results.<locals>.<lambda>c S s | j jS �N)�valueZ
subFailure)�fr r r r r )r Z addCallbackZ
addErrback)Z deferredsZconsume_errors�dr r r �gather_results s �r c @ s2 e Zd ZdZd dd�Zdd� Zdd� Zd d
� ZdS ) �AllOutputProcessProtocolz<A process protocol for getting stdout, stderr and exit code.Nc C s: || _ t�� | _t�� | _| jj| _|| _|| _d| _ d S �Nr )
�deferred�io�BytesIO�outBuf�errBuf�writeZ errReceived�stdin�
line_received�
_partial_line)�selfr! r' r( r r r �__init__ s
z!AllOutputProcessProtocol.__init__c C s* | j d k r&| j�t| j �� | j�� d S r )r' Z transportr& r
Z
closeStdin)r* r r r �connectionMade% s
z'AllOutputProcessProtocol.connectionMadec C sX | j �|� | jd krd S |�d�}| j|d |d< |�� | _|D ]}| �|� qDd S )N�
r )r$ r& r( �splitr) �pop)r* �data�lines�liner r r � outReceived* s
z$AllOutputProcessProtocol.outReceivedc C st | j r| �| j � d| _ | j�� }| j�� }|j}|j}|jr^tt |||j��}| j
� |� n| j
� |||f� d S r )
r) r( r$ �getvaluer% r ZexitCode�signalr r r! Zerrback�callback)r* �reason�out�err�e�codeZfailurer r r � processEnded: s
z%AllOutputProcessProtocol.processEnded)NN)r
r r r r+ r, r3 r< r r r r r s
r r Tc
s` | g}
|
� |� t� } t| | |d�} t|�}tj| | |
|||||d�� |s\� fdd�}
|
� _| S )a�
Spawn a process using Twisted reactor.
Return a deferred which will be called with process stdout, stderr and exit
code.
@param wait_pipes: if set to False, don't wait for stdin/stdout pipes to
close when process ends.
@param line_received: an optional callback called with every line of
output from the process as parameter.
@note: compared to reactor.spawnProcess, this version does NOT require the
executable name as first element of args.
)r' r( )�args�env�path�uid�gid�usePTYc sJ � j r<� js<t� j �D ] } t| t�r,| �� | �� qi � _ t�� � dS )a� A less strict version of Process.maybeCallProcessEnded.
This behaves exactly like the original method, but in case the
process has ended already and sent us a SIGCHLD, it doesn't wait
for the stdin/stdout pipes to close, because the child process
itself might have passed them to its own child processes.
@note: Twisted 8.2 now has a processExited hook that could
be used in place of this workaround.
N) Zpipes�pidr �
isinstancer ZdoReadZ stopReadingr �maybeCallProcessEnded)�pipe�Zprocessr r rE h s
z,spawn_process.<locals>.maybeCallProcessEnded)�extendr r r r Z spawnProcessrE )�
executabler= r> r? r@ rA rB Z
wait_pipesr( r' Z list_args�resultZprotocolrE r rG r �
spawn_processI s$
�
� rK )F)r" Ztwisted.internet.deferr r Ztwisted.internet.protocolr Ztwisted.internet.processr r Ztwisted.internetr Ztwisted.python.failurer Ztwisted.python.compatr r
Zlandscape.lib.encodingr � Exceptionr r r rK r r r r �<module> s
0
�