
    `_                         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e          Zdd	Z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                   6    e Zd ZdZeedfdZd Zd	dZ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                    t          |t                    rt          j        |dd          | _        n|| _        | j        j        r3| j                            d           | j                                         ||| _        n1| 	                    ||
                    ||                     || _        || _        |                                  ||                     |           d S d S )NFutf-8echoencoding)
isinstance
basestringpexpectspawnchildr	   setecho
waitnoechoprompt
set_promptformatcontinuation_prompt_expect_promptrun_command)selfcmd_or_spawnorig_promptprompt_change
new_promptr   extra_init_cmds          0lib/python3.11/site-packages/pexpect/replwrap.py__init__zREPLWrapper.__init__!   s     lJ// 	& |%'RRRDJJ%DJ:? 	$ Ju%%%J!!### %DKKOOK%,,Z9LMMO O O$DK#6 %^,,,,, &%    c                 n    | j                             |           | j                             |           d S )N)r   expectsendline)r   r   r   s      r   r   zREPLWrapper.set_prompt<   s4    
+&&&
M*****r    Fc                 T    | j                             | j        | j        g||          S )N)timeoutasync_)r   expect_exactr   r   )r   r&   r'   s      r   r   zREPLWrapper._expect_prompt@   s5    z&&T5M'N/6v ' G G 	Gr    c                    |                                 }|                    d          r|                    d           |st          d          |rddlm}  || ||          S g }| j                            |d                    |dd         D ]Q}|                     |           |                    | j        j	                   | j                            |           R|                     |          dk    rL| j        
                    t          j                   |                     d           t          d	|z             d                    || j        j	        gz             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
ValueError_asyncr-   r   r#   r   beforekillsignalSIGINTjoin)r   commandr&   r'   cmdlinesr-   reslines           r   r   zREPLWrapper.run_commandD   s     %%''D!! 	 OOB 	53444 	C666666))$'BBB
HQK(((QRRL 	& 	&D000JJtz()))J%%%% w//144JOOFM******R&' ( ( (xxtz011222r    )r$   F)
__name__
__module____qualname____doc__PEXPECT_PROMPTPEXPECT_CONTINUATION_PROMPTr   r   r   r    r    r   r   r      sx           +%@ $- - - -6+ + +G G G G)3 )3 )3 )3 )3 )3r    r   pythonc                 $    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   )r8   s    r   rC   rC   o   s    w)TUUUr    bashc                    t           j                            t           j                            t                    d          }t          j        | d|gdd          }t          dd         dz   t          dd         z   }t          dd         dz   t          dd         z   }d		                    ||          }t          |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pathr7   dirname__file__r   r   r@   rA   r   r   )r8   bashrcr   ps1ps2r   s         r   rE   rE   s   s    W\\"'//(33[AAFM'J#7e#*, , ,E !
y
(>!""+=
=C
%bqb
)I
58STUTVTV8W
WC<CCCMMMufm&8: : : :r    )rC   )rE   )r?   os.pathrH   r5   sysr   version_infoPY3strr   r@   rA   objectr   rC   rE   rB   r    r   <module>rU      s       



 
a J$1 \3 \3 \3 \3 \3& \3 \3 \3|V V V V: : : : : :r    