o [hJ@s0ddlmZddlmZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl mZddlmZddlmZddlmZddlmZdd lmZe jr[dd lmZdd lmZGd d d Zejd%ddZGdddejZGdddZGdddej Z!d&dd Z"Gd!d"d"Z#Gd#d$d$Z$dS)') annotationsN) TracebackType)_compat) formatting)termui)utils)_find_binary_reader)ReadableBuffer)Commandc@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)selfr rrM/var/www/html/premium_crap/venv/lib/python3.10/site-packages/click/testing.py__init__s zEchoingStdin.__init__xstrt.AnycCs t|j|SN)getattrr)rrrrr __getattr__s zEchoingStdin.__getattr__rvbytescCs|js |j||Sr)rrwrite)rrrrr_echo"s zEchoingStdin._echonintcC||j|Sr)r"rreadrr$rrrr'(zEchoingStdin.readcCr&r)r"rread1r(rrrr*+r)zEchoingStdin.read1cCr&r)r"rreadliner(rrrr+.r)zEchoingStdin.readline list[bytes]csfddjDS)Ncsg|]}|qSrr".0rrrr 2sz*EchoingStdin.readlines..)r readlinesr0rr0rr21zEchoingStdin.readlinescabc.Iterator[bytes]cstfddjDS)Nc3s|]}|VqdSrr-r.r0rr 5sz(EchoingStdin.__iter__..)iterrr0rr0r__iter__4r3zEchoingStdin.__iter__cCs t|jSr)reprrr0rrr__repr__7s zEchoingStdin.__repr__N)r rrrrr)rrrr)rr rr )r#)r$r%rr )rr,)rr4rr) __name__ __module__ __qualname__rrr"r'r*r+r2r7r9rrrrr s        r streamEchoingStdin | Nonercabc.Iterator[None]ccs*|dur dVdSd|_dVd|_dS)NTF)r)r>rrr _pause_echo;s   rAcs>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)superrrC)rrC __class__rrrKs  zBytesIOCopy.__init__cst|jdSr)rEflushrCr0rFrrrHOs zBytesIOCopy.flushbr r%cs|j|t|Sr)rCr!rE)rrIrFrrr!Ss  zBytesIOCopy.write)rCrDrrrr)rIr rr%)r;r<r=__doc__rrHr! __classcell__rrrFrrBEs rBc@s$eZdZdZd ddZd ddZdS) StreamMixerzMixes `` and `` streams. The result is available in the ``output`` attribute. .. versionadded:: 8.2 rrcCs*t|_t|jd|_t|jd|_dS)N)rC)ioBytesIOrrBstdoutstderrr0rrrr`s zStreamMixer.__init__cCs"|j|j|jdS)z Guarantee that embedded file-like objects are closed in a predictable order, protecting against races between self.output being closed and other streams being flushed on close .. versionadded:: 8.2.2 N)rQcloserPrr0rrr__del__es  zStreamMixer.__del__NrJ)r;r<r=rKrrSrrrrrMXs rMcs:eZdZdfd d Zedd d ZedddZZS)_NamedTextIOWrapperbufferrnamermodekwargsrrrc s$tj|fi|||_||_dSr)rEr_name_mode)rrUrVrWrXrFrrrss z_NamedTextIOWrapper.__init__cC|jSr)rYr0rrrrVzz_NamedTextIOWrapper.namecCr[r)rZr0rrrrW~r\z_NamedTextIOWrapper.mode) rUrrVrrWrrXrrrr:)r;r<r=rpropertyrVrWrLrrrFrrTrs  rTr 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 isinstancerencoderNrO)r r_rrrrmake_input_streams    rgc@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||_||_||_||_||_||_||_||_dSrrirkrlrmrnrorprr) rrirkrlrmrnrorprrrrrrs  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  )rmdecoderir_rvr0rrrrsz Result.outputcCru)z&The standard output as unicode string.rvrwrx)rkryrir_rvr0rrrrPsz Result.stdoutcCru)zThe standard error as unicode string. .. versionchanged:: 8.2 No longer raise an exception, always returns the `` string. rvrwrx)rlryrir_rvr0rrrrQsz Result.stderrcCs,|jrt|jnd}dt|jd|dS)Nokay< >)rpr8typer;)rexc_strrrrr9szResult.__repr__r)rirjrkr rlr rmr rnrror%rprqrrrsr:) r;r<r=rKrr]rrPrQr9rrrrrhs     rhc@s~eZdZdZ    d,d-ddZd.ddZ d/d0ddZej   d1d2ddZ     d3d4d%d&Z ej d/d5d*d+Z dS)6rjaThe 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. utf-8NFTr_renv$cabc.Mapping[str, str | None] | None echo_stdinboolcatch_exceptionsrrcCs ||_|pi|_||_||_dSr)r_rrr)rr_rrrrrrrs  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)rV)rrrrrget_default_prog_names zCliRunner.get_default_prog_name overridescabc.Mapping[str, str | None]cCst|j}|r |||S)z8Returns the environment overrides for invoking a script.)dictrupdate)rrrrrrmake_envs  zCliRunner.make_envr r^color8cabc.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|ddfdd } t|ddfdd } t|d dd} | d!d"fdd }tj}tj}tj}tj}tj}| t_| t_| t_|t_|t_i}zs|D]%\}}tj|||<|durztj|=WqtyYqw|tj|<q| j| j| jfVW|D]\}}|durztj|=WqtyYqw|tj|<q|t_|t_|t_|t_|t_|t_|t_|t_| t_dS|D]"\}}|dur9ztj|=Wqty8Yqw|tj|<q|t_|t_|t_|t_|t_|t_|t_|t_| t_w)#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. NPzr)encodingrVrWrzwzbackslashreplace)rrVrWerrorsprompt str | Nonerrc sdtj|pdz td}Wnty!}zt|d}~wwtj|dtj|S)Nrwrx)sysrPr!nextrstrip StopIterationEOFErrorrH)rvale text_inputrr visible_input`s z*CliRunner.isolation..visible_inputc sRtj|pddtjztdWSty(}zt|d}~ww)Nrrxrw)rrPr!rHrrrr)rrrrr hidden_inputks z)CliRunner.isolation..hidden_inputechorcSs*tjd}|rtj|tj|S)Nr)rstdinr'rPr!rH)rcharrrr_getcharts   z%CliRunner.isolation.._getcharr>t.IO[t.Any] | Noner bool | Nonecs|dur S| Srr)r>r) default_colorrrshould_strip_ansisz.CliRunner.isolation..should_strip_ansir)rrrr)rrrr)NN)r>rrrrr)rgr_rrrPrQr FORCED_WIDTHrrMrrbrcBinaryIOr rT _CHUNK_SIZErArvisible_prompt_funchidden_prompt_funcrrrritemsosenvironget Exceptionr)rr rr 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_ansiold_envkeyvaluer)rrr isolations "               zCliRunner.isolationargsstr | cabc.Sequence[str] | Nonerextrarrhc Ksd}|dur |j}|j|||d} d} d} d} t|tr"t|}z|d} Wnty7||} Ynwzz|j d|p@d| d|} Wn\t y}z4t }t d|j}|durad}|dkrg|} t|ts|t jt|t jdd }|} WYd}~n!d}~wty}z|s|} d } t }WYd}~nd}~wwWt jt j| d}| d }| d }nt jt j| d}| d }| d }wWdn1swYt||||| | | |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. N)r rrr prog_namer)rrzint | t.Any | Nonerxrrt)rrrershlexsplitpopKeyErrorrmain SystemExitrrrrbrccoder%rPr!rrHrQgetvaluerh)rrrr rrrrrr outstreamsrnrprorre_coderPrQrrrrinvokesv5            -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)rgetcwdtempfilemkdtempchdirshutilrmtreeOSError)rrcwddtrrrrisolated_filesystem"s0      zCliRunner.isolated_filesystem)rNFT) r_rrrrrrrrr)rr rrr)rrrr)NNF)r r^rrrrrr)NNNNF)rr rrr r^rrrrrrrrrrh)rrrr) r;r<r=rKrrr contextlibcontextmanagerrrrrrrrrjs2    qrj)r>r?rr@)r r^r_rrr)% __future__rcollections.abcabccabcrrNrrrrtypingrbtypesrrrrrrr TYPE_CHECKING _typeshedr corer r rrArOrBrM TextIOWrapperrTrgrhrjrrrrs6         "   N