o
0i @ s d Z ddlZddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ej Z ej
Z
ej Z G d d
d
ej ej
ZG d d d Zdddejfd
dZdddejdddZdS ))create_subprocess_execcreate_subprocess_shell N )events) protocols)streams)tasks)loggerc sV e Zd Z fddZdd Zdd Zdd Zd d
Zd d Zd
d Z dd Z
Z S )SubprocessStreamProtocolc sH t j|d || _d | _ | _| _d | _d| _g | _| j
| _ d S )N)loopF) super__init___limitstdinstdoutstderr
_transport_process_exited _pipe_fds_loopZ
create_future
_stdin_closed)selflimitr __class__ 9/opt/alt/python310/lib64/python3.10/asyncio/subprocess.pyr
s z!SubprocessStreamProtocol.__init__c C sn | j jg}| jd ur|d| j | jd ur!|d| j | jd ur/|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 ur"tj| j| jd| _| j| | j d |d}|d urAtj| j| jd| _
| j
| | j d |d}|d urWtj || d | jd| _ d S 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 ur || d S 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 ur| | | |d u r | jd d S | j| d S |dkr0| j}n
|dkr8| j}nd }|d urL|d u rG| n|| || j v rW| 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_lostJ s,
z-SubprocessStreamProtocol.pipe_connection_lostc C s d| _ | d S )NT)r r3 r r r r process_exitede s z'SubprocessStreamProtocol.process_exitedc C s0 t | jdkr| jr| j d | _d S d S d S )Nr )lenr r r r1 r7 r r r r3 i s
z/SubprocessStreamProtocol._maybe_close_transportc C s || j u r| jS d S N)r r )r streamr r r _get_close_waitern s
z*SubprocessStreamProtocol._get_close_waiter) r
__module__ __qualname__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
t s zProcess.__init__c C s d| j j d| j dS )N)r r rA 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 d H S r: )r Z_waitr7 r r r wait s z Process.waitc C s | j | d S r: )r send_signal)r signalr r r rF s zProcess.send_signalc C | j d S r: )r terminater7 r r r rI zProcess.terminatec C rH r: )r killr7 r r r rK rJ z Process.killc
s | j }| j| |rtd| t| z
| j I d H W n tt fy> } z|r4td| | W Y d }~nd }~ww |rGtd| | 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 inputrN 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 r&|dkr dnd}td| | | I d H }| j rA|dkr8dnd}td| | | |S )Nr% r r r z%r communicate: read %sz%r communicate: close %s)
r r( r r r rL r rN readr1 )r r. r+ r; nameoutputr r r _read_stream s
zProcess._read_streamNc s |d ur | |}n| }| jd ur| d}n| }| jd ur)| d}n| }t|||I d H \}}}| I d H ||fS r- )rR rS r rW r r ZgatherrE )r rQ r r r r r r communicate s
zProcess.communicater: )r r= r> r
r# propertyrD rE rF rI rK rR rS rW rX r r r r r@ s s
r@ c sJ t fdd}j|| f|||d|I d H \}}t||S )Nc t dS Nr$ r
r r$ r r z)create_subprocess_shell..r r r )r get_running_loopZsubprocess_shellr@ ) cmdr r r r kwdsprotocol_factoryr+ r& r r$ r r s r )r r r r c
sP t fdd}j|| g|R |||d|I d H \}} t|| S )Nc rZ r[ r\ r r$ r r r] r^ z(create_subprocess_exec..r_ )r r` Zsubprocess_execr@ )
Zprogramr r r r argsrb rc r+ r& r r$ r r s r )__all__
subprocess r r r r logr PIPEZSTDOUTZDEVNULLZFlowControlMixinZSubprocessProtocolr
r@ Z_DEFAULT_LIMITr r r r r r s( bU