U
|
ß=®dÔ> ã @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl m
|
Z
|
ddl m Z ddl m Z ddl mZ ddlmZ ejrddlmZ G d d
|
d
|
Ze jeje ejd d d d ZG dd dejZejeje e ej ej f e ej dddZ!G dd dZ"G dd dZ#dS )é N)Ú TracebackTypeé )Ú
|
formatting)Útermui)Úutils)Ú_find_binary_reader)Ú BaseCommandc @ s° e Zd ZejejddddZeejdddZe e dd d
|
Z
|
de e d d dZ de e d ddZ d e e d ddZeje dddZeje dddZedddZdS ) Ú EchoingStdinN)ÚinputÚoutputÚreturnc C s || _ || _d| _d S )NF)Ú_inputÚ_outputÚ_paused)Úselfr
|
r © r úDd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\click/testing.pyÚ__init__ s zEchoingStdin.__init__)Úxr c C s t | j|S ©N)Úgetattrr )r r r r r Ú __getattr__ s zEchoingStdin.__getattr__)Úrvr c C s | j s| j |¡ |S r )r r Úwrite)r r r r r Ú_echo s zEchoingStdin._echoéÿÿÿÿ)Únr c C s | | j |¡¡S r )r r Úread©r r r r r r # s zEchoingStdin.readc C s | | j |¡¡S r )r r Úread1r r r r r & s zEchoingStdin.read1c C s | | j |¡¡S r )r r Úreadliner r r r r ) s zEchoingStdin.readline©r c s fdd j ¡ D S )Nc s g | ]} |¡qS r ©r ©Ú.0r ©r r r Ú
|
<listcomp>- s z*EchoingStdin.readlines.<locals>.<listcomp>)r Ú readlinesr% r r% r r' , s zEchoingStdin.readlinesc s t fdd jD S )Nc 3 s | ]} |¡V qd S r r" r# r% r r Ú <genexpr>0 s z(EchoingStdin.__iter__.<locals>.<genexpr>)Úiterr r% r r% r Ú__iter__/ s zEchoingStdin.__iter__c C s
|
t | jS r )Úreprr r% r r r Ú__repr__2 s zEchoingStdin.__repr__)r )r )r )Ú__name__Ú
|
__module__Ú __qualname__ÚtÚBinaryIOr ÚstrÚAnyr Úbytesr Úintr r r ÚListr' ÚIteratorr* r, r r r r r s r )Ústreamr c c s& | d krd V nd| _ d V d| _ d S )NTF)r )r8 r r r Ú _pause_echo6 s
|
r9 c sR e Zd Zejeeejdd fdd ZeedddZ eeddd Z
|
Z S )
|
Ú_NamedTextIOWrapperN)ÚbufferÚnameÚmodeÚkwargsr c s t j|f| || _|| _d S r )Úsuperr Ú_nameÚ_mode)r r; r< r= r> ©Ú __class__r r r A s z _NamedTextIOWrapper.__init__r! c C s | j S r )r@ r% r r r r< H s z_NamedTextIOWrapper.namec C s | j S r )rA r% r r r r= L s z_NamedTextIOWrapper.mode) r- r. r/ r0 r1 r2 r3 r Úpropertyr< r= Ú __classcell__r r rB r r: @ s þr: )r
|
Úcharsetr c C sb t | dr6tt tjtj | ¡}|d k r.|S td| d krDd} nt| trX| |¡} t
|
| ¡S )Nr z.Could not find binary reader for input stream.ó ) Úhasattrr r0 ÚcastÚIOr3 Ú TypeErrorÚ
|
isinstancer2 ÚencodeÚioÚBytesIO)r
|
rF r r r r Úmake_input_streamQ s
|
|
|
rP c @ s e Zd ZdZddeeje ejeeje ejej
|
ej e e e f dddZ eeddd Zeedd
|
d Zeedd d ZedddZdS )ÚResultz3Holds the captured result of an invoked CLI script.NÚ CliRunner©ÚrunnerÚ stdout_bytesÚ stderr_bytesÚ return_valueÚ exit_codeÚ exceptionÚexc_infoc C s. || _ || _|| _|| _|| _|| _|| _d S r rS )r rT rU rV rW rX rY rZ r r r r h s zResult.__init__r! c C s | j S )z(The (standard) output as unicode string.)Ústdoutr% r r r r
s z Result.outputc C s | j | jjd¡ dd¡S )z&The standard output as unicode string.Úreplaceú
|
Ú
|
)rU ÚdecoderT rF r\ r% r r r r[ s ÿz Result.stdoutc C s, | j dkrtd| j | jjd¡ dd¡S )z%The standard error as unicode string.Nz stderr not separately capturedr\ r] r^ )rV Ú
|
ValueErrorr_ rT rF r\ r% r r r Ústderr s
|
ÿz Result.stderrc C s, | j rt| j nd}dt| j d| dS )NÚokayú<ú ú>)rY r+ Útyper- )r Zexc_strr r r r, s zResult.__repr__)N)r- r. r/ Ú__doc__r4 r0 ÚOptionalr3 r5 Ú BaseExceptionÚTupleÚTyper r rD r2 r r[ ra r, r r r r rQ e s( öÿø rQ c
|
@ s e Zd ZdZdeejejeeje f eeddddZ d ed
|
d d Z
|
dejejeeje f ejeeje f d ddZ e j d ejejeeejej f ejejeeje f eejejejejej f dddZd d ejejeeje f ejejeeejej f ejejeeje f eeejedddZe j d ejejedf eje dddZdS ) rR aø The 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 stdin writes
|
to stdout. This is useful for showing examples in
|
some circumstances. Note that regular prompts
|
will automatically echo the input.
|
:param mix_stderr: if this is set to `False`, then stdout and stderr are
|
preserved as independent streams. This is useful for
|
Unix-philosophy apps that have predictable stdout and
|
noisy stderr, such that each may be measured
|
independently
|
úutf-8NFT)rF ÚenvÚ
|
echo_stdinÚ
|
mix_stderrr c C s || _ |p i | _|| _|| _d S r )rF rm rn ro )r rF rm rn ro r r r r ² s
|
zCliRunner.__init__r )Úclir c C s
|
|j pdS )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)r< )r rp r r r Úget_default_prog_name¾ s z CliRunner.get_default_prog_name)Ú overridesr c C s t | j}|r| |¡ |S )z8Returns the environment overrides for invoking a script.)Údictrm Úupdate)r rs r r r r Úmake_envÅ s
|
|
zCliRunner.make_env)r
|
rm Úcolorr c # s~ t || j}d}tj}tj}tj}tj} dt_| |¡}t
|
¡ }
|
| j r^t t jt||
|
¡ }}t|| jddd t_| j rd_t|
|
| jdddt_d} | jrªtjt_n t
|
¡ } t| | jd dd
|
d t_t|dt jt td fd d } t|dt jt td fdd } t|ttddd}| dt jt jt j t jt td fdd }tj}tj}tj }t j }| t_| t_|t_ |t _ i }zl| ¡ D ]T\}}t j! #|¡||< |dkrÞz t j!|= W n t"k
|
rÚ Y nX n
|
|t j!|< q|
|
| fV W 5 | ¡ D ]D\}}|dkr:z t j!|= W n t"k
|
r6 Y nX n
|
|t j!|< q|t_|t_|t_|t_|t_|t_ |t _ | t_X dS )an A context manager that sets up the isolation for invoking of a
|
command line tool. This sets up stdin 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.
|
|
.. versionchanged:: 8.0
|
``stderr`` is opened with ``errors="backslashreplace"``
|
instead of the default ``"strict"``.
|
|
.. versionchanged:: 4.0
|
Added the ``color`` parameter.
|
NéP z<stdin>Úr)Úencodingr< r= r z<stdout>Úwz<stderr>Úbackslashreplace)rz r< r= Úerrors)Úpromptr c s> t j | p d¡ ¡ d¡}t j | d¡ t j ¡ |S )NÚ r] r^ )Úsysr[ r r ÚrstripÚflush)r~ Úval©Ú
|
text_inputr r Ú visible_input s
|
|
z*CliRunner.isolation.<locals>.visible_inputc s. t j | p d d¡ t j ¡ ¡ d¡S )Nr r^ r] )r r[ r r r r )r~ r r r Ú hidden_input s
|
z)CliRunner.isolation.<locals>.hidden_input)Úechor c S s* t j d¡}| r t j |¡ t j ¡ |S )Nr )r Ústdinr r[ r r )r Úcharr r r Ú_getchar# s
|
|
z%CliRunner.isolation.<locals>._getchar)r8 rw r c s |d kr S | S r r )r8 rw )Ú default_colorr r Úshould_strip_ansi/ s z.CliRunner.isolation.<locals>.should_strip_ansi)N)N)NN)$rP rF r r r[ ra r Z FORCED_WIDTHrv rN rO rn r0 rI r1 r r: Ú _CHUNK_SIZEro r9 rh r2 ÚboolrJ r3 r Zvisible_prompt_funcZhidden_prompt_funcr r r ÚitemsÚosÚenvironÚ ExceptionÚget)r r
|
rm rw Z bytes_inputZ
|
echo_inputZ old_stdinZ
|
old_stdoutZ
|
old_stderrZold_forced_widthZ bytes_outputZ bytes_errorr r r r Zold_visible_prompt_funcZold_hidden_prompt_funcZold__getchar_funcZold_should_strip_ansiZold_envÚkeyÚvaluer )r r
r Ú isolationÎ s®
|
ÿ ÿ ÿ
|
û ÿ þ
|
|
zCliRunner.isolation)rp Úargsr
|
rm Úcatch_exceptionsrw Úextrar c K s¬ d}| j |||dz} d}
|
d} d} t|tr8t |¡}z| d¡} W n tk
|
rd | |¡} Y nX zòz |j f |pvd| d|}
|
W nÎ tk
|
r } ztt ¡ }t tjtjttjf |j¡}|dkrÎd}|dkrÚ|} t|tstj t|¡ tj d¡ d}|} W 5 d}~X Y n< tk
|
rT } z |s4 |} d} t ¡ }W 5 d}~X Y nX W 5 tj
|
¡ | d ¡ }| j r~d}n | d ¡ }X W 5 Q R X t| |||
|
| | |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``.
|
: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.
|
|
.. 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
|
rm rw r Ú prog_namer r )r r r^ rS )r rL r2 ÚshlexÚsplitÚpopÚKeyErrorrr r r[ r Úgetvaluero ÚmainÚ
|
SystemExitrZ r0 rI rh ÚUnionr5 r3 Úcoder r rQ )r rp r r
|
rm r rw r rZ Z
|
outstreamsrW rY rX r r[ ra ÚeZe_coder r r Úinvoke] sZ ,
|
|
|
ùzCliRunner.invokezos.PathLike[str])Útemp_dirr c c sf t ¡ }tj|d}t |¡ z
|
|V W 5 t |¡ |dkr`zt |¡ W n tk
|
r^ Y nX X dS )aã A 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.
|
)ÚdirN)r ÚgetcwdÚtempfileÚmkdtempÚchdirÚshutilÚrmtreeÚOSError)r r§ ÚcwdÚdtr r r Úisolated_filesystem s
|
|
|
z CliRunner.isolated_filesystem)rl NFT)N)NNF)NNNTF)N)r- r. r/ rg r2 r0 rh ÚMappingr r rr rv Ú
|
contextlibÚcontextmanagerr£ r4 rJ r3 r7 rj rN rO r ÚSequencerQ r¦ r² r r r r rR sd ûú ÿþ ü û ù ÷ e ÿþrR )$r´ rN r r r r rª Útypingr0 Útypesr r r r r Ú_compatr Ú TYPE_CHECKINGÚcorer r rµ rh r7 r9 Ú TextIOWrapperr: r£ r2 r4 rJ r3 r1 rP rQ rR r r r r Ú<module> s0 " þ :
|
