U
e5d @ s d Z ddlZddlZddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ej
Z
ej Z ej Z G d d
d
ej
ejZG d d d Zddddejfd
dZddddejdddZdS ))create_subprocess_execcreate_subprocess_shell N )events) protocols)streams)tasks)loggerc sX e Zd ZdZ fddZdd Zdd Zdd Zd
d Zd d
Z dd Z
dd Z Z S )SubprocessStreamProtocolz0Like StreamReaderProtocol, but for a subprocess.c sH t j|d || _d | _ | _| _d | _d| _g | _| j
| _ d S )NloopF) super__init___limitstdinstdoutstderr
_transport_process_exited _pipe_fds_loopZ
create_future
_stdin_closed)selflimitr __class__ */usr/lib64/python3.8/asyncio/subprocess.pyr s z!SubprocessStreamProtocol.__init__c C sn | j jg}| jd k r&|d| j | jd k rB|d| j | jd k r^|d| j dd|S )Nzstdin=zstdout=zstderr=z<{}> )r __name__r appendr r formatjoin)r infor r r __repr__ s
z!SubprocessStreamProtocol.__repr__c C s || _ |d}|d k rDtj| j| jd| _| j| | j d |d}|d k rtj| j| jd| _
| j
| | j d |d}|d k rtj || d | jd| _ d S )Nr r r r )protocolreaderr )
r get_pipe_transportr StreamReaderr r r 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|| d S )Nr r& )r r Z feed_data)r fddatar( r r r pipe_data_receivedA s z+SubprocessStreamProtocol.pipe_data_receivedc C s |dkrN| j }|d k r | | | |d kr>| jd n | j| d S |dkr^| j}n|dkrn| j}nd }|d k r|d kr| n
|| || j kr| j
| | d S )Nr r r& ) r closeZconnection_lostr Z
set_resultZ
set_exceptionr r Zfeed_eofr remove_maybe_close_transport)r r. excpiper( r r r pipe_connection_lostK s*
z-SubprocessStreamProtocol.pipe_connection_lostc C s d| _ | d S )NT)r r3 r r r r process_exitedf s z'SubprocessStreamProtocol.process_exitedc C s( t | jdkr$| jr$| j d | _d S )Nr )lenr r r r1 r7 r r r r3 j s
z/SubprocessStreamProtocol._maybe_close_transportc C s || j kr| jS d S N)r r )r streamr r r _get_close_waitero s
z*SubprocessStreamProtocol._get_close_waiter)
r
__module__ __qualname____doc__r r$ r- r0 r6 r8 r3 r<
__classcell__r r r r r
s
r
c @ sj e Zd Zdd Zdd Zedd Zdd Zd d
Zd d Z d
d Z
dd Z dd Z dd Z
dddZdS )Processc C s8 || _ || _|| _|j| _|j| _|j| _| | _d S r: )r Z _protocolr r r r Zget_pidpid)r r, r' r r r r r u s zProcess.__init__c C s d| j j d| j dS )N)r r rB r7 r r r r$ ~ s zProcess.__repr__c C s
| j S r: )r Zget_returncoder7 r r r
returncode s zProcess.returncodec s | j I dH S )z?Wait until the process exit and return the process return code.N)r Z_waitr7 r r r wait s z Process.waitc C s | j | d S r: )r send_signal)r signalr r r rG s zProcess.send_signalc C s | j d S r: )r terminater7 r r r rI s zProcess.terminatec C s | j d S r: )r killr7 r r r rJ s z Process.killc
s | j }| j| |r,td| t| z| j I d H W n8 tt fk
rx } z|rhtd| | W 5 d }~X Y nX |rtd| | j
d S )Nz%%r communicate: feed stdin (%s bytes)z %r communicate: stdin got %rz%r communicate: close stdin) r get_debugr writer debugr9 ZdrainBrokenPipeErrorConnectionResetErrorr1 )r inputrM r4 r r r _feed_stdin s
zProcess._feed_stdinc s d S r: r r7 r r r _noop s z
Process._noopc s | j |}|dkr | j}n| j}| j rJ|dkr8dnd}td| | | I d H }| j r|dkrndnd}td| | | |S )Nr& r r r z%r communicate: read %sz%r communicate: close %s)
r r) r r r rK r rM readr1 )r r. r, r; nameoutputr r r _read_stream s
zProcess._read_streamNc s |d k r| |}n| }| jd k r2| d}n| }| jd k rP| d}n| }tj|||| jdI d H \}}}| I d H ||fS )Nr r& r ) rQ rR r rV r r Zgatherr rF )r rP r r r r r r communicate s
zProcess.communicate)N)r r= r> r r$ propertyrE rF rG rI rJ rQ rR rV rW r r r r rA t s
rA c
sb d krt ntjdtdd fdd}j|| f|||d|I d H \}} t|| S )NZThe loop argument is deprecated since Python 3.8 and scheduled for removal in Python 3.10.r&
stacklevelc s t dS Nr% r
r r% r r s z)create_subprocess_shell..r r r )r get_event_loopwarningswarnDeprecationWarningZsubprocess_shellrA )
cmdr r r r r kwdsprotocol_factoryr, r' r r% r r s$
r )r r r r r c sf d krt ntjdtdd fdd}j|| f||||d|I d H \} }
t| |
S )NrY r& rZ c s t dS r\ r] r r% r r r^ s z(create_subprocess_exec..r_ )r r` ra rb rc Zsubprocess_execrA ) Zprogramr r r r r argsre rf r, r' r r% r r s(
r )__all__
subprocessra r r r r logr PIPEZSTDOUTZDEVNULLZFlowControlMixinZSubprocessProtocolr
rA Z_DEFAULT_LIMITr r r r r r s. bV