U
e([V* @ s d dl Z d dlZd dlZd dlZd dlZddlmZmZ eje
ddd Z er`d dl m
Z
nd dlZddlmZ G dd deZd d
ZG d d d eZG d
d deZdS ) N ) iteritemsPY2.)StringIO)_find_binary_readerc @ sP e Zd Zdd Zdd Zdd Zddd Zdd
d Zd d
Zdd Z dd Z
dS ) EchoingStdinc C s || _ || _d S N)_input_output)selfinputoutput r /testing.py__init__ s zEchoingStdin.__init__c C s t | j|S r )getattrr
)r xr r r __getattr__ s zEchoingStdin.__getattr__c C s | j | |S r )r write)r rvr r r _echo! s zEchoingStdin._echoc C s | | j|S r )r r
readr nr r r r % s zEchoingStdin.readc C s | | j|S r )r r
readliner r r r r ( s zEchoingStdin.readlinec 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>/ 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__1 s zEchoingStdin.__repr__N)r )r ) __name__
__module__ __qualname__r r r r r r" r% r' r r r r r s
r c C sf t | dr.tr| S t| }|d k r&|S td| d kr<d} nt| tsP| |} tr\t| S t | S )Nr z.Could not find binary reader for input stream. )
hasattrr r TypeError
isinstancebytesencoder ioBytesIO)r
charsetr r r r make_input_stream5 s
r4 c @ s. e Zd ZdZd ddZedd Zdd ZdS )
Resultz3Holds the captured result of an invoked CLI script.Nc C s" || _ || _|| _|| _|| _d S r runner output_bytes exit_code exceptionexc_info)r r7 r8 r9 r: r; r r r r K s
zResult.__init__c C s | j | jjdddS )z The output as unicode string.replace
)r8 decoder7 r3 r< r r r r r X s z
Result.outputc C s d| j rt| j pdf S )Nz <Result %s>Zokay)r: r& r r r r r' ^ s zResult.__repr__)N)r( r) r* __doc__r propertyr r' r r r r r5 H s
r5 c @ sT e Zd ZdZdddZdd Zddd Zejdd
d Z dd
dZ
ejdd Z dS ) CliRunneraI 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. This is
UTF-8 by default and should not be changed currently as
the reporting to Click only works in Python 2 properly.
: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.
NFc C s& |d kr d}|| _ |pi | _|| _d S )Nzutf-8)r3 env
echo_stdin)r r3 rC rD r r r r t s
zCliRunner.__init__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)name)r clir r r get_default_prog_name{ s z CliRunner.get_default_prog_namec C s t | j}|r|| |S )z8Returns the environment overrides for invoking a script.)dictrC update)r Z overridesr r r r make_env s
zCliRunner.make_envc # s t | jtj}tj}tj}tjj}dtj_| |}t
r`t t_ t_}| j rt
|n@t }| j rxt
|tj| jdtj|| jd t_t_t_d fdd } d
fdd }
dd } | d fd
d } tjj}
tjj}tjj}tjj}| tj_|
tj_| tj_| tj_i }zht|D ]T\}}tj|||< |dkrrz tj|= W n tk
rn Y nX n
|tj|< q*|V W 5 t|D ]D\}}|dkrz tj|= W n tk
r Y nX n
|tj|< q|t_|t_|t_|
tj_|tj_|tj_|tj_|tj_X dS )a 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.
.. versionadded:: 4.0
The ``color`` parameter was added.
: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.
P )encodingNc s< t j| p d d}t j|d t j |S )N r= r> )sysstdoutr r rstripflush)promptvalr
r r
visible_input s
z*CliRunner.isolation.<locals>.visible_inputc s, t j| p dd t j dS )NrN r> r= )rO rP r rR r rQ )rS rU r r hidden_input s
z)CliRunner.isolation.<locals>.hidden_inputc S s* t jd}| r&t j| t j |S )Nr )rO stdinr rP r rR )Zechocharr r r _getchar s
z%CliRunner.isolation.<locals>._getcharc s |d kr S | S r r )streamcolor)
default_colorr r should_strip_ansi s z.CliRunner.isolation.<locals>.should_strip_ansi)N)N)NN) r4 r3 rO rX rP stderrclickpkgZ
formattingZ FORCED_WIDTHrK r r rD r r1 r2
TextIOWrapperZtermuiZvisible_prompt_funcZhidden_prompt_funcrZ Zutilsr^ r osenviron Exceptionget)r r
rC r\ Z old_stdinZ
old_stdoutZ
old_stderrZold_forced_widthZ bytes_outputrV rW rZ r^ Zold_visible_prompt_funcZold_hidden_prompt_funcZold__getchar_funcZold_should_strip_ansiZold_envkeyvaluer )r] r
r isolation sv
zCliRunner.isolationTc K s d}| j |||d} d}
d} zz$|jf |p.d| |d| W n tk
r }
zJ|
jdkrd|
}
t }|
j} t
| t stj t
| tj d d} W 5 d}
~
X Y n8 tk
r }
z|s |
}
d} t }W 5 d}
~
X Y nX W 5 tj | } X W 5 Q R X t| | | |
|d S )
ae 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.
.. versionadded:: 3.0
The ``catch_exceptions`` parameter was added.
.. versionchanged:: 3.0
The result object now has an `exc_info` attribute with the
traceback if available.
.. versionadded:: 4.0
The ``color`` parameter was added.
:param cli: the command to invoke
:param args: the arguments to invoke
: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.
N)r
rC r\ r r )argsZ prog_namer> r r r6 )rh rO rP rR getvaluemainrH
SystemExitcoder; r. intr strrd r5 )r rG ri r
rC Zcatch_exceptionsr\ Zextrar; outr: r9 r er r r invoke sB
zCliRunner.invokec c s^ t }t }t | z
|V W 5 t | zt| W n ttfk
rV Y nX X dS )zA context manager that creates a temporary folder and changes
the current working directory to it for isolated filesystem tests.
N) rb getcwdtempfileZmkdtempchdirshutilZrmtreeOSErrorIOError)r cwdtr r r isolated_filesystem3 s
z CliRunner.isolated_filesystem)NNF)N)NNF)NNNTF) r( r) r* r@ r rH rK
contextlibcontextmanagerrh rr r{ r r r r rB d s
i
@rB )rb rO rv rt r| Z_compatr r modulesr( rsplitr` Z cStringIOr r1 r objectr r4 r5 rB r r r r <module> s