a
XC?h�" � @ sx d dl Z d dlZd dlZddlmZ ddlmZ ddlmZ G dd� dej�Z G dd � d ej
�Z G d
d � d e ej �Z
dS ) � N� )� protocols)�
transports)�loggerc s� e Zd Zd0� fdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z e
j fdd�Z dd� Z
dd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Z� Z S )1�BaseSubprocessTransportNc
s& t � �|
� d| _|| _|| _d | _d | _d | _g | _t �
� | _ i | _ d| _
|tjkr`d | j d< |tjkrtd | j d< |tjkr�d | j d< z"| jf ||||||d�| �� W n | �� � Y n0 | jj| _| j| jd< | j�� �rt|ttf�r�|} n|d } t�d| | j� | j�| �| �� d S )NFr r � )�args�shell�stdin�stdout�stderr�bufsize�
subprocesszprocess %r created: pid %s) �super�__init__�_closed� _protocol�_loop�_proc�_pid� _returncode�
_exit_waiters� collections�deque�_pending_calls�_pipes� _finishedr �PIPE�_start�close�pidZ_extra� get_debug�
isinstance�bytes�strr �debugZ create_task�_connect_pipes)
�self�loop�protocolr r r
r r r
�waiterZextra�kwargsZprogram�� __class__� �� ) r- �__name__r �appendr r r �get�pipe�format�join)r' �infor
r r r. r. r/ �__repr__7 s,
z BaseSubprocessTransport.__repr__c K s t �d S �N)�NotImplementedError)r' r r r
r r r
r+ r. r. r/ r T s z BaseSubprocessTransport._startc C s
|| _ d S r: �r )r' r) r. r. r/ � set_protocolW s z$BaseSubprocessTransport.set_protocolc C s | j S r: r< �r' r. r. r/ � get_protocolZ s z$BaseSubprocessTransport.get_protocolc C s | j S r: )r r> r. r. r/ �
is_closing] s z"BaseSubprocessTransport.is_closingc C s� | j r
d S d| _ | j�� D ]}|d u r(q|j�� q| jd ur�| jd u r�| j�� d u r�| j� � rlt
� d| � z| j� � W n t
y� Y n0 d S )NTz$Close running child process: kill %r)r r �valuesr5 r r r Zpollr r! r Zwarning�kill�ProcessLookupError)r' �protor. r. r/ r ` s$
� �
z BaseSubprocessTransport.closec C s& | j s"|d| ��t| d� | �� d S )Nzunclosed transport )�source)r �ResourceWarningr )r' Z_warnr. r. r/ �__del__{ s z BaseSubprocessTransport.__del__c C s | j S r: )r r> r. r. r/ �get_pid� s z BaseSubprocessTransport.get_pidc C s | j S r: )r r> r. r. r/ �get_returncode� s z&BaseSubprocessTransport.get_returncodec C s || j v r| j | jS d S d S r: )r r5 )r' �fdr. r. r/ �get_pipe_transport� s
z*BaseSubprocessTransport.get_pipe_transportc C s | j d u rt� �d S r: )r rC r> r. r. r/ � _check_proc� s
z#BaseSubprocessTransport._check_procc C s | � � | j�|� d S r: )rL r � send_signal)r' �signalr. r. r/ rM � s z#BaseSubprocessTransport.send_signalc C s | � � | j�� d S r: )rL r � terminater> r. r. r/ rO � s z!BaseSubprocessTransport.terminatec C s | � � | j�� d S r: )rL r rB r> r. r. r/ rB � s z BaseSubprocessTransport.killc
� sb z� j }� j}|jd urB|�� fdd�|j�I d H \}}|� jd< |jd urv|�� fdd�|j�I d H \}}|� jd< |jd ur�|�� fdd�|j�I d H \}}|� jd< |�� j j
� � � j D ]\}}|j|g|�R � q�d � _ W nZ t t
fy� � Y nb t�y> } z*|d u�r*|�� �s*|�|� W Y d }~n(d }~0 0 |d u�r^|�� �s^|�d � d S )Nc s
t � d�S )Nr )�WriteSubprocessPipeProtor. r> r. r/ �� � z8BaseSubprocessTransport._connect_pipes..r c s
t � d�S )Nr ��ReadSubprocessPipeProtor. r> r. r/ rQ � rR r c s
t � d�S )Nr rS r. r> r. r/ rQ � rR r )r r r
Zconnect_write_piper r Zconnect_read_piper � call_soonr �connection_mader �
SystemExit�KeyboardInterrupt�
BaseException� cancelledZ
set_exception�
set_result) r' r* �procr( �_r5 �callback�data�excr. r> r/ r& � s@
�
�
�
z&BaseSubprocessTransport._connect_pipesc G s4 | j d ur | j �||f� n| jj|g|�R � d S r: )r r3 r rU )r' �cbr_ r. r. r/ �_call� s
z BaseSubprocessTransport._callc C s | � | jj||� | �� d S r: )rb r Zpipe_connection_lost� _try_finish)r' rJ r` r. r. r/ �_pipe_connection_lost� s z-BaseSubprocessTransport._pipe_connection_lostc C s | � | jj||� d S r: )rb r Zpipe_data_received)r' rJ r_ r. r. r/ �_pipe_data_received� s z+BaseSubprocessTransport._pipe_data_receivedc C sp | j �� rt�d| |� || _| jjd u r2|| j_| �| jj � | �
� | j D ]}|� � sN|�
|� qNd | _ d S )Nz %r exited with return code %r)r r! r r8 r r �
returncoderb r Zprocess_exitedrc r rZ r[ )r' rf r* r. r. r/ �_process_exited� s
z'BaseSubprocessTransport._process_exitedc � s0 | j d ur| j S | j�� }| j�|� |I d H S r: )r r Z
create_futurer r3 )r' r* r. r. r/ �_wait� s
z BaseSubprocessTransport._waitc C s> | j d u rd S tdd� | j�� D ��r:d| _| �| jd � d S )Nc s s | ]}|d uo|j V qd S r: )� disconnected)�.0�pr. r. r/ � � s �z6BaseSubprocessTransport._try_finish..T)r �allr rA r rb �_call_connection_lostr> r. r. r/ rc � s
�z#BaseSubprocessTransport._try_finishc C s< z"| j �|� W d | _d | _d | _ nd | _d | _d | _ 0 d S r: )r �connection_lostr r �r' r` r. r. r/ rn � s �z-BaseSubprocessTransport._call_connection_lost)NN) r2 �
__module__� __qualname__r r9 r r= r? r@ r �warnings�warnrG rH rI rK rL rM rO rB r& rb rd re rg rh rc rn �
__classcell__r. r. r, r/ r
s0 �+ & r c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )rP c C s || _ || _d | _d| _d S )NF)r\ rJ r5 ri )r' r\ rJ r. r. r/ r s z!WriteSubprocessPipeProto.__init__c C s
|| _ d S r: )r5 )r' Z transportr. r. r/ rV s z(WriteSubprocessPipeProto.connection_madec C s d| j j� d| j� d| j�d�S )N�)r- r2 rJ r5 r> r. r. r/ r9
s z!WriteSubprocessPipeProto.__repr__c C s d| _ | j�| j|� d | _d S )NT)ri r\ rd rJ rp r. r. r/ ro
s z(WriteSubprocessPipeProto.connection_lostc C s | j j�� d S r: )r\ r �
pause_writingr> r. r. r/ rx s z&WriteSubprocessPipeProto.pause_writingc C s | j j�� d S r: )r\ r �resume_writingr> r. r. r/ ry s z'WriteSubprocessPipeProto.resume_writingN) r2 rq rr r rV r9 ro rx ry r. r. r. r/ rP � s rP c @ s e Zd Zdd� ZdS )rT c C s | j �| j|� d S r: )r\ re rJ )r' r_ r. r. r/ �
data_received s z%ReadSubprocessPipeProto.data_receivedN)r2 rq rr rz r. r. r. r/ rT s rT )r r rs � r r �logr ZSubprocessTransportr Z BaseProtocolrP �ProtocolrT r. r. r. r/ � s v �