
W                 @   s   d  d g 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 d d
 l	 m
 Z
 e j Z e j Z e j Z Gd d   d e j e j  Z Gd d   d  Z e d d d d e j d d   Z e d d d d d d d d d e j d d    Z d S)create_subprocess_execcreate_subprocess_shell    N   )events)	protocols)streams)tasks)	coroutine)loggerc                   sd   e  Z d  Z d Z   f d d   Z d d   Z d d   Z d d	   Z d
 d   Z d d   Z	   S)SubprocessStreamProtocolz0Like StreamReaderProtocol, but for a subprocess.c                s@   t    j d |  | |  _ d  |  _ |  _ |  _ d  |  _ d  S)Nloop)super__init___limitstdinstdoutstderr
_transport)selflimitr   )	__class__ ?/afs/.cs/s/python-3.5.2/amd64_ubu14/lib/python3.5/subprocess.pyr      s    	z!SubprocessStreamProtocol.__init__c             C   s   |  j  j g } |  j d  k	 r2 | j d |  j  |  j d  k	 rU | j d |  j  |  j d  k	 rx | j d |  j  d d j |  S)Nzstdin=%rz	stdout=%rz	stderr=%rz<%s> )r   __name__r   appendr   r   join)r   infor   r   r   __repr__   s    z!SubprocessStreamProtocol.__repr__c             C   s   | |  _  | j d  } | d  k	 rU t j d |  j d |  j  |  _ |  j j |  | j d  } | d  k	 r t j d |  j d |  j  |  _ |  j j |  | j d  } | d  k	 r t j	 | d |  d d  d |  j |  _
 d  S)Nr   r   r      r   protocolreader)r   get_pipe_transportr   StreamReaderr   _loopr   Zset_transportr   StreamWriterr   )r   	transportZstdout_transportZstderr_transportZstdin_transportr   r   r   connection_made&   s"    	z(SubprocessStreamProtocol.connection_madec             C   sS   | d k r |  j  } n | d k r0 |  j } n d  } | d  k	 rO | j |  d  S)Nr   r   )r   r   Z	feed_data)r   fddatar!   r   r   r   pipe_data_received<   s    z+SubprocessStreamProtocol.pipe_data_receivedc             C   s   | d k r< |  j  } | d  k	 r+ | j   |  j |  d  S| d k rT |  j } n | d k rl |  j } n d  } | d  k r | d  k r | j   n | j |  d  S)Nr   r   r   )r   closeZconnection_lostr   r   Zfeed_eofset_exception)r   r(   excpiper!   r   r   r   pipe_connection_lostF   s    	
z-SubprocessStreamProtocol.pipe_connection_lostc             C   s   |  j  j   d  |  _  d  S)N)r   r+   )r   r   r   r   process_exitedY   s    z'SubprocessStreamProtocol.process_exited)
r   
__module____qualname____doc__r   r   r'   r*   r/   r0   r   r   )r   r   r      s   

r   c               @   s   e  Z d  Z d d   Z d d   Z e d d    Z e d d    Z d	 d
   Z	 d d   Z
 d d   Z e d d    Z e d d    Z e d d    Z e d d d   Z d S)Processc             C   sR   | |  _  | |  _ | |  _ | j |  _ | j |  _ | j |  _ | j   |  _ d  S)N)r   Z	_protocolr$   r   r   r   Zget_pidpid)r   r&   r    r   r   r   r   r   _   s    			zProcess.__init__c             C   s   d |  j  j |  j f S)Nz<%s %s>)r   r   r5   )r   r   r   r   r   h   s    zProcess.__repr__c             C   s   |  j  j   S)N)r   Zget_returncode)r   r   r   r   
returncodek   s    zProcess.returncodec             c   s   |  j  j   Ed HS)zdWait until the process exit and return the process return code.

        This method is a coroutine.N)r   _wait)r   r   r   r   waito   s    zProcess.waitc             C   s   |  j  j |  d  S)N)r   send_signal)r   signalr   r   r   r9   v   s    zProcess.send_signalc             C   s   |  j  j   d  S)N)r   	terminate)r   r   r   r   r;   y   s    zProcess.terminatec             C   s   |  j  j   d  S)N)r   kill)r   r   r   r   r<   |   s    zProcess.killc             c   s   |  j  j   } |  j j |  | r> t j d |  t |   y |  j j   Ed  HWnD t t	 f k
 r } z | r t j d |  |  WYd  d  } ~ Xn X| r t j d |   |  j j
   d  S)Nz%%r communicate: feed stdin (%s bytes)z%r communicate: stdin got %rz%r communicate: close stdin)r$   	get_debugr   writer
   debuglenZdrainBrokenPipeErrorConnectionResetErrorr+   )r   inputr?   r-   r   r   r   _feed_stdin   s    	&zProcess._feed_stdinc             C   s   d  S)Nr   )r   r   r   r   _noop   s    zProcess._noopc             c   s   |  j  j |  } | d k r* |  j } n | d k s< t  |  j } |  j j   r | d k rf d n d } t j d |  |  | j	   Ed  H} |  j j   r | d k r d n d } t j d |  |  | j
   | S)Nr   r   r   r   z%r communicate: read %sz%r communicate: close %s)r   r"   r   AssertionErrorr   r$   r=   r
   r?   readr+   )r   r(   r&   streamnameoutputr   r   r   _read_stream   s    	
zProcess._read_streamNc             c   s   | d  k	 r |  j  |  } n |  j   } |  j d  k	 rK |  j d  } n |  j   } |  j d  k	 rx |  j d  } n |  j   } t j | | | d |  j Ed  H\ } } } |  j   Ed  H| | f S)Nr   r   r   )	rD   rE   r   rK   r   r   Zgatherr$   r8   )r   rC   r   r   r   r   r   r   communicate   s    zProcess.communicate)r   r1   r2   r   r   propertyr6   r	   r8   r9   r;   r<   rD   rE   rK   rL   r   r   r   r   r4   ^   s   	r4   c       
   
   +   so    d  k r t  j       f d d   }  j | |  d | d | d | | Ed  H\ } }	 t | |	   S)Nc                  s   t  d   d   S)Nr   r   )r   r   )r   r   r   r   <lambda>   s   z)create_subprocess_shell.<locals>.<lambda>r   r   r   )r   get_event_loopZsubprocess_shellr4   )
cmdr   r   r   r   r   kwdsprotocol_factoryr&   r    r   )r   r   r   r      s    r   r   r   r   r   c            /   sr    d  k r t  j       f d d   }  j | |  | d | d | d | | Ed  H\ }	 }
 t |	 |
   S)Nc                  s   t  d   d   S)Nr   r   )r   r   )r   r   r   r   rN      s   z(create_subprocess_exec.<locals>.<lambda>r   r   r   )r   rO   Zsubprocess_execr4   )Zprogramr   r   r   r   r   argsrQ   rR   r&   r    r   )r   r   r   r      s    	)__all__
subprocess r   r   r   r   Z
coroutinesr	   logr
   PIPESTDOUTDEVNULLZFlowControlMixinZSubprocessProtocolr   r4   Z_DEFAULT_LIMITr   r   r   r   r   r   <module>   s(   			K]	