a
    If"                     @   sx   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 G dd dejZ	G dd	 d	ej
ZG d
d deejZdS )    N   )	protocols)
transports)loggerc                       s   e Zd Zd0 fdd	Zdd Zdd Zdd	 Zd
d Zdd Zdd Z	e
jfddZdd Zdd Zdd Zdd Zdd Zdd Zdd Zd d! Zd"d# Zd$d% Zd&d' Zd(d) Zd*d+ Zd,d- Zd.d/ Z  ZS )1BaseSubprocessTransportNc              	      s&  t  |
 d| _|| _|| _d | _d | _d | _g | _t	
 | _i | _d| _|tjkr`d | jd< |tjkrtd | jd< |tjkrd | jd< z"| jf ||||||d| W n   |    Y n0 | jj| _| j| jd< | j rt|ttfr|}n|d }td|| j | j| |	 d S )NFr   r      )argsshellstdinstdoutstderrbufsize
subprocesszprocess %r created: pid %s)super__init___closed	_protocol_loop_proc_pid_returncode_exit_waiterscollectionsdeque_pending_calls_pipes	_finishedr   PIPE_startclosepidZ_extra	get_debug
isinstancebytesstrr   debugZcreate_task_connect_pipes)selfloopprotocolr   r	   r
   r   r   r   waiterZextrakwargsZprogram	__class__ (lib/python3.9/asyncio/base_subprocess.pyr      sH    








z BaseSubprocessTransport.__init__c                 C   s  | j jg}| jr|d | jd ur6|d| j  | jd urT|d| j  n | jd urj|d n
|d | jd}|d ur|d|j  | jd}| jd	}|d ur||u r|d
|j  n6|d ur|d|j  |d ur|d|j  d	d
|S )Nclosedzpid=zreturncode=Zrunningznot startedr   zstdin=r   r   zstdout=stderr=zstdout=zstderr=z<{}> )r-   __name__r   appendr   r   r   getpipeformatjoin)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 u r(q|j  q| jd ur| jd u r| j d u r| j	 rlt
d|  z| j  W n ty   Y n0 d S )NTz$Close running child process: kill %r)r   r   valuesr5   r   r   r   Zpollr   r!   r   ZwarningkillProcessLookupError)r'   protor.   r.   r/   r   `   s$    

zBaseSubprocessTransport.closec                 C   s&   | j s"|d| t| d |   d S )Nzunclosed 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 v r| j | jS d S d S r:   )r   r5   )r'   fdr.   r.   r/   get_pipe_transport   s    
z*BaseSubprocessTransport.get_pipe_transportc                 C   s   | j d u rt 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           	   
      sr  z j } j}|jd urB| fdd|jI d H \}}| jd< |jd urv| fdd|jI d H \}}| jd< |jd ur| fdd|jI d H \}}| jd<  jd usJ |	 j
j   jD ]\}}|j	|g|R   qd  _W n\ ttfy    Y nb tyN } z*|d ur:| s:|| W Y d }~n(d }~0 0 |d urn| sn|d  d S )Nc                      s
   t  dS )Nr   )WriteSubprocessPipeProtor.   r>   r.   r/   <lambda>       z8BaseSubprocessTransport._connect_pipes.<locals>.<lambda>r   c                      s
   t  dS )Nr   ReadSubprocessPipeProtor.   r>   r.   r/   rQ      rR   r   c                      s
   t  dS )Nr   rS   r.   r>   r.   r/   rQ      rR   r   )r   r   r
   Zconnect_write_piper   r   Zconnect_read_piper   r   	call_soonr   connection_made
SystemExitKeyboardInterruptBaseException	cancelledZset_exception
set_result)	r'   r*   procr(   _r5   callbackdataexcr.   r>   r/   r&      sB    









 z&BaseSubprocessTransport._connect_pipesc                 G   s4   | j d ur| j ||f n| jj|g|R   d S r:   )r   r3   r   rU   )r'   cbr_   r.   r.   r/   _call   s    
zBaseSubprocessTransport._callc                 C   s   |  | jj|| |   d S r:   )rb   r   Zpipe_connection_lost_try_finish)r'   rJ   r`   r.   r.   r/   _pipe_connection_lost   s    z-BaseSubprocessTransport._pipe_connection_lostc                 C   s   |  | jj|| d S r:   )rb   r   Zpipe_data_received)r'   rJ   r_   r.   r.   r/   _pipe_data_received   s    z+BaseSubprocessTransport._pipe_data_receivedc                 C   s   |d usJ || j d u s$J | j | j r<td| | || _ | jjd u rV|| j_| | jj	 | 
  | jD ]}| sr|| qrd | _d S )Nz%r exited with return code %r)r   r   r!   r   r8   r   
returncoderb   r   Zprocess_exitedrc   r   rZ   r[   )r'   rf   r*   r.   r.   r/   _process_exited   s    

z'BaseSubprocessTransport._process_exitedc                    s0   | j dur| j S | j }| j| |I dH S )zdWait until the process exit and return the process return code.

        This method is a coroutine.N)r   r   Zcreate_futurer   r3   )r'   r*   r.   r.   r/   _wait   s
    

zBaseSubprocessTransport._waitc                 C   sH   | j r
J | jd u rd S tdd | j D rDd| _ | | jd  d S )Nc                 s   s   | ]}|d uo|j V  qd S r:   )disconnected).0pr.   r.   r/   	<genexpr>   s   z6BaseSubprocessTransport._try_finish.<locals>.<genexpr>T)r   r   allr   rA   rb   _call_connection_lostr>   r.   r.   r/   rc      s    

z#BaseSubprocessTransport._try_finishc                 C   s<   z"| j | W d | _d | _d | _ nd | _d | _d | _ 0 d S r:   )r   connection_lostr   r   r'   r`   r.   r.   r/   rn      s    z-BaseSubprocessTransport._call_connection_lost)NN)r2   
__module____qualname__r   r9   r   r=   r?   r@   r   warningswarnrG   rH   rI   rK   rL   rM   rO   rB   r&   rb   rd   re   rg   rh   rc   rn   __classcell__r.   r.   r,   r/   r   
   s0    +&	r   c                   @   s<   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zd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| jdS )N<z fd=z pipe=>)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 Zd Zdd ZdS )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   ZSubprocessTransportr   ZBaseProtocolrP   ProtocolrT   r.   r.   r.   r/   <module>   s    v