B
� fV � @ s� d Z ddlZddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ej Z ej
Z
ej Z G d d
� d
ej ej
�ZG d d � d �Zddddejfd
d�Zddddejd�dd�ZdS ))�create_subprocess_exec�create_subprocess_shell� N� )�events)� protocols)�streams)�tasks)�loggerc sP e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
� Z S )�SubprocessStreamProtocolz0Like StreamReaderProtocol, but for a subprocess.c s< t � j|d� || _d | _ | _| _d | _d| _g | _d S )N)�loopF) �super�__init__�_limit�stdin�stdout�stderr�
_transport�_process_exited� _pipe_fds)�self�limitr )� __class__� �7/opt/alt/python37/lib64/python3.7/asyncio/subprocess.pyr
s z!SubprocessStreamProtocol.__init__c C sn | j jg}| jd k r&|�d| j��� | jd k rB|�d| j��� | jd k r^|�d| j��� d�d�|��S )Nzstdin=zstdout=zstderr=z<{}>� )r �__name__r �appendr r �format�join)r �infor r r �__repr__ s
z!SubprocessStreamProtocol.__repr__c C s� || _ |�d�}|d k rDtj| j| jd�| _| j�|� | j� d� |�d�}|d k r�tj| j| jd�| _
| j
�|� | j� d� |�d�}|d k r�tj || d | jd�| _ d S )Nr )r r � r )�protocol�readerr )
r �get_pipe_transportr � StreamReaderr �_loopr Z
set_transportr r r � StreamWriterr )r � transportZstdout_transportZstderr_transportZstdin_transportr r r �connection_made' s&
z(SubprocessStreamProtocol.connection_madec C s: |dkr| j }n|dkr | j}nd }|d k r6|�|� d S )Nr r! )r r Z feed_data)r �fd�datar# r r r �pipe_data_received? s z+SubprocessStreamProtocol.pipe_data_receivedc C s� |dkr,| j }|d k r |�� | �|� d S |dkr<| j}n|dkrL| j}nd }|d k rt|d krj|�� n
|�|� || jkr�| j�|� | � � d S )Nr r r! )
r �closeZconnection_lostr r Zfeed_eofZ
set_exceptionr �remove�_maybe_close_transport)r r* �exc�piper# r r r �pipe_connection_lostI s$
z-SubprocessStreamProtocol.pipe_connection_lostc C s d| _ | �� d S )NT)r r/ )r r r r �process_exited` s z'SubprocessStreamProtocol.process_exitedc C s( t | j�dkr$| jr$| j�� d | _d S )Nr )�lenr r r r- )r r r r r/ d s
z/SubprocessStreamProtocol._maybe_close_transport) r �
__module__� __qualname__�__doc__r
r r) r, r2 r3 r/ �
__classcell__r r )r r r
s
r
c @ sj e Zd Zdd� Zdd� Zedd� �Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dd� Z
ddd�ZdS )�Processc C s8 || _ || _|| _|j| _|j| _|j| _|�� | _d S )N)r Z _protocolr& r r r Zget_pid�pid)r r( r" r r r r r
k s zProcess.__init__c C s d| j j� d| j� d�S )N�<r �>)r r r: )r r r r r t s zProcess.__repr__c C s
| j �� S )N)r Zget_returncode)r r r r �
returncodew s zProcess.returncodec � s | j �� I dH S )z?Wait until the process exit and return the process return code.N)r Z_wait)r r r r �wait{ s z Process.waitc C s | j �|� d S )N)r � send_signal)r �signalr r r r? s zProcess.send_signalc C s | j �� d S )N)r � terminate)r r r r rA � s zProcess.terminatec C s | j �� d S )N)r �kill)r r r r rB � s z Process.killc
� s� | j �� }| j�|� |r,t�d| t|�� y| j�� I d H W n8 tt fk
rx } z|rht�d| |� W d d }~X Y nX |r�t�d| � | j�
� d S )Nz%%r communicate: feed stdin (%s bytes)z %r communicate: stdin got %rz%r communicate: close stdin) r& � get_debugr �writer �debugr4 Zdrain�BrokenPipeError�ConnectionResetErrorr- )r �inputrE r0 r r r � _feed_stdin� s
zProcess._feed_stdinc � s d S )Nr )r r r r �_noop� s z
Process._noopc � s� | j �|�}|dkr | j}n|dks(t�| j}| j�� rV|dkrDdnd}t�d| |� |� � I d H }| j�� r�|dkrzdnd}t�d| |� |�
� |S )Nr! r r r z%r communicate: read %sz%r communicate: close %s) r r$ r �AssertionErrorr r& rC r rE �readr- )r r* r( �stream�name�outputr r r � _read_stream� s
zProcess._read_streamNc � s� |d k r| � |�}n| �� }| jd k r2| �d�}n| �� }| jd k rP| �d�}n| �� }tj|||| jd�I d H \}}}| �� I d H ||fS )Nr r! )r ) rI rJ r rP r r Zgatherr& r> )r rH r r r r r r � communicate� s
zProcess.communicate)N)r r5 r6 r
r �propertyr= r> r? rA rB rI rJ rP rQ r r r r r9 j s r9 c
� sP �d krt �� �� �fdd�}�j|| f|||d�|��I d H \}} t|| ��S )Nc s t � �d�S )N)r r )r
r )r r r r �<lambda>� s z)create_subprocess_shell.<locals>.<lambda>)r r r )r �get_event_loopZsubprocess_shellr9 )
�cmdr r r r r �kwds�protocol_factoryr( r" r )r r r r � s r )r r r r r c � sT �d krt �� �� �fdd�}�j|| f|�|||d�|��I d H \} }
t| |
��S )Nc s t � �d�S )N)r r )r
r )r r r r rS � s z(create_subprocess_exec.<locals>.<lambda>)r r r )r rT Zsubprocess_execr9 ) Zprogramr r r r r �argsrV rW r( r" r )r r r r � s r )�__all__�
subprocess� r r r r �logr �PIPEZSTDOUTZDEVNULLZFlowControlMixinZSubprocessProtocolr
r9 Z_DEFAULT_LIMITr r r r r r �<module> s" XV