]]# @ s d d l Z d d l Z d d l Z d d l m Z d d l m Z d d l m Z d d l m Z d d l m
Z
Gd d d e j Z Gd
d d e j
Z Gd d
d
e e j Z d S) N )compat) protocols)
transports) coroutine)loggerc sE e Z d Z d d f d d Z d d Z d d Z d d Z d
d Z d d
Z d d Z e
j r d d Z d d Z
d d Z d d Z d d Z d d Z d d Z d d Z e d d! Z d" d# Z d$ d% Z d& d' Z d( d) Z e d* d+ Z d, d- Z d. d/ Z S)0BaseSubprocessTransportNc
s t j |
d | _ | | _ | | _ d | _ d | _ d | _ g | _ t j
| _ i | _ d | _
| t j k r d | j d <| t j k r d | j d <| t j k r d | j d )
r/ __name__r appendr r r getpipejoin)r) infor r
r r0 r0 r1 __repr__9 s,
z BaseSubprocessTransport.__repr__c K s
t d S)N)NotImplementedError)r) r
r r r
r r r. r0 r0 r1 r V s z BaseSubprocessTransport._startc C s
| | _ d S)N)r )r) r+ r0 r0 r1 set_protocolY s z$BaseSubprocessTransport.set_protocolc C s | j S)N)r )r) r0 r0 r1 get_protocol\ s z$BaseSubprocessTransport.get_protocolc C s | j S)N)r )r) r0 r0 r1
is_closing_ s z"BaseSubprocessTransport.is_closingc C s | j r
d Sd | _ x3 | j j D]" } | d k r; q& | j j q& W| j d k r | j d k r | j j d k r | j j r t
j d | y | j j Wn t
k
r Yn Xd S)NTz$Close running child process: kill %r)r r valuesr8 r! r r pollr r# r warningkillProcessLookupError)r) protor0 r0 r1 r! b s
z BaseSubprocessTransport.closec C s+ | j s' t j d | t | j d S)Nzunclosed transport %r)r warningswarnResourceWarningr! )r) r0 r0 r1 __del__ s z BaseSubprocessTransport.__del__c C s | j S)N)r )r) r0 r0 r1 get_pid s z BaseSubprocessTransport.get_pidc C s | j S)N)r )r) r0 r0 r1 get_returncode s z&BaseSubprocessTransport.get_returncodec C s% | | j k r | j | j Sd Sd S)N)r r8 )r) fdr0 r0 r1 get_pipe_transport s z*BaseSubprocessTransport.get_pipe_transportc C s | j d k r t d S)N)r rD )r) r0 r0 r1 _check_proc s z#BaseSubprocessTransport._check_procc C s | j | j j | d S)N)rN r send_signal)r) signalr0 r0 r1 rO s
z#BaseSubprocessTransport.send_signalc C s | j | j j d S)N)rN r terminate)r) r0 r0 r1 rQ s
z!BaseSubprocessTransport.terminatec C s | j | j j d S)N)rN r rC )r) r0 r0 r1 rC s
z BaseSubprocessTransport.killc # s y7 j } j } | j d k r] | j f d d | j Ed H\ } } | j d <| j d k r | j f d d | j Ed H\ } } | j d <| j d k r | j f d d | j Ed H\ } } | j d <| j j j
x' j D] \ } } | j | | q
Wd _ WnK t k
r} z+ | d k rr| j
rr| j | WYd d } ~ Xn' X| d k r| j
r| j d d S)Nc s
t d S)Nr )WriteSubprocessPipeProtor0 )r) r0 r1 s z8BaseSubprocessTransport._connect_pipes..r c s
t d S)Nr )ReadSubprocessPipeProtor0 )r) r0 r1 rS s r c s
t d S)Nr )rT r0 )r) r0 r1 rS s r )r r r Zconnect_write_piper r
Zconnect_read_piper call_soonr connection_mader Exception cancelled
set_exception
set_result) r) r, procr* _r8 callbackdataexcr0 )r) r1 r( s6
z&BaseSubprocessTransport._connect_pipesc G s? | j d k r( | j j | | f n | j j | | d S)N)r r6 r rU )r) cbr^ r0 r0 r1 _call s z BaseSubprocessTransport._callc C s' | j | j j | | | j d S)N)ra r Zpipe_connection_lost _try_finish)r) rL r_ r0 r0 r1 _pipe_connection_lost s z-BaseSubprocessTransport._pipe_connection_lostc C s | j | j j | | d S)N)ra r Zpipe_data_received)r) rL r^ r0 r0 r1 _pipe_data_received s z+BaseSubprocessTransport._pipe_data_receivedc C s | j j r" t j d | | | | _ | j j d k rI | | j _ | j | j j | j
x* | j D] } | j sp | j
| qp Wd | _ d S)Nz %r exited with return code %r)r r# r r: r r
returncodera r Zprocess_exitedrb r rX rZ )r) re r, r0 r0 r1 _process_exited s
z'BaseSubprocessTransport._process_exitedc c s> | j d k r | j S| j j } | j j | | Ed HS)N)r r Z
create_futurer r6 )r) r, r0 r0 r1 _wait s
z BaseSubprocessTransport._waitc C sU | j d k r d St d d | j j D rQ d | _ | j | j d d S)Nc s s$ | ] } | d k o | j Vq d S)N) disconnected).0pr0 r0 r1 s z6BaseSubprocessTransport._try_finish..T)r allr r@ r ra _call_connection_lost)r) r0 r0 r1 rb s z#BaseSubprocessTransport._try_finishc C s7 z | j j | Wd d | _ d | _ d | _ Xd S)N)r connection_lostr r )r) r_ r0 r0 r1 rm s
z-BaseSubprocessTransport._call_connection_lost) r5
__module__ __qualname__r r; r r= r> r? r! r ZPY34rI rJ rK rM rN rO rQ rC r r( ra rc rd rf rg rb rm r0 r0 )r/ r1 r s0 ) % r c @ sX e Z d Z d d Z d d Z d d Z d d Z d d
Z d d Z d
S)rR c C s( | | _ | | _ d | _ d | _ d S)NF)r[ rL r8 rh )r) r[ rL r0 r0 r1 r s z!WriteSubprocessPipeProto.__init__c C s
| | _ d S)N)r8 )r) Z transportr0 r0 r1 rV
s z(WriteSubprocessPipeProto.connection_madec C s d | j j | j | j f S)Nz<%s fd=%s pipe=%r>)r/ r5 rL r8 )r) r0 r0 r1 r; s z!WriteSubprocessPipeProto.__repr__c C s, d | _ | j j | j | d | _ d S)NT)rh r[ rc rL )r) r_ r0 r0 r1 rn s z(WriteSubprocessPipeProto.connection_lostc C s | j j j d S)N)r[ r
pause_writing)r) r0 r0 r1 rq s z&WriteSubprocessPipeProto.pause_writingc C s | j j j d S)N)r[ r resume_writing)r) r0 r0 r1 rr s z'WriteSubprocessPipeProto.resume_writingN) r5 ro rp r rV r; rn rq rr r0 r0 r0 r1 rR s rR c @ s e Z d Z d d Z d S)rT c C s | j j | j | d S)N)r[ rd rL )r) r^ r0 r0 r1
data_received# s z%ReadSubprocessPipeProto.data_receivedN)r5 ro rp rs r0 r0 r0 r1 rT s rT )r r rF r r r Z
coroutinesr logr ZSubprocessTransportr Z BaseProtocolrR ZProtocolrT r0 r0 r0 r1 s