xjddlZddlZddlmZmZddlmZGddejZGddeZ Gd d e Z dS) N) GeneratorIterable)ResponseNotAcceptedc.eZdZdZdedeefdZdS) 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 streamreturnct)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)selfr s @C:\Projects\Historical\venv\Lib\site-packages\invoke/watchers.pysubmitzStreamWatcher.submit&s "!N)__name__ __module__ __qualname____doc__strrrrrrrsD: "S "Xc] " " " " " "rrc leZdZdZdededdfdZdedededeefd Zdedeeddffd 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 patternresponser Nc0||_||_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 rrs r__init__zResponder.__init__>s    rr index_attrct||}||d}tj||tj}|r!t |||t |z|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 r rrrnewmatchess rpattern_matcheszResponder.pattern_matchesOs_&j))UVVn*Wc2400  8 D*ec#hh&6 7 7 7rc#ZK|||jdD] }|jV dS)Nr)r(rr)r r _s rrzResponder.submitksD%%fdlGDD  A-      r) rrrrrrrr(rrrrrrr5sst"$'58 #8 S YsD$%?      rrcZeZdZdZdedededdffd Zdedeeddfffd ZxZS) 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 rrsentinelr Ncvt||||_d|_d|_dS)NrF)superrr- failure_indextried)r rrr- __class__s rrzFailingResponder.__init__|s7 (+++   rr ct|}|||jd}|jr1|r/d|j|j}t||rd|_|S)Nr0z(Auto-response to r"{}" failed with {!r}!T)r/rr(r-r1formatrr)r r rfailederrr2s rrzFailingResponder.submits77>>&))%%fdm_MM : +& +<CC dmC&c** *  DJr) rrrrrrrr __classcell__)r2s@rr,r,qsscd SYsD$%?rr,) r! threadingtypingrr exceptionsrlocalrrr,rrrr<s &&&&&&&&++++++*"*"*"*"*"IO*"*"*"Z9 9 9 9 9 9 9 9 x     y     r