iXddlmZddlmZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddlmZddlmZddlmZddlmZddlmZdd lmZe j,r dd lmZdd lmZGd d Zej8ddZGddej<ZGddZ GddejBZ" ddZ#GddZ$GddZ%y)) annotationsN) TracebackType)_compat) formatting)termui)utils)_find_binary_reader)ReadableBuffer)CommandcZeZdZd dZd dZd dZdddZdddZdddZddZ ddZ dd Z y ) EchoingStdinc.||_||_d|_y)NF)_input_output_paused)selfinputoutputs M/root/aria/mcps/aria-brain/venv/lib/python3.12/site-packages/click/testing.py__init__zEchoingStdin.__init__s   c.t|j|SN)getattrrrxs r __getattr__zEchoingStdin.__getattr__ st{{A&&rcT|js|jj||Sr)rrwrite)rrvs r_echozEchoingStdin._echo#s || LL  r " rcV|j|jj|Sr)r"rreadrns rr$zEchoingStdin.read)s zz$++**1-..rcV|j|jj|Sr)r"rread1r%s rr(zEchoingStdin.read1,s zz$++++A.//rcV|j|jj|Sr)r"rreadliner%s rr*zEchoingStdin.readline/s zz$++..q122rcz|jjDcgc]}|j|c}Scc}wr)r readlinesr"rs rr,zEchoingStdin.readlines2s+'+{{'<'<'>?! 1 ???s8c@tfdjDS)Nc3@K|]}j|ywr)r").0rrs r z(EchoingStdin.__iter__..6s7aDJJqM7s)iterrrs`r__iter__zEchoingStdin.__iter__5s74;;777rc,t|jSr)reprrr2s r__repr__zEchoingStdin.__repr__8sDKK  rN)r t.BinaryIOrr7returnNone)rstrr8t.Any)r!bytesr8r<))r&intr8r<)r8z list[bytes])r8zcabc.Iterator[bytes]r8r:) __name__ __module__ __qualname__rrr"r$r(r*r,r3r6rrrrs0 ' /03@8!rrc#>K|dyd|_dd|_yw)NTF)rstreams r _pause_echorG<s ~  sc<eZdZdZdfd Zdfd Zdfd ZxZS) BytesIOCopyzdPatch ``io.BytesIO`` to let the written stream be copied to another. .. versionadded:: 8.2 c0t|||_yr)superrcopy_to)rrL __class__s rrzBytesIOCopy.__init__Ls  rcVt||jjyr)rKflushrLrrMs rrOzBytesIOCopy.flushPs   rcX|jj|t||Sr)rLr rK)rbrMs rr zBytesIOCopy.writeTs$ 1w}Qr)rLz io.BytesIOr8r9r8r9)rRr r8r>)r@rArB__doc__rrOr __classcell__rMs@rrIrIFs   rrIceZdZdZddZy) StreamMixerzMixes `` and `` streams. The result is available in the ``output`` attribute. .. versionadded:: 8.2 ctj|_t|j|_t|j|_y)N)rL)ioBytesIOrrIstdoutstderrr2s rrzStreamMixer.__init__as/"$**, "-dkk"B "-dkk"B rNrS)r@rArBrTrrCrrrXrXYs CrrXcxeZdZdZdd d fdZd dZd fd Zed dZed dZ xZ S) _NamedTextIOWrapperaA :class:`~io.TextIOWrapper` with custom ``name`` and ``mode`` that does not close its underlying buffer. An optional ``original_fd`` preserves the file descriptor of the stream being replaced, so that C-level consumers that call :meth:`fileno` (``faulthandler``, ``subprocess``, ...) still work. Inspired by pytest's ``capsys``/``capfd`` split: see :doc:`/testing` for details. .. versionchanged:: 8.3.3 Added ``original_fd`` parameter and :meth:`fileno` override. r=) original_fdc Pt||fi|||_||_||_yr)rKr_name_mode _original_fd)rbuffernamemoder`kwargsrMs rrz_NamedTextIOWrapper.__init__us- *6*  'rcy)zThe buffer this object contains belongs to some other object, so prevent the default ``__del__`` implementation from closing that buffer. .. versionadded:: 8.3.2 NrCr2s rclosez_NamedTextIOWrapper.closesrcV|jdk\r |jSt| S)axReturn the file descriptor of the original stream, if one was provided at construction time. This allows C-level consumers (``faulthandler``, ``subprocess``, signal handlers, ...) to obtain a valid fd without crashing, even though the Python-level writes are redirected to an in-memory buffer. .. versionadded:: 8.3.3 r)rdrKfilenorPs rrlz_NamedTextIOWrapper.filenos,    !$$ $w~rc|jSr)rbr2s rrfz_NamedTextIOWrapper.name zzrc|jSr)rcr2s rrgz_NamedTextIOWrapper.modernr) rer7rfr:rgr:r`r>rhr;r8r9rS)r8r>r?) r@rArBrTrrjrlpropertyrfrgrUrVs@rr_r_gs~ & ( ( ( (  ( (  (   rr_ct|dr.ttjd|}||St d|d}n!t |t r|j|}tj|S)Nr$ t.IO[t.Any]z.Could not find binary reader for input stream.r) hasattrr tcast TypeError isinstancer:encoderZr[)rcharsetr!s rmake_input_streamrzsjuf  u!= > >IHII } E3  W% ::e rcxeZdZdZ d d dZed dZed dZed dZd dZ y) ResultaCHolds the captured result of an invoked CLI script. :param runner: The runner that created the result :param stdout_bytes: The standard output as bytes. :param stderr_bytes: The standard error as bytes. :param output_bytes: A mix of ``stdout_bytes`` and ``stderr_bytes``, as the user would see it in its terminal. :param return_value: The value returned from the invoked command. :param exit_code: The exit code as integer. :param exception: The exception that happened if one did. :param exc_info: Exception information (exception type, exception instance, traceback type). .. versionchanged:: 8.2 ``stderr_bytes`` no longer optional, ``output_bytes`` introduced and ``mix_stderr`` has been removed. .. versionadded:: 8.0 Added ``return_value``. Nc t||_||_||_||_||_||_||_||_yrrunner stdout_bytes stderr_bytes output_bytes return_value exit_code exceptionexc_info) rrrrrrrrrs rrzResult.__init__s@ ((((""  rc|jj|jjdj ddS)aThe terminal output as unicode string, as the user would see it. .. versionchanged:: 8.2 No longer a proxy for ``self.stdout``. Now has its own independent stream that is mixing `` and ``, in the order they were written. replace  )rdecoderryrr2s rrz Result.outputs9  '' (;(;YGOO D  rc|jj|jjdj ddS)z&The standard output as unicode string.rrr)rrrryrr2s rr\z Result.stdouts9  '' (;(;YGOO D  rc|jj|jjdj ddS)zThe standard error as unicode string. .. versionchanged:: 8.2 No longer raise an exception, always returns the `` string. rrr)rrrryrr2s rr]z Result.stderrs9  '' (;(;YGOO D  rc|jrt|jnd}dt|jd|dS)Nokay< >)rr5typer@)rexc_strs rr6zResult.__repr__s8*...$t~~&f4:&&'q 33rr)r CliRunnerrr<rr<rr<rr;rr>rzBaseException | Nonerz?tuple[type[BaseException], BaseException, TracebackType] | Noner?) r@rArBrTrrprr\r]r6rCrrr|r|s>!!! !  !  !!(!!*        4rr|ceZdZdZ d d dZd dZ d d dZej d ddZ d ddZ ej d ddZ y)raThe CLI runner provides functionality to invoke a Click command line script for unittesting purposes in a isolated environment. This only works in single-threaded systems without any concurrency as it changes the global interpreter state. :param charset: the character set for the input and output data. :param env: a dictionary with environment variables for overriding. :param echo_stdin: if this is set to `True`, then reading from `` writes to ``. This is useful for showing examples in some circumstances. Note that regular prompts will automatically echo the input. :param catch_exceptions: Whether to catch any exceptions other than ``SystemExit`` when running :meth:`~CliRunner.invoke`. .. versionchanged:: 8.2 Added the ``catch_exceptions`` parameter. .. versionchanged:: 8.2 ``mix_stderr`` parameter has been removed. NcD||_|xsi|_||_||_yr)ryenv echo_stdincatch_exceptions)rryrrrs rrzCliRunner.__init__s% 25)$ 0rc"|jxsdS)zGiven a command object it will return the default program name for it. The default is the `name` attribute or ``"root"`` if not set. root)rf)rclis rget_default_prog_namezCliRunner.get_default_prog_name's xx!6!rcVt|j}|r|j||S)z8Returns the environment overrides for invoking a script.)dictrupdate)r overridesr!s rmake_envzCliRunner.make_env.s%$((^  IIi  rc#l Kt||j}d}tj}tj}tj }t j} dt _|j|}t} dd} | |} | |} |jr:tjtjt|| jx}}t||jddxt_|jrd_t| j|jdd | t_t| j |jd d d | t_t#|ddfd }t#|ddfd }t#|dd}| d dfd }t$j&}t$j(}t$j*}t,j.}t0j.}t2j4j6|t$_|t$_|t$_|t,_|t0_ d dfd }|t2j4_i} |j9D]O\}}t:j<j?|||<| t:j<|==|t:j<|<Q| j| j | jBf|j9D]-\}}| t:j<|=|t:j<|</|t_|t_|t_|t$_|t$_|t$_|t,_|t0_| t _t2j4_y#t@$rY|t_|t_|t_|t$_|t$_|t$_|t,_|t0_| t _t2j4_wxYww)a\A context manager that sets up the isolation for invoking of a command line tool. This sets up `` with the given input data and `os.environ` with the overrides from the given dictionary. This also rebinds some internals in Click to be mocked (like the prompt functionality). This is automatically done in the :meth:`invoke` method. :param input: the input stream to put into `sys.stdin`. :param env: the environment overrides as dictionary. :param color: whether the output should contain color codes. The application can still override this explicitly. .. versionadded:: 8.2 An additional output stream is returned, which is a mix of `` and `` streams. .. versionchanged:: 8.2 Always returns the `` stream. .. versionchanged:: 8.0 `` is opened with ``errors="backslashreplace"`` instead of the default ``"strict"``. .. versionchanged:: 4.0 Added the ``color`` parameter. NPcb |jS#ttjf$rYywxYw)Nr=)rlAttributeErrorrZUnsupportedOperationrEs r _safe_filenoz)CliRunner.isolation.._safe_filenoks0 }}&"B$;$;<  s ..zr)encodingrfrgrzw)rrfrgr`zbackslashreplace)rrfrgerrorsr`c<tjj|xsd tj d}tjj|dtjj|S#t $r}t |d}~wwxYw)Nrr)sysr\r nextrstrip StopIterationEOFErrorrO)promptvale text_inputs r visible_inputz*CliRunner.isolation..visible_inputs~ JJ  V\r * (:&--f5 JJ  uBZ ( JJ   J ! (ja' (sB B BBctjj|xsddtjj t j dS#t $r}t|d}~wwxYw)Nrrr)rr\r rOrrrr)rrrs r hidden_inputz)CliRunner.isolation..hidden_inputse JJ   "~R0 1 JJ    (J'..v66  (ja' (sA!! A:* A55A:ctjjd}|rtjj |tjj |S)Nr)rstdinr$r\r rO)echochars r_getcharz%CliRunner.isolation.._getchars>99>>!$D   & JJ   Krc| S| SrrC)rFcolor default_colors rshould_strip_ansiz.CliRunner.isolation..should_strip_ansis}(((9 rch|tj}|tj}|f|||d|y)abDefault ``pdb.Pdb`` to real terminal streams during ``CliRunner`` isolation. Without this patch, ``pdb.Pdb.__init__`` inherits from ``cmd.Cmd`` which falls back to ``sys.stdin``/``sys.stdout`` when no explicit streams are provided. During isolation those are ``BytesIO``-backed wrappers, so the debugger reads from an empty buffer and writes to captured output, making interactive debugging impossible. By defaulting to ``sys.__stdin__``/``sys.__stdout__`` (the original terminal streams Python preserves regardless of redirection), debuggers can interact with the user while ``click.echo`` output is still captured normally. This covers ``pdb.set_trace()``, ``breakpoint()``, ``pdb.post_mortem()``, and debuggers that subclass ``pdb.Pdb`` (ipdb, pdbpp). Explicit ``stdin``/``stdout`` arguments are honored and not overridden. Debuggers that do not subclass ``pdb.Pdb`` (pudb, debugpy) are not covered. N) completekeyrr\)r __stdin__ __stdout__)rrrr\rh old_pdb_inits r_patched_pdb_initz.CliRunner.isolation.._patched_pdb_inits@:} ~  "-U6 MS r)rFrrr8r>r)rz str | Noner8r:)rboolr8r:)NN)rFzt.IO[t.Any] | Noner bool | Noner8r)tabNN) rzpdb.Pdbrr:rt.IO[str] | Noner\rrhr;r8r9)"rzryrrr\r]r FORCED_WIDTHrrXrrtruBinaryIOrr_ _CHUNK_SIZErGrvisible_prompt_funchidden_prompt_funcrr rrpdbPdbritemsosenvironget Exceptionr)rrrr bytes_input echo_input old_stdin old_stdout old_stderrold_forced_width stream_mixerr old_stdout_fd old_stderr_fdrrrrold_visible_prompt_funcold_hidden_prompt_funcold__getchar_funcold_should_strip_ansiold__compat_should_strip_ansirold_envkeyvaluerrrs @@@r isolationzCliRunner.isolation7sJD(t||<  II ZZ ZZ %22"$ mmC "}   %Z0 $Z0 ??'(vv Ll6I6IJ( K*"5 $,,YS"   J ??&'J "(   \\%   )   \\%%    Z   !  Z  ( ! ( Z   !  EI & 6A   #)"<"<!'!:!:"OO % 7 7(/(A(A%ww'' %2"$0!""3$5! %&*'+ # # # $# % #   #   # J- ,!iik , U!zz~~c2 =JJsO',BJJsO , && (;(;\=P=PQ Q%mmo , U=JJsO',BJJsO ,$CJ#CJ!CI)@F &(>F %/FO&;E #(EG %&6J #+CGG /%% &mmo , U=JJsO$',BJJsO ,$CJ#CJ!CI)@F &(>F %/FO&;E #(EG %&6J #+CGG sJR4 :O(O ;O(R4)O:BR4 OO(OO( O%"R4$O%%R4(R1PR1 P R1P BR11R4c &d}| |j}|j|||5} d} d} d} t|trt j |} |j d} |jd|xsd| d|} tj"j)tj*j)| dj-}| d j-}| d j-} dddt/|   | S#t$r|j|} YwxYw#t$r}tj}tjd|j}|d}|dk7r|} t|t sItj"j%t|tj"j%dd }|} Yd}~Zd}~wt&$r&}|s|} d } tj}Yd}~d}~wwxYw#tj"j)tj*j)| dj-}| d j-}| d j-}wxYw#1swYxYw) a{Invokes a command in an isolated environment. The arguments are forwarded directly to the command line script, the `extra` keyword arguments are passed to the :meth:`~clickpkg.Command.main` function of the command. This returns a :class:`Result` object. :param cli: the command to invoke :param args: the arguments to invoke. It may be given as an iterable or a string. When given as string it will be interpreted as a Unix shell command. More details at :func:`shlex.split`. :param input: the input data for `sys.stdin`. :param env: the environment overrides. :param catch_exceptions: Whether to catch any other exceptions than ``SystemExit``. If :data:`None`, the value from :class:`CliRunner` is used. :param extra: the keyword arguments to pass to :meth:`main`. :param color: whether the output should contain color codes. The application can still override this explicitly. .. versionadded:: 8.2 The result object has the ``output_bytes`` attribute with the mix of ``stdout_bytes`` and ``stderr_bytes``, as the user would see it in its terminal. .. versionchanged:: 8.2 The result object always returns the ``stderr_bytes`` stream. .. versionchanged:: 8.0 The result object has the ``return_value`` attribute with the value returned from the invoked command. .. versionchanged:: 4.0 Added the ``color`` parameter. .. versionchanged:: 3.0 Added the ``catch_exceptions`` parameter. .. versionchanged:: 3.0 The result object has the ``exc_info`` attribute with the traceback if available. N)rrrr prog_namerC)argsrzint | t.Any | Nonerrr~)rrrwr:shlexsplitpopKeyErrorrmain SystemExitrrrtrucoder>r\r rrOr]getvaluer|)rrrrrrrextrar outstreamsrrrrre_coder\r]rs rinvokezCliRunner.invoke sGj  ##44  ^^%S^ >+ 2*L.2II$${{4( <!IIk2  2'sxxVTZR9VPUV 2   "   "#A//1#A//1#A//1W+ 2Z%  G < 66s;  <  #<<> 4aff=>FQ; !I!&#.JJ$$S[1JJ$$T*F"  *'  <<>  *   "   "#A//1#A//1#A//1W+ 2+ 2st,JD$D.=A5JD+(J*D++J. H 7BGH  H #H>H H  H  A7JJJc#Ktj}tj|}tj| |tj||ddl} |j |yy#t$rYywxYw#tj||&ddl} |j |w#t$rYwwxYwwxYww)aA context manager that creates a temporary directory and changes the current working directory to it. This isolates tests that affect the contents of the CWD to prevent them from interfering with each other. :param temp_dir: Create the temporary directory under this directory. If given, the created directory is not removed when exiting. .. versionchanged:: 8.0 Added the ``temp_dir`` parameter. )dirNr)rgetcwdtempfilemkdtempchdirshutilrmtreeOSError)rtemp_dircwddtrs risolated_filesystemzCliRunner.isolated_filesystem~s iik   ( +   H HHSMMM"%   HHSMMM"%  sfACBC#A64C6 B?CBCC"B43C4 C=C?CCC)zutf-8NFT) ryr:r$cabc.Mapping[str, str | None] | Nonerrrrr8r9)rr r8r:r)rr r8zcabc.Mapping[str, str | None])NNF)r str | bytes | t.IO[t.Any] | Nonerr rrr8z8cabc.Iterator[tuple[io.BytesIO, io.BytesIO, io.BytesIO]])NNNNF)rr rzstr | cabc.Sequence[str] | Nonerr rr rrrrrr;r8r|)rzstr | os.PathLike[str] | Noner8zcabc.Iterator[str]) r@rArBrTrrr contextlibcontextmanagerrrr rCrrrrsY.48 !% 1 12 1 1  1  1"AE= &3748 S,/S,2S, S, B S,S,p152648(,o o .o 0 o 2 o & o o o  o b8<5 rr)rFzEchoingStdin | Noner8zcabc.Iterator[None])rr ryr:r8r7)& __future__rcollections.abcabccabcr rZrrrrrtypingrttypesrrrrrr r TYPE_CHECKING _typeshedr corer rrrGr[rIrX TextIOWrapperr_rzr|rrCrrrs" (??(!!D  "** & C C9"**9x +69(K4K4\XXr