� j f"�@s�ddlZddlZddlZddlmZddlmZddlmZddlmZddlm Z ddl mZGd d �d ej�Z Gdd�dej�ZGd d�deej�ZdS)�N�)�compat)�futures)� protocols)� transports)� coroutine)�loggercs0eZdZdd�fdd�Zdd�Zdd�Zdd �Zd d�Zej rldd �Z ndd�Zdd�Zdd�Z dd�Zdd�Zdd�Zdd�Zedd��Zdd�Zd d!�Zd"d#�Zd$d%�Zed&d'��Zd(d)�Zd*d+�Z�S),�BaseSubprocessTransportNc s�t�j| �d|_||_||_d|_d|_d|_g|_t j �|_i|_d|_ |tjkr�d|jd� ) r.�__name__r�appendrrr�get�pipe�join)r*�infor rrr/r/r0�__repr__:s, z BaseSubprocessTransport.__repr__cKs t�dS)N)�NotImplementedError)r*rrr rrrr-r/r/r0r!WszBaseSubprocessTransport._startcCs|jS)N)r)r*r/r/r0� is_closingZsz"BaseSubprocessTransport.is_closingcCs�|jr dSd|_x6|jj�D]%}|dkr>q&n|jj�q&W|jdk r�|jdkr�|jj�dkr�|jj �r�t jd|�ny|jj�Wq�t k r�Yq�XndS)NTz$Close running child process: kill %r)rr�valuesr6r"rrZpollrr$rZwarning�kill�ProcessLookupError)r*�protor/r/r0r"]s zBaseSubprocessTransport.closecCs.|js*tjd|t�|j�ndS)Nzunclosed transport %r)r�warnings�warn�ResourceWarningr")r*r/r/r0�__del__{s zBaseSubprocessTransport.__del__cCs|jS)N)r)r*r/r/r0�get_pid�szBaseSubprocessTransport.get_pidcCs|jS)N)r)r*r/r/r0�get_returncode�sz&BaseSubprocessTransport.get_returncodecCs%||jkr|j|jSdSdS)N)rr6)r*�fdr/r/r0�get_pipe_transport�sz*BaseSubprocessTransport.get_pipe_transportcCs|jdkrt��ndS)N)rr>)r*r/r/r0�_check_proc�sz#BaseSubprocessTransport._check_proccCs|j�|jj|�dS)N)rHr�send_signal)r*�signalr/r/r0rI�s z#BaseSubprocessTransport.send_signalcCs|j�|jj�dS)N)rHr� terminate)r*r/r/r0rK�s z!BaseSubprocessTransport.terminatecCs|j�|jj�dS)N)rHrr=)r*r/r/r0r=�s zBaseSubprocessTransport.killc #s�yU�j}�j}|jdk r`|j�fdd�|j�DdH\}}|�jd�sz8BaseSubprocessTransport._connect_pipes..rcs t�d�S)Nr)�ReadSubprocessPipeProtor/)r*r/r0rM�srcs t�d�S)Nr )rNr/)r*r/r0rM�sr )rrr Zconnect_write_piperrZconnect_read_piperr�AssertionError� call_soonr�connection_made� Exception� cancelledZ set_exception� set_result) r*r,�procr+�_r6�callback�data�excr/)r*r0r)�s8 #z&BaseSubprocessTransport._connect_pipescGs?|jdk r(|jj||f�n|jj||�dS)N)rr4rrP)r*�cbrXr/r/r0�_call�szBaseSubprocessTransport._callcCs'|j|jj||�|j�dS)N)r[rZpipe_connection_lost�_try_finish)r*rFrYr/r/r0�_pipe_connection_lost�sz-BaseSubprocessTransport._pipe_connection_lostcCs|j|jj||�dS)N)r[rZpipe_data_received)r*rFrXr/r/r0�_pipe_data_received�sz+BaseSubprocessTransport._pipe_data_receivedcCs�|dk st|��|jdks6t|j��|jj�r[tjd||�n||_|j|jj�|j �x-|j D]"}|j�s�|j|�q�q�Wd|_ dS)Nz%r exited with return code %r) rOrrr$rr8r[rZprocess_exitedr\rrSrT)r*� returncoder,r/r/r0�_process_exited�s z'BaseSubprocessTransport._process_exitedccsD|jdk r|jStjd|j�}|jj|�|DdHS)zdWait until the process exit and return the process return code. This method is a coroutine.Nr+)rrZFuturerrr4)r*r,r/r/r0�_wait�s zBaseSubprocessTransport._waitcCsh|jst�|jdkr#dStdd�|jj�D��rdd|_|j|jd�ndS)Ncss$|]}|dk o|jVqdS)N)�disconnected)�.0�pr/r/r0� �sz6BaseSubprocessTransport._try_finish..T)rrOr�allrr<r[�_call_connection_lost)r*r/r/r0r\�s z#BaseSubprocessTransport._try_finishcCs7z|jj|�Wdd|_d|_d|_XdS)N)r�connection_lostrr)r*rYr/r/r0rg�s z-BaseSubprocessTransport._call_connection_lost)r3� __module__�__qualname__rr9r!r;r"rZPY34rCrDrErGrHrIrKr=rr)r[r]r^r`rar\rgr/r/)r.r0r s,) % r c@sXeZdZdd�Zdd�Zdd�Zdd�Zd d �Zdd�Zd S)rLcCs(||_||_d|_d|_dS)NF)rUrFr6rb)r*rUrFr/r/r0r�s z!WriteSubprocessPipeProto.__init__cCs ||_dS)N)r6)r*Z transportr/r/r0rQsz(WriteSubprocessPipeProto.connection_madecCsd|jj|j|jfS)Nz<%s fd=%s pipe=%r>)r.r3rFr6)r*r/r/r0r9sz!WriteSubprocessPipeProto.__repr__cCs,d|_|jj|j|�d|_dS)NT)rbrUr]rF)r*rYr/r/r0rhs z(WriteSubprocessPipeProto.connection_lostcCs|jjj�dS)N)rUr� pause_writing)r*r/r/r0rksz&WriteSubprocessPipeProto.pause_writingcCs|jjj�dS)N)rUr�resume_writing)r*r/r/r0rlsz'WriteSubprocessPipeProto.resume_writingN) r3rirjrrQr9rhrkrlr/r/r/r0rL�srLc@seZdZdd�ZdS)rNcCs|jj|j|�dS)N)rUr^rF)r*rXr/r/r0� data_receivedsz%ReadSubprocessPipeProto.data_receivedN)r3rirjrmr/r/r/r0rNsrN)rrr@�rrrrZ coroutinesr�logrZSubprocessTransportr ZBaseProtocolrLZProtocolrNr/r/r/r0�s�