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)QueueEmpty ) SpawnBasePY3)EOF) string_typesc s e Zd ZerdjdZndZd fdd Zd Zd
d Zd d
Z dd Z
dd Z dd Z d ddZ
dd Zdd Zdd Z ZS )
PopenSpawn
ascii Nstrictc
s t t| j|||||| d tdtjtjtj||d}
tjdkrjtj } | j
tj O _
| |
d< tj |
d< t
|trtjdkrtj|tjdkd}tj|f|
| _| jj| _d | _| j | _t | _tj | j d
| _ | j j d | j j d S ) N)timeoutmaxreadsearchwindowsizelogfileencoding codec_errorsr )bufsizestdinstderrstdoutcwdenvwin32 startupinfoZ
creationflagsposix)r F)targetT)!superr __init__dict
subprocessPIPEZSTDOUTsysplatformZ STARTUPINFOZdwFlagsZSTARTF_USESHOWWINDOWZCREATE_NEW_PROCESS_GROUP
isinstancer shlexsplitosnamePopenprocpidclosedZ string_type_bufr _read_queue threadingZThread_read_incomingZ _read_threadZ setDaemonstart) selfcmdr 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|rt||k ry| jj }W n t k
r P Y q^X |d krd| _P || j
j |dd7 }q^W |d | ||d }| _ | j |d |S ) NTzEnd Of File (EOF).r g .AF)finalread)
r/ _read_reached_eofZflag_eofr r timelenr0 Z
get_nowaitr Z_decoderdecode_log)r4 sizer bufZt0Zincomingrr8 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 )zThis 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_encoderencoder r, r rL r? )r4 rK br8 r8 r9 rJ } s
zPopenSpawn.send c C s | j |}|| j | j S )zWraps 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 )zSends a Unix signal to the subprocess.
Use constants from the :mod:`signal` module to specify which signal.
r N) r$ r% signalSIGINTZ CTRL_C_EVENTZSIGBREAKZCTRL_BREAK_EVENTSIGTERMr) 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 s