o  jud@sPddlmZddlmZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddlmZddlmZddlmZddlmZddlmZdd lmZe jr_dd lmZdd lmZe jd ZGd ddZejd(ddZGdddZ Gdddej!Z"GdddZ#Gdddej$Z%d)d"d#Z&Gd$d%d%Z'Gd&d'd'Z(dS)*) annotationsN) TracebackType)_compat) formatting)termui)utils)_find_binary_reader)ReadableBuffer)Command)sysfdc@sleZdZd#ddZd$d d Zd%ddZd&d'ddZd&d'ddZd&d'ddZd(ddZ d)ddZ d*d d!Z d"S)+ EchoingStdininput t.BinaryIOoutputreturnNonecCs||_||_d|_dS)NF)_input_output_paused)selfrrr@/home/kuhnn/.local/lib/python3.10/site-packages/click/testing.py__init__ zEchoingStdin.__init__xstrt.AnycCs t|j|SN)getattrr)rrrrr __getattr__"s zEchoingStdin.__getattr__rvbytescCs|js |j||Sr)rrwrite)rr"rrr_echo%s zEchoingStdin._echonintcC||j|Sr)r%rreadrr'rrrr*+zEchoingStdin.readcCr)r)r%rread1r+rrrr-.r,zEchoingStdin.read1cCr)r)r%rreadliner+rrrr.1r,zEchoingStdin.readline list[bytes]csfddjDS)Ncsg|]}|qSrr%.0rrrr 5sz*EchoingStdin.readlines..)r readlinesr3rr3rr54zEchoingStdin.readlinescabc.Iterator[bytes]cstfddjDS)Nc3s|]}|VqdSrr0r1r3rr 8sz(EchoingStdin.__iter__..)iterrr3rr3r__iter__7r6zEchoingStdin.__iter__cCs t|jSr)reprrr3rrr__repr__:s zEchoingStdin.__repr__N)rrrrrr)rrrr)r"r#rr#)r&)r'r(rr#)rr/)rr7rr) __name__ __module__ __qualname__rr!r%r*r-r.r5r:r<rrrrrs        rstreamEchoingStdin | Nonercabc.Iterator[None]ccs*|dur dVdSd|_dVd|_dS)NTF)r)rArrr _pause_echo>s   rDc@s.eZdZdZdddZddd Zdd d Zd S) _FDCaptureaVRedirect a file descriptor to a temporary file for capture. Saves the current target of *targetfd* via :func:`os.dup`, then redirects it to a temporary file via :func:`os.dup2`. On :meth:`stop`, restores the original ``fd`` and returns the captured bytes. Inspired by Pytest's ``FDCapture``. .. versionadded:: 8.4.0 targetfdr(rrcCs||_d|_d|_dSNr&) _targetfdsaved_fd_tmpfile)rrFrrrrSrz_FDCapture.__init__cCs4t|j|_tjdd|_t|j|jdS)Nr) buffering) osduprHrItempfile TemporaryFilerJdup2filenor3rrrstartXsz_FDCapture.startr#cCs^|jdus Jdt|j|jt|jd|_|jd|j}|jd|_|S)Nz!_FDCapture.start() was not calledr&r)rJrLrPrIrHcloseseekr*)rdatarrrstop]s    z_FDCapture.stopN)rFr(rrrr)rr#)r>r?r@__doc__rrRrVrrrrrEHs  rEcs>eZdZdZdfdd Zdfdd Zdfd d ZZS) BytesIOCopyzdPatch ``io.BytesIO`` to let the written stream be copied to another. .. versionadded:: 8.2 copy_to io.BytesIOrrcst||_dSr)superrrZ)rrZ __class__rrros  zBytesIOCopy.__init__cst|jdSr)r\flushrZr3r]rrr_ss zBytesIOCopy.flushbr r(cs|j|t|Sr)rZr$r\)rr`r]rrr$ws  zBytesIOCopy.write)rZr[rrrW)r`r rr()r>r?r@rXrr_r$ __classcell__rrr]rrYis rYc@seZdZdZdddZdS) StreamMixerzMixes `` and `` streams. The result is available in the ``output`` attribute. .. versionadded:: 8.2 rrcCs*t|_t|jd|_t|jd|_dS)N)rZ)ioBytesIOrrYstdoutstderrr3rrrrs zStreamMixer.__init__NrW)r>r?r@rXrrrrrrb|srbcsVeZdZdZdfd d Zdd dZdfdd ZedddZedddZ Z S)_NamedTextIOWrapperaA :class:`~io.TextIOWrapper` with custom ``name`` and ``mode`` that does not close its underlying buffer. When ``CliRunner`` runs in ``fd`` mode, ``_original_fd`` is patched to point at the saved (pre-redirection) ``fd``, so C-level consumers that call :meth:`fileno` (like ``faulthandler`` or ``subprocess``) keep working. In the default ``sys`` mode ``_original_fd`` stays at ``-1`` and :meth:`fileno` raises :exc:`io.UnsupportedOperation`, matching the pre-``8.3.3`` behavior. bufferrnamermodekwargsrrrc s*tj|fi|||_||_d|_dSrG)r\r_name_mode _original_fd)rrhrirjrkr]rrrs z_NamedTextIOWrapper.__init__cCsdS)zThe buffer this object contains belongs to some other object, so prevent the default ``__del__`` implementation from closing that buffer. .. versionadded:: 8.3.2 Nrr3rrrrSsz_NamedTextIOWrapper.closer(cs|jdkr|jStS)zReturn the file descriptor of the saved original stream when ``CliRunner`` runs in ``fd`` mode. Otherwise delegate to :class:`~io.TextIOWrapper`, which raises :exc:`io.UnsupportedOperation` for a ``BytesIO``-backed buffer. r)rnr\rQr3r]rrrQs  z_NamedTextIOWrapper.filenocC|jSr)rlr3rrrriz_NamedTextIOWrapper.namecCror)rmr3rrrrjrpz_NamedTextIOWrapper.mode) rhrrirrjrrkrrrrW)rr(r=) r>r?r@rXrrSrQpropertyrirjrarrr]rrgs  rgr str | bytes | t.IO[t.Any] | NonecharsetrrcCsZt|drttd|}|dur|Std|durd}n t|tr(||}t |S)Nr*z t.IO[t.Any]z.Could not find binary reader for input stream.) hasattrr tcast TypeError isinstancerencodercrd)rrsr"rrrmake_input_streams    r{c@sReZdZdZ ddddZedddZedddZedddZdddZ dS) 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``. Nrunner CliRunner stdout_bytesr# stderr_bytes output_bytes return_valuer exit_coder( exceptionBaseException | Noneexc_info?tuple[type[BaseException], BaseException, TracebackType] | Nonec Cs4||_||_||_||_||_||_||_||_dSrr}rrrrrrr) rr}rrrrrrrrrrrs  zResult.__init__rrcC|j|jjdddS)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  )rdecoder}rsrr3rrrrsz Result.outputcCr)z&The standard output as unicode string.rrr)rrr}rsrr3rrrresz Result.stdoutcCr)zThe standard error as unicode string. .. versionchanged:: 8.2 No longer raise an exception, always returns the `` string. rrr)rrr}rsrr3rrrrfsz Result.stderrcCs,|jrt|jnd}dt|jd|dS)Nokay< >)rr;typer>)rexc_strrrrr<szResult.__repr__r)r}r~rr#rr#rr#rrrr(rrrrr=) r>r?r@rXrrqrrerfr<rrrrr|s     r|c@seZdZdZ     d/d0ddZd1ddZ d2d3ddZej   d4d5d d!Z     d6d7d(d)Z ej d2d8d-d.Z dS)9r~aThe 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`. :param capture: Selects the output capture strategy. ``sys`` (default) captures Python-level writes only and leaves :meth:`sys.stdout.fileno` raising :exc:`io.UnsupportedOperation`, so user code that calls :func:`os.dup2` on ``sys.stdout.fileno()`` cannot clobber the host runner's stdout. ``fd`` redirects file descriptors ``1`` and ``2`` via :func:`os.dup2` to a temporary file, also catching output from stale stream references, C extensions, and subprocesses. ``fd`` is not supported on Windows. .. versionchanged:: 8.4.0 Added the ``capture`` parameter. The default ``sys`` mode no longer exposes the original fd through :meth:`fileno`, reverting the change introduced in ``8.3.3`` that broke Pytest's ``fd``-level capture teardown. Use ``capture="fd"`` to restore that behavior with proper isolation. :issue:`3384` .. versionchanged:: 8.2 Added the ``catch_exceptions`` parameter. .. versionchanged:: 8.2 ``mix_stderr`` parameter has been removed. utf-8NFTr rsrenv$cabc.Mapping[str, str | None] | None echo_stdinboolcatch_exceptionscapture CaptureModerrcCs`|dvr td|d|dkrtjdkrtd|d||_|p#i|_||_||_||_dS)N>r r zcapture=z) is not valid. Choose from 'sys' or 'fd'.r win32z( is not supported on Windows. Use 'sys'.) ValueErrorr platformrsrrrr)rrsrrrrrrrrDs    zCliRunner.__init__clir cCs |jpdS)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)ri)rrrrrget_default_prog_nameZs zCliRunner.get_default_prog_name overridescabc.Mapping[str, str | None]cCst|j}|r |||S)z8Returns the environment overrides for invoking a script.)dictrupdate)rrr"rrrmake_envas  zCliRunner.make_envrrrcolor8cabc.Iterator[tuple[io.BytesIO, io.BytesIO, io.BytesIO]]c#st||j}d}tj}tj}tj}tj} dt_||}t } |j r0t t j t|| j}}t||jdddt_|j rBd_t| j|jdddt_t| j|jd dd d t_t|d*d+fdd } t|d*d+fdd } t|d,dd} | d-d.fdd }tj}tj}tj}tj}tj}tjj| t_| t_| t_|t_|t_   d/d0fd(d) }|tj_i}zy|D]%\}}tj|||<|durztj|=Wqt yYqw|tj|<q| j| j| j!fVW|D]\}}|dur ztj|=Wqt y Yqw|tj|<q|t_|t_|t_|t_|t_|t_|t_|t_| t_tj_dS|D]"\}}|durRztj|=Wq6t yQYq6w|tj|<q6|t_|t_|t_|t_|t_|t_|t_|t_| t_tj_w)1a\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. NPzr)encodingrirjrzwzbackslashreplace)rrirjerrorsprompt str | Nonerrc sdtj|pdz td}Wnty!}zt|d}~wwtj|dtj|S)Nrr)r rer$nextrstrip StopIterationEOFErrorr_)rvale text_inputrr visible_inputs z*CliRunner.isolation..visible_inputc sRtj|pddtjztdWSty(}zt|d}~ww)Nrrr)r rer$r_rrrr)rrrrr hidden_inputs z)CliRunner.isolation..hidden_inputechorcSs*tjd}|rtj|tj|S)Nr)r stdinr*rer$r_)rcharrrr_getchars   z%CliRunner.isolation.._getcharrAt.IO[t.Any] | Noner bool | Nonecs|dur S| Srr)rAr) default_colorrrshould_strip_ansisz.CliRunner.isolation..should_strip_ansitabrpdb.Pdb completekeyrt.IO[str] | Nonererkrrcs8|durtj}|durtj}|f|||d|dS)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)rrre)r __stdin__ __stdout__)rrrrerk) old_pdb_initrr_patched_pdb_inits z.CliRunner.isolation.._patched_pdb_initr)rrrr)rrrr)NN)rArrrrr)rNN) rrrrrrrerrkrrr)"r{rsr rrerfr FORCED_WIDTHrrbrrvrwBinaryIOrrg _CHUNK_SIZErDrvisible_prompt_funchidden_prompt_funcrrrrpdbPdbritemsrLenvironget Exceptionr)rrrr bytes_input echo_input old_stdin old_stdout old_stderrold_forced_width stream_mixerrrrrold_visible_prompt_funcold_hidden_prompt_funcold__getchar_funcold_should_strip_ansiold__compat_should_strip_ansirold_envkeyvaluer)rrrr isolationjs "      %         zCliRunner.isolationargsstr | cabc.Sequence[str] | Nonerextrarr|c Ksd}|dur |j}d} d} |jdkr2td} td} z | | Wn ty1d} } Ynw|j|||d4} | durN| durN| jtj_ | jtj _ d} d} d}t |t r^t |}z|d}Wntys||}Ynwzz|jd|p|d|d |} Wn\ty}z4t}td |j}|durd}|dkr|} t |tstjt |tjd d}|}WYd}~n!d}~wty}z|s΂|} d}t}WYd}~nd}~wwWtjtj | dur| dur| }| }|r| d||r| d|| d}| d}| d}nCtjtj | durU| durU| }| }|rK| d||rU| d|| d}| d}| d}wWdn 1sswYt||||| || |d S) 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. Nr r)rrrr prog_namer)rrzint | t.Any | Nonerr)rrrErROSErrorrrIr rernrfryrshlexsplitpopKeyErrorrmain SystemExitrrvrwcoder(r$rr_rVgetvaluer|)rrrrrrrrrcap_outcap_err outstreamsrrrrre_codefd_outfd_errrerfrrrrinvoke0s5                  @zCliRunner.invoketemp_dirstr | os.PathLike[str] | Nonecabc.Iterator[str]ccst}tj|d}t|z&|VWt||dur5ddl}z||WdSty4YdSwdSt||durUddl}z||WwtyTYwww)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)rLgetcwdrNmkdtempchdirshutilrmtreer)rrcwddtrrrrisolated_filesystems0      zCliRunner.isolated_filesystem)rNFTr ) rsrrrrrrrrrrr)rr rrr)rrrr)NNF)rrrrrrrrr)NNNNF)rr rrrrrrrrrrrrrrr|)rrrr) r>r?r@rXrrr contextlibcontextmanagerrrrrrrrr~s6&   I r~)rArBrrC)rrrrsrrr)) __future__rcollections.abcabccabcrrcrLrrr rNtypingrvtypesrrrrrrr TYPE_CHECKING _typeshedr corer Literalrrr rDrErdrYrb TextIOWrapperrgr{r|r~rrrrs<          "  ! 3N