3
\ @ s d dgZ ddlZddlmZ ddlmZ ddlmZ ddlmZ dd lmZ dd
l m
Z
ej Z ej Z ej
Z
G d d d ejejZG d
d dZeddddejfddZeddddejddd ZdS )create_subprocess_execcreate_subprocess_shell N )events) protocols)streams)tasks) coroutine)loggerc sP e Zd ZdZ fddZdd Zdd Zdd Zd
d Zd d
Z dd Z
Z S )SubprocessStreamProtocolz0Like StreamReaderProtocol, but for a subprocess.c s< t j|d || _d | _ | _| _d | _d| _g | _d S )N)loopF) super__init___limitstdinstdoutstderr
_transport_process_exited _pipe_fds)selflimitr ) __class__ "/usr/lib64/python3.6/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 rtj| j| jd| _
| j
j| | jj d |jd}|d k rtj || d | jd| _ d S )Nr )r r r )protocolreaderr )
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 fddatar# 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* excpiper# 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 | jdkr$| 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____doc__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 ZedddZdS )Processc C s8 || _ || _|| _|j| _|j| _|j| _|j | _d S )N)r Z _protocolr& r r r Zget_pidpid)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 r: )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 dH S )zdWait until the process exit and return the process return code.
This method is a coroutine.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 |rtjd| | 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 ZdrainBrokenPipeErrorConnectionResetErrorr- )r inputrC 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& rA r
rC readr- )r r* r( streamnameoutputr 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|||| jdE d H \}}}| j E d H ||fS )Nr r! )r ) rG rH r rM r r Zgatherr& r< )r rF r r r r r r communicate s
zProcess.communicate)N)r r5 r6 r r propertyr; r r< r= r? r@ rG rH rM rN r r r r r9 k s r9 c
+ sP d krt j fdd}j|| f|||d|E d H \}} t|| S )Nc s t dS )N)r r )r r )r r r r <lambda> s z)create_subprocess_shell.<locals>.<lambda>)r r r )r get_event_loopZsubprocess_shellr9 )
cmdr r r r r kwdsprotocol_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 dS )N)r r )r r )r r r r rP s z(create_subprocess_exec.<locals>.<lambda>)r r r )r rQ Zsubprocess_execr9 ) Zprogramr r r r r argsrS rT r( r" r )r r r r s )__all__
subprocess r r r r Z
coroutinesr logr
PIPEZSTDOUTZDEVNULLZFlowControlMixinZSubprocessProtocolr r9 Z_DEFAULT_LIMITr r r r r r <module> s( X]