3
.�]! � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZyddlm Z m
Z
W n$ e k
rt ddl m Z m
Z
Y nX ddl m
Z
mZ ddlmZ ddlmZ G dd � d e
�ZdS )
zJProvides an interface like pexpect.spawn interface using subprocess.Popen
� N)�Queue�Empty� )� SpawnBase�PY3)�EOF)� string_typesc s� e Zd Zerdjd�ZndZd � fdd� Zd Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z d dd�Z
dd� Zdd� Zdd � Z� ZS ) �
PopenSpawn�
�ascii� �� N�strictc
s� t t| �j|||||| d� tdtjtjtj||d�}
tjdkrjtj � } | j
tj O _
| |
d< tj |
d< t
|t�r�tjdkr�tj|tjdkd�}tj|f|
�| _| jj| _d | _| j� | _t� | _tj | j d
�| _ | j j d � | j j � d S ) N)�timeout�maxread�searchwindowsize�logfile�encoding� codec_errorsr )�bufsize�stdin�stderr�stdout�cwd�env�win32� startupinfoZ
creationflags�posix)r F)�targetT)!�superr �__init__�dict�
subprocess�PIPEZSTDOUT�sys�platformZ STARTUPINFOZdwFlagsZSTARTF_USESHOWWINDOWZCREATE_NEW_PROCESS_GROUP�
isinstancer �shlex�split�os�name�Popen�proc�pid�closedZ string_type�_bufr � _read_queue� threadingZThread�_read_incomingZ _read_threadZ setDaemon�start) �self�cmdr r r r r r r r �kwargsr )� __class__� �!/usr/lib/python3.6/popen_spawn.pyr s*
zPopenSpawn.__init__Fc
C s� | j }| jr8|r*||d � | _ |d |� S d| _td��|dkrH| j}n |d krTd}tj� }xntj� | |k r�|r�t|�|k r�y| jj� }W n t k
r� P Y q^X |d kr�d| _P || j
j |dd�7 }q^W |d |� ||d � }| _ | j |d� |S ) NTzEnd Of File (EOF).r g ��.AF)�final�read���)
r/ �_read_reached_eofZflag_eofr r �time�lenr0 Z
get_nowaitr Z_decoder�decode�_log)r4 �sizer �bufZt0Zincoming�rr8 r8 r9 �read_nonblocking: s0 " zPopenSpawn.read_nonblockingc C sz | j jj� }xhd}ytj|d�}W n. tk
rP } z| j|d� W Y dd}~X nX |sf| jjd� dS | jj|� qW dS )z6Run in a thread to move output from a pipe to a queue.� i r; N) r, r �filenor) r; �OSErrorrA r0 Zput)r4 rG rC �er8 r8 r9 r2 ] s zPopenSpawn._read_incomingc C s | j |� dS )zHThis is similar to send() except that there is no return value.
N)�send)r4 �sr8 r8 r9 �writen s zPopenSpawn.writec C s x|D ]}| j |� qW dS )z�This calls write() for each element in the sequence.
The sequence can be any iterable object producing strings, typically a
list of strings. This does not add line separators. There is no return
value.
N)rJ )r4 ZsequencerK r8 r8 r9 �
writeliness s
zPopenSpawn.writelinesc C sR | j |�}| j|d� | jj|dd�}tr8| jjj|�S | jjj|� t|�S dS )zbSend data to the subprocess' stdin.
Returns the number of bytes written.
rJ F)r: N) Z_coerce_send_stringrA Z_encoder�encoder r, r rL r? )r4 rK �br8 r8 r9 rJ } s
zPopenSpawn.send� c C s | j |�}|| j | j� S )z�Wraps send(), sending string ``s`` to child process, with os.linesep
automatically appended. Returns number of bytes written. )rJ �linesep)r4 rK �nr8 r8 r9 �sendline� s
zPopenSpawn.sendlinec C s8 | j j� }|dkr || _d| _nd| _| | _d| _|S )zSWait for the subprocess to finish.
Returns the exit code.
r NT)r, �waitZ
exitstatusZ signalstatusZ
terminated)r4 Zstatusr8 r8 r9 rT � s
zPopenSpawn.waitc C sT t jdkr@|tjtjgkr"tj}n |tjtjgkr:tj}ntj}tj | j
j |� dS )z�Sends a Unix signal to the subprocess.
Use constants from the :mod:`signal` module to specify which signal.
r N) r$ r% �signal�SIGINTZ CTRL_C_EVENTZSIGBREAKZCTRL_BREAK_EVENT�SIGTERMr) �killr, r- )r4 Zsigr8 r8 r9 rX � s
zPopenSpawn.killc C s | j jj� dS )z+Closes the stdin pipe from the writing end.N)r, r �close)r4 r8 r8 r9 �sendeof� s zPopenSpawn.sendeof)r r
NNNNNr )rP )�__name__�
__module__� __qualname__r rN Zcrlfr r= rE r2 rL rM rJ rS rT rX rZ �
__classcell__r8 r8 )r7 r9 r s #
r )�__doc__r) r1 r" r$ r> rU r' Zqueuer r � ImportErrorZ spawnbaser r �
exceptionsr Zutilsr r r8 r8 r8 r9 �<module> s