3
\� � @ s� d dgZ ddlZddlmZ ddlmZ ddlmZ ddlmZ dd lmZ dd
l m
Z
ej Z ej Z ej
Z
G d d � d ejej�ZG d
d� d�Zeddddejfdd��Zeddddejd�dd ��ZdS )�create_subprocess_exec�create_subprocess_shell� N� )�events)� protocols)�streams)�tasks)� coroutine)�loggerc sL e Zd Z� fdd�Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z � Z
S )�SubprocessStreamProtocolc 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__� �*/usr/lib64/python3.6/asyncio/subprocess.pyr s z!SubprocessStreamProtocol.__init__c C sf | j jg}| jd k r$|jd| j � | jd k r>|jd| j � | jd k rX|jd| j � ddj|� S )Nzstdin=%rz stdout=%rz stderr=%rz<%s>� )r �__name__r �appendr r �join)r �infor r r �__repr__ s
z!SubprocessStreamProtocol.__repr__c C s� || _ |jd�}|d k rDtj| j| jd�| _| jj|� | jj d� |jd�}|d k r�tj| j| jd�| _
| j
j|� | jj d� |jd�}|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|j|� 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 |j� | j|� d S |dkr<| j}n|dkrL| j}nd }|d krt|d krj|j� n
|j|� || jkr�| jj|� | 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_lostJ s$
z-SubprocessStreamProtocol.pipe_connection_lostc C s d| _ | j� d S )NT)r r/ )r r r r �process_exiteda s z'SubprocessStreamProtocol.process_exitedc C s( t | j�dkr$| jr$| jj� d | _d S )Nr )�lenr r r r- )r r r r r/ e s
z/SubprocessStreamProtocol._maybe_close_transport) r �
__module__� __qualname__r r r) r, r2 r3 r/ �
__classcell__r r )r r r s
r c @ s~ e Zd Zdd� Zdd� Zedd� �Zedd� �Zd d
� Z d d � Z
d
d� Z edd� �Z edd� �Z
edd� �Zeddd��ZdS )�Processc C s8 || _ || _|| _|j| _|j| _|j| _|j� | _d S )N)r Z _protocolr& r r r Zget_pid�pid)r r( r" r r r r r l s zProcess.__init__c C s d| j j| jf S )Nz<%s %s>)r r r9 )r r r r r u s zProcess.__repr__c C s
| j j� S )N)r Zget_returncode)r r r r �
returncodex s zProcess.returncodec c s | j j� E d H S )N)r Z_wait)r r r r �wait| s z Process.waitc C s | j j|� d S )N)r � send_signal)r �signalr r r r< � s zProcess.send_signalc C s | j j� d S )N)r � terminate)r r r r r> � s zProcess.terminatec C s | j j� d S )N)r �kill)r r r r r? � s z Process.killc c s� | j j� }| jj|� |r,tjd| t|�� y| jj� E d H W n8 tt fk
rx } z|rhtjd| |� W Y d d }~X nX |r�tjd| � | jj
� 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 �inputrB r0 r r r � _feed_stdin� s
zProcess._feed_stdinc C s d S )Nr )r r r r �_noop� s z
Process._noopc c s� | j j|�}|dkr | j}n| j}| jj� rJ|dkr8dnd}tjd| |� |j� E d H }| jj� r�|dkrndnd}tjd| |� |j � |S )Nr! r r r z%r communicate: read %sz%r communicate: close %s)
r r$ r r r& r@ r
rB �readr- )r r* r( �stream�name�outputr r r � _read_stream� s
zProcess._read_streamNc c s� |d k r| j |�}n| j� }| jd k r2| jd�}n| j� }| jd k rP| jd�}n| j� }tj|||| jd�E d H \}}}| j� E d H ||fS )Nr r! )r ) rF rG r rL r r Zgatherr& r; )r rE r r r r r r � communicate� s
zProcess.communicate)N)r r5 r6 r r �propertyr: r r; r< r> r? rF rG rL rM r r r r r8 k s r8 c
+ sP �d krt j� �� �fdd�}�j|| f|||d�|��E 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_shellr8 )
�cmdr r r r r �kwds�protocol_factoryr( r" r )r r r r � s )r r r r r c / sT �d krt j� �� �fdd�}�j|| f|�|||d�|��E d H \} }
t| |
��S )Nc s t � �d�S )N)r r )r r )r r r r rO � s z(create_subprocess_exec.<locals>.<lambda>)r r r )r rP Zsubprocess_execr8 ) Zprogramr r r r r �argsrR rS r( r" r )r r r r � s )�__all__�
subprocess� r r r r Z
coroutinesr �logr
�PIPEZSTDOUTZDEVNULLZFlowControlMixinZSubprocessProtocolr r8 Z_DEFAULT_LIMITr r r r r r �<module> s( X]