U
if" @ sx d dl Z d dlZd dlZddlmZ ddlmZ ddlmZ G dd dejZ 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 fddZ 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 )1BaseSubprocessTransportNc
s& t |
d| _|| _|| _d | _d | _d | _g | _t
| _ i | _ d| _
|tjkr`d | j d< |tjkrtd | j d< |tjkrd | j d< z"| jf ||||||d| W n | Y nX | jj| _| j| jd< | j rt|ttfr|} n|d } td| | j | j| | d S )NFr r )argsshellstdinstdoutstderrbufsize
subprocesszprocess %r created: pid %s) super__init___closed _protocol_loop_proc_pid _returncode
_exit_waiters collectionsdeque_pending_calls_pipes _finishedr PIPE_startclosepidZ_extra get_debug
isinstancebytesstrr debugZ create_task_connect_pipes)
selfloopprotocolr r r
r r r
waiterZextrakwargsZprogram __class__ ) r- __name__r appendr r r getpipeformatjoin)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 kr(q|j q| jd k r| jd kr| j d kr| j rlt
d| z| j W n t
k
r Y nX d S )NTz$Close running child process: kill %r)r r valuesr5 r r r Zpollr r! r ZwarningkillProcessLookupError)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 kr| 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 krt 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
s` z j } j}|jd k rB| fdd|jI d H \}}| jd< |jd k rv| fdd|jI d H \}}| jd< |jd k r| fdd|jI d H \}}| jd< | j j
j D ]\}}|j|f| qd _ W nZ t t
fk
r Y n` tk
r< } z"|d k r,| s,|| W 5 d }~X Y n X |d k r\| s\|d d S )Nc s
t dS )Nr )WriteSubprocessPipeProtor. r> r. r/ z8BaseSubprocessTransport._connect_pipes..r c s
t dS )Nr ReadSubprocessPipeProtor. r> r. r/ rQ rR r c s
t dS )Nr rS r. r> r. r/ rQ rR r )r r r
Zconnect_write_piper r Zconnect_read_piper call_soonr connection_mader
SystemExitKeyboardInterrupt
BaseException cancelledZ
set_exception
set_result) r' r* procr( _r5 callbackdataexcr. r> r/ r& s@
z&BaseSubprocessTransport._connect_pipesc G s2 | j d k r | j ||f n| jj|f| 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 rtd| | || _| jjd kr2|| 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 k r| 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 krd S tdd | j D r:d| _| | jd d S )Nc s s | ]}|d k o|j V qd S r: ) disconnected).0pr. 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 5 d | _ d | _d | _X d S r: )r r r connection_lostr' r` r. r. r/ rn s
z-BaseSubprocessTransport._call_connection_lost)NN) r2
__module__ __qualname__r r9 r r= r? r@ r warningswarnrG rH rI rK rL rM rO rB r& rb rd re rg rh rc rn
__classcell__r. r. r, r/ r
s2 + & 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| jdS )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 ZProtocolrT r. r. r. r/ s v