U
    ºâ~Z2  ã                   @   sj   d Z ddlZddlZddlZddlZejd dkZer:eZ	dZ
dZG dd„ deƒZdd	d„Zddd
„ZdS )zEGeneric wrapper for read-eval-print-loops, a.k.a. interactive shells
é    Né   z[PEXPECT_PROMPT>z[PEXPECT_PROMPT+c                   @   s<   e Zd ZdZeedfdd„Zdd„ Zddd	„Zdd
d„Z	dS )Ú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tj|ddd| _n|| _| jjrB| j d¡ | j ¡  |d krR|| _n|  	|| 
||¡¡ || _|| _|  ¡  |d k	rŒ|  |¡ d S )NFúutf-8©ÚechoÚencoding)Ú
isinstanceÚ
basestringÚpexpectÚspawnÚchildr   ZsetechoZ
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   ú2/usr/lib/python3/dist-packages/pexpect/replwrap.pyÚ__init__!   s     


ÿzREPLWrapper.__init__c                 C   s   | j  |¡ | j  |¡ d S )N)r   ZexpectÚsendline)r   r   r   r   r   r   r   <   s    zREPLWrapper.set_promptéÿÿÿÿc                 C   s   | j j| j| jg|dS )N©Útimeout)r   Zexpect_exactr   r   )r   r   r   r   r   r   @   s    ÿzREPLWrapper._expect_promptc                 C   s¾   |  ¡ }| d¡r| d¡ |s(tdƒ‚g }| j |d ¡ |dd… D ]*}| j|d | | jj¡ | j |¡ qH| j|ddkrª| j t	j
¡ | jdd td| ƒ‚d || jjg ¡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.
        Ú
Ú zNo command was givenr   é   Nr   z2Continuation prompt found - input was incomplete:
)Ú
splitlinesÚendswithÚappendÚ
ValueErrorr   r   r   ZbeforeÚkillÚsignalÚSIGINTÚjoin)r   Úcommandr   ZcmdlinesÚresÚliner   r   r   r   D   s$    

ÿzREPLWrapper.run_command)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 | ddƒ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   r2   g   s    Úbashc                 C   s€   t j t j t¡d¡}tj| d|gddd}tdd… d tdd…  }tdd… d tdd…  }d	 	||¡}t
|d
|ddS )z<Start a bash shell and return a :class:`REPLWrapper` object.z	bashrc.shz--rcfileFr   r   Né   z\[\]z%PS1='{0}' PS2='{1}' PROMPT_COMMAND=''z\$zexport PAGER=cat)r   )ÚosÚpathr(   ÚdirnameÚ__file__r
   r   r0   r1   r   r   )r)   Zbashrcr   Zps1Zps2r   r   r   r   r3   k   s    ÿÿ)r2   )r3   )r/   Úos.pathr5   r&   Úsysr
   Úversion_infoZPY3Ústrr	   r0   r1   Úobjectr   r2   r3   r   r   r   r   Ú<module>   s   V
