B
    cŒ<[2  ã               @   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 )NFz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   ú/lib/python3.7/site-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 ¡ x:|dd… D ]*}| j|d | | jj¡ | j |¡ qJW | 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   é   N)r   z2Continuation prompt found - input was incomplete:
)Ú
splitlinesÚendswithÚappendÚ
ValueErrorr
   r   r   ZbeforeÚkillÚsignalÚSIGINTÚjoin)r   Úcommandr   ZcmdlinesZ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   r.   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--rcfileFzutf-8)r   r   Né   z\[\]z%PS1='{0}' PS2='{1}' PROMPT_COMMAND=''z\$zexport PAGER=cat)r   )ÚosÚpathr%   ÚdirnameÚ__file__r   r	   r,   r-   r   r   )r&   Zbashrcr
   Zps1Zps2r   r   r   r   r/   k   s    )r.   )r/   )r+   Úos.pathr1   r#   Úsysr   Úversion_infoZPY3Ústrr   r,   r-   Úobjectr   r.   r/   r   r   r   r   Ú<module>   s   V
