o Th@sbddlZddlZddlmZmZddlmZGdddejZGdddeZ Gd d d e Z dS) N) GeneratorIterable)ResponseNotAcceptedc@s&eZdZdZdedeefddZdS) StreamWatchera} A class whose subclasses may act on seen stream data from subprocesses. Subclasses must exhibit the following API; see `Responder` for a concrete example. * ``__init__`` is completely up to each subclass, though as usual, subclasses *of* subclasses should be careful to make use of `super` where appropriate. * `submit` must accept the entire current contents of the stream being watched, as a string, and may optionally return an iterable of strings (or act as a generator iterator, i.e. multiple calls to ``yield ``), which will each be written to the subprocess' standard input. .. note:: `StreamWatcher` subclasses exist in part to enable state tracking, such as detecting when a submitted password didn't work & erroring (or prompting a user, or etc). Such bookkeeping isn't easily achievable with simple callback functions. .. note:: `StreamWatcher` subclasses `threading.local` so that its instances can be used to 'watch' both subprocess stdout and stderr in separate threads. .. versionadded:: 1.0 streamreturncCst)a Act on ``stream`` data, potentially returning responses. :param str stream: All data read on this stream since the beginning of the session. :returns: An iterable of ``str`` (which may be empty). .. versionadded:: 1.0 )NotImplementedError)selfrr B/var/www/html/venv/lib/python3.10/site-packages/invoke/watchers.pysubmit&s zStreamWatcher.submitN)__name__ __module__ __qualname____doc__strrr r r r r rsrc @s`eZdZdZdededdfddZdeded edeefd d Zdedeeddffd d Z dS) Responderz A parameterizable object that submits responses to specific patterns. Commonly used to implement password auto-responds for things like ``sudo``. .. versionadded:: 1.0 patternresponserNcCs||_||_d|_dS)aY Imprint this `Responder` with necessary parameters. :param pattern: A raw string (e.g. ``r"\[sudo\] password for .*:"``) which will be turned into a regular expression. :param response: The string to submit to the subprocess' stdin when ``pattern`` is detected. rN)rrindex)r rrr r r __init__>s  zResponder.__init__r index_attrcCsBt||}||d}t||tj}|rt|||t||S)a Generic "search for pattern in stream, using index" behavior. Used here and in some subclasses that want to track multiple patterns concurrently. :param str stream: The same data passed to ``submit``. :param str pattern: The pattern to search for. :param str index_attr: The name of the index attribute to use. :returns: An iterable of string matches. .. versionadded:: 1.0 N)getattrrefindallSsetattrlen)r rrrrnewmatchesr r r pattern_matchesOs  zResponder.pattern_matchesccs$|||jdD]}|jVq dS)Nr)r!rr)r r_r r r r ks zResponder.submit) rrrrrrrr!rr r r r r r5s rcsReZdZdZdedededdffdd Zd edeeddfffd d ZZS) FailingRespondera Variant of `Responder` which is capable of detecting incorrect responses. This class adds a ``sentinel`` parameter to ``__init__``, and its ``submit`` will raise `.ResponseNotAccepted` if it detects that sentinel value in the stream. .. versionadded:: 1.0 rrsentinelrNcs$t||||_d|_d|_dS)NrF)superrr$ failure_indextried)r rrr$ __class__r r r|s zFailingResponder.__init__rcsLt|}|||jd}|jr|rd|j|j}t||r$d|_|S)Nr&z(Auto-response to r"{}" failed with {!r}!T)r%r r!r$r'formatrr)r rrfailederrr(r r r s  zFailingResponder.submit) rrrrrrrr __classcell__r r r(r r#qs (r#) r threadingtypingrr exceptionsrlocalrrr#r r r r s -<