3
.�]2 � @ sj d Z ddlZddlZddlZddlZejd dkZer:eZ dZ
dZ G dd� de �Z
d d d�Zd
d d
�ZdS )zEGeneric wrapper for read-eval-print-loops, a.k.a. interactive shells
� N� z[PEXPECT_PROMPT>z[PEXPECT_PROMPT+c @ s< e Zd ZdZeedfdd�Zdd� Zd
dd �Zdd
d �Z dS )� REPLWrappera� Wrapper for a REPL.
:param cmd_or_spawn: This can either be an instance of :class:`pexpect.spawn`
in which a REPL has already been started, or a str command to start a new
REPL process.
:param str orig_prompt: The prompt to expect at first.
:param str prompt_change: A command to change the prompt to something more
unique. If this is ``None``, the prompt will not be changed. This will
be formatted with the new and continuation prompts as positional
parameters, so you can use ``{}`` style formatting to insert them into
the command.
:param str new_prompt: The more unique prompt to expect after the change.
:param str extra_init_cmd: Commands to do extra initialisation, such as
disabling pagers.
Nc C s� t |t�r tj|ddd�| _n|| _| jjrB| jjd� | jj� |d krR|| _n| j ||j
||�� || _|| _ | j � |d k r�| j
|� d S )NFzutf-8)�echo�encoding)�
isinstance�
basestring�pexpect�spawn�childr ZsetechoZ
waitnoecho�prompt�
set_prompt�format�continuation_prompt�_expect_prompt� run_command)�selfZ cmd_or_spawn� orig_prompt�
prompt_changeZ
new_promptr �extra_init_cmd� r � /usr/lib/python3.6/replwrap.py�__init__! s
zREPLWrapper.__init__c C s | j j|� | j j|� d S )N)r
�expect�sendline)r r r r r r r < s zREPLWrapper.set_prompt� c C s | j j| j| jg|d�S )N)�timeout)r
Z expect_exactr r )r r r r r r @ s zREPLWrapper._expect_promptc C s� |j � }|jd�r |jd� |s(td��g }| jj|d � x:|dd� D ]*}| j|d� |j| jj� | jj|� qJW | j|d�dkr�| jjt j
� | jdd� td| ��dj || jjg �S ) a Send a command to the REPL, wait for and return output.
:param str command: The command to send. Trailing newlines are not needed.
This should be a complete block of input that will trigger execution;
if a continuation prompt is found after sending input, :exc:`ValueError`
will be raised.
:param int timeout: How long to wait for the next prompt. -1 means the
default from the :class:`pexpect.spawn` object (default 30 seconds).
None means to wait indefinitely.
�
� zNo command was givenr r N)r z2Continuation prompt found - input was incomplete:
) �
splitlines�endswith�append�
ValueErrorr
r r Zbefore�kill�signal�SIGINT�join)r �commandr Zcmdlines�res�liner r r r D s"
zREPLWrapper.run_command���)r) r) )r) )
�__name__�
__module__� __qualname__�__doc__�PEXPECT_PROMPT�PEXPECT_CONTINUATION_PROMPTr r r r r r r r r s
r �pythonc C s t | dd�S )z>Start a Python shell and return a :class:`REPLWrapper` object.z>>> z(import sys; sys.ps1={0!r}; sys.ps2={1!r})r )r& r r r r0 g s �bashc C s� t jjt jjt�d�}tj| d|gddd�}tdd� d tdd� }tdd� d tdd� }d j ||�}t
|d
|d d �S )
z<Start a bash shell and return a :class:`REPLWrapper` object.z bashrc.shz--rcfileFzutf-8)r r N� z\[\]z%PS1='{0}' PS2='{1}' PROMPT_COMMAND=''z\$zexport PAGER=cat)r ) �os�pathr% �dirname�__file__r r r. r/ r
r )r& Zbashrcr
Zps1Zps2r r r r r1 k s )r0 )r1 )r- Zos.pathr3 r# �sysr � version_infoZPY3�strr r. r/ �objectr r0 r1 r r r r �<module> s V