o
    ÿé!^  ã                   @   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r!| j d¡ | j ¡  |d u r)|| _n|  	|| 
||¡¡ || _|| _|  ¡  |d urH|  |¡ d S 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éÿÿÿÿFc                 C   s   | j j| j| jg||dS )N)ÚtimeoutÚasync_)r   Zexpect_exactr   r   )r   r   r   r   r   r   r   @   s   ÿzREPLWrapper._expect_promptc                 C   sÚ   |  ¡ }| d¡r| d¡ |stdƒ‚|r"ddlm} || ||ƒS g }| j |d ¡ |dd… D ]}| j|d | | jj	¡ | j |¡ q2| j|ddkrc| j 
tj¡ | jdd td	| ƒ‚d || jj	g ¡S )
a0  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.
        :param bool async_: On Python 3.4, or Python 3.3 with asyncio
          installed, passing ``async_=True`` will make this return an
          :mod:`asyncio` Future, which you can yield from to get the same
          result that this method would normally give directly.
        Ú
Ú zNo command was givené   )Úrepl_run_command_asyncr   N)r   z2Continuation prompt found - input was incomplete:
)Ú
splitlinesÚendswithÚappendÚ
ValueErrorZ_asyncr!   r   r   r   ZbeforeÚkillÚsignalÚSIGINTÚjoin)r   Úcommandr   r   Zcmdlinesr!   ÚresÚliner   r   r   r   D   s*   

ÿzREPLWrapper.run_command)r   F)
Ú__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   r3   o   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   r1   r2   r   r   )r*   Zbashrcr   Zps1Zps2r   r   r   r   r4   s   s   ÿÿ)r3   )r4   )r0   Úos.pathr6   r'   Úsysr
   Úversion_infoZPY3Ústrr	   r1   r2   Úobjectr   r3   r4   r   r   r   r   Ú<module>   s    
^