
    0g$                    f   S SK Jr  S SKrS SKrS SKrS SKrS SKrS SKrS SKrS SK	r	S SK
r
S SKrS SKrS SKrS SKJr  \R"                  S:  a  S SKJr  OS SKr\R(                  SS j5       r\R(                   S   SS jj5       r    SS jrS	 r\" \\5      r S
 rS r\R(                  \R8                  4S j5       r\R<                  " \\" 5       S9r\R(                  SS\4   SS jj5       r  " S S5      r! " S S\RD                  \RF                  5      r" " S S\RF                  5      r$g)    )annotationsN)Iterator)      )tarfilec              #     #    [         R                  " 5       n[         R                  " U 5         U v   [         R                  " U5        g! [         R                  " U5        f = f7f)z
>>> tmp_path = getfixture('tmp_path')
>>> with pushd(tmp_path):
...     assert os.getcwd() == os.fspath(tmp_path)
>>> assert os.getcwd() != os.fspath(tmp_path)
N)osgetcwdchdir)dirorigs     9/usr/lib/python3/dist-packages/jaraco/context/__init__.pypushdr      s=      99;DHHSM	
s   ,A%A
 A%
A""A%c              #    #    Uc?  [         R                  R                  U 5      R                  SS5      R                  SS5      n[         R                  " U5         [
        R                  R                  U 5      n[        R                  " USS9 nUR                  U[        S9  SSS5        Uv   [        R                  " U5        g! , (       d  f       N)= f! [        R                  " U5        f = f7f)a  
Get a URL to a tarball, download, extract, yield, then clean up.

Assumes everything in the tarball is prefixed with a common
directory. That common path is stripped and the contents
are extracted to ``target_dir``, similar to passing
``-C {target} --strip-components 1`` to the ``tar`` command.

Uses the streaming protocol to extract the contents from a
stream in a single pass without loading the whole file into
memory.

>>> import urllib.request
>>> url = getfixture('tarfile_served')
>>> target = getfixture('tmp_path') / 'out'
>>> tb = tarball(url, target_dir=target)
>>> import pathlib
>>> with tb as extracted:
...     contents = pathlib.Path(extracted, 'contents.txt').read_text(encoding='utf-8')
>>> assert not os.path.exists(extracted)

If the target is not specified, contents are extracted to a
directory relative to the current working directory named after
the name of the file as extracted from the URL.

>>> target = getfixture('tmp_path')
>>> with pushd(target), tarball(url):
...     target.joinpath('served').is_dir()
True
Nz.tar.gz z.tgzzr|*)fileobjmode)pathfilter)r	   r   basenamereplacemkdirurllibrequesturlopenr   open
extractallstrip_first_componentshutilrmtree)url
target_dirreqtfs       r   tarballr%   )   s     D WW%%c*229bAII&RTU
HHZ"nn$$S)\\#E2bMMz2GMH 3j!	 32 	j!s6   AC44C C%C 1C4
CC C11C4c                L    U R                   R                  SS5      u  o l         U $ )N/   )namesplit)memberr   _s      r   r   r   W   s#     [[&&sA.NA{M    c                 H    S n[         R                  " U[        U 5      5      $ )aQ  
Compose any number of dependent context managers into a single one.

The last, innermost context manager may take arbitrary arguments, but
each successive context manager should accept the result from the
previous as a single parameter.

Like :func:`jaraco.functools.compose`, behavior works from right to
left, so the context manager should be indicated from outermost to
innermost.

Example, to create a context manager to change to a temporary
directory:

>>> temp_dir_as_cwd = _compose(pushd, temp_dir)
>>> with temp_dir_as_cwd() as dir:
...     assert os.path.samefile(os.getcwd(), dir)
c                @   ^ ^ U U4S jn[         R                  " U5      $ )Nc               ?     >#    T" U 0 UD6 nT" U5       nUv   S S S 5        S S S 5        g ! , (       d  f       N= f! , (       d  f       g = f7fN )argskwargssavedresinnerouters       r   composed/_compose.<locals>.compose_two.<locals>.composedt   s8     ''5%,#	 3?'',,''s)   	A	<+<	A
9	<
A
A)
contextlibcontextmanager)r7   r8   r9   s   `` r   compose_two_compose.<locals>.compose_twos   s    	 ((22r-   )	functoolsreducereversed)cmgrsr=   s     r   _composerC   _   s     (3 K%99r-   c                P   Uu  p4nU [         R                  [         R                  [         R                  4;   an  UR                  [        R
                  :X  aP  [         R                  " U[        R                  [        R                  -  [        R                  -  5        U " U5        ge )z6
Add support for removing read-only files on Windows.
N)r	   rmdirremoveunlinkerrnoEACCESchmodstatS_IRWXUS_IRWXGS_IRWXO)funcr   exc_infor,   excs        r   remove_readonlyrR      se     IAA"))RYY//CII4M
t||dll2T\\ABT
r-   c                     [         R                  " 5       S:X  a'  [        R                  " [        R
                  [        S9$ [        R
                  $ )NWindows)onerror)platformsystemr?   partialr   r    rR   r2   r-   r   robust_removerrY      s<     ??	) 	&--A ]]r-   c              #  j   #    [         R                  " 5       n Uv   U " U5        g! U " U5        f = f7f)z
Create a temporary directory context. Pass a custom remover
to override the removal behavior.

>>> import pathlib
>>> with temp_dir() as the_dir:
...     assert os.path.isdir(the_dir)
>>> assert not os.path.exists(the_dir)
N)tempfilemkdtemp)removertemp_dirs     r   r^   r^      s/      !Hs   3& 	3
03)r]   Tc              #    #    SU ;   a  SOSnU" 5        nUSX/nUR                  SU/[        U5      -  5        U(       a  [        R                  OSn[        R                  " XgUS9  Uv   SSS5        g! , (       d  f       g= f7f)a0  
Check out the repo indicated by url.

If dest_ctx is supplied, it should be a context manager
to yield the target directory for the check out.

>>> repo = repo_context('https://github.com/jaraco/jaraco.context')
>>> with repo as dest:
...     listing = os.listdir(dest)
>>> 'README.rst' in listing
True
githgclonez--branchN)stdoutstderr)extendbool
subprocessDEVNULL
check_call)r!   branchquietdest_ctxexerepo_dircmdstreams           r   repo_contextrq      sn       C<%TC	xGS+

J'$v,67',##$c@ 
s   BAA4+	B4
B>Bc                      \ rS rSrSrSr\44S jrS r\	S 5       r
\	S 5       r\	S 5       rS	 rS
 r\S.S jrS rSrg)ExceptionTrap   aA  
A context manager that will catch certain exceptions and provide an
indication they occurred.

>>> with ExceptionTrap() as trap:
...     raise Exception()
>>> bool(trap)
True

>>> with ExceptionTrap() as trap:
...     pass
>>> bool(trap)
False

>>> with ExceptionTrap(ValueError) as trap:
...     raise ValueError("1 + 1 is not 3")
>>> bool(trap)
True
>>> trap.value
ValueError('1 + 1 is not 3')
>>> trap.tb
<traceback object at ...>

>>> with ExceptionTrap(ValueError) as trap:
...     raise Exception()
Traceback (most recent call last):
...
Exception

>>> bool(trap)
False
)NNNc                    Xl         g r1   )
exceptions)selfrv   s     r   __init__ExceptionTrap.__init__   s    $r-   c                    U $ r1   r2   rw   s    r   	__enter__ExceptionTrap.__enter__       r-   c                     U R                   S   $ Nr   rP   r{   s    r   typeExceptionTrap.type       }}Qr-   c                     U R                   S   $ )Nr(   r   r{   s    r   valueExceptionTrap.value   r   r-   c                     U R                   S   $ )N   r   r{   s    r   tbExceptionTrap.tb   r   r-   c                f    US   nU=(       a    [        X R                  5      nU(       a  Xl        U$ r   )
issubclassrv   rP   )rw   rP   r   matchess       r   __exit__ExceptionTrap.__exit__   s+    {<:dOO<$Mr-   c                ,    [        U R                  5      $ r1   )rf   r   r{   s    r   __bool__ExceptionTrap.__bool__  s    DIIr-   _testc               N   ^ ^^ [         R                  " T5      UUU 4S j5       nU$ )aQ  
Wrap func and replace the result with the truth
value of the trap (True if an exception occurred).

First, give the decorator an alias to support Python 3.8
Syntax.

>>> raises = ExceptionTrap(ValueError).raises

Now decorate a function that always fails.

>>> @raises
... def fail():
...     raise ValueError('failed')
>>> fail()
True
c                    > [        TR                  5       nT" U 0 UD6  S S S 5        T" W5      $ ! , (       d  f       N= fr1   )rs   rv   )r3   r4   trapr   rO   rw   s      r   wrapper%ExceptionTrap.raises.<locals>.wrapper  s6    t/4d%f% 0; 0/s   	0
>)r?   wraps)rw   rO   r   r   s   ``` r   raisesExceptionTrap.raises  s'    & 
		 
	
 r-   c                >    U R                  U[        R                  S9$ )aJ  
Wrap func and replace the result with the truth
value of the trap (True if no exception).

First, give the decorator an alias to support Python 3.8
Syntax.

>>> passes = ExceptionTrap(ValueError).passes

Now decorate a function that always fails.

>>> @passes
... def fail():
...     raise ValueError('failed')

>>> fail()
False
r   )r   operatornot_)rw   rO   s     r   passesExceptionTrap.passes"  s    & {{4x}}{55r-   )rP   rv   N)__name__
__module____qualname____firstlineno____doc__rP   	Exceptionrx   r|   propertyr   r   r   r   r   rf   r   r   __static_attributes__r2   r-   r   rs   rs      ss    B  H#,, %             %) 66r-   rs   c                      \ rS rSrSrSrg)suppressi8  z
A version of contextlib.suppress with decorator support.

>>> @suppress(KeyError)
... def key_error():
...     {}['']
>>> key_error()
r2   N)r   r   r   r   r   r   r2   r-   r   r   r   8  s    r-   r   c                  .    \ rS rSrSrSS jrS rS rSrg)	on_interruptiC  a  
Replace a KeyboardInterrupt with SystemExit(1).

Useful in conjunction with console entry point functions.

>>> def do_interrupt():
...     raise KeyboardInterrupt()
>>> on_interrupt('error')(do_interrupt)()
Traceback (most recent call last):
...
SystemExit: 1
>>> on_interrupt('error', code=255)(do_interrupt)()
Traceback (most recent call last):
...
SystemExit: 255
>>> on_interrupt('suppress')(do_interrupt)()
>>> with __import__('pytest').raises(KeyboardInterrupt):
...     on_interrupt('ignore')(do_interrupt)()
c                   Xl         X l        g r1   actioncode)rw   r   r   s      r   rx   on_interrupt.__init__X  s    	r-   c                    U $ r1   r2   r{   s    r   r|   on_interrupt.__enter__\  r~   r-   c                    U[         Ld  U R                  S:X  a  g U R                  S:X  a  [        U R                  5      UeU R                  S:H  $ )Nignoreerrorr   )KeyboardInterruptr   
SystemExitr   )rw   exctypeexcinstexctbs       r   r   on_interrupt.__exit___  sE    ++t{{h/F[[G#TYY'W4{{j((r-   r   N)r   r(   )	r   r   r   r   r   rx   r|   r   r   r2   r-   r   r   r   C  s    ()r-   r   )r   zstr | os.PathLikereturnIterator[str | os.PathLike]r1   )r"   zstr | os.PathLike | Noner   r   )r+   tarfile.TarInfor   r   )rj   z
str | Nonerk   rf   )%
__future__r   r;   rH   r?   r   r	   rV   r   rK   rg   sysr[   urllib.requestr   typingr   version_info	backportsr   r<   r   r%   r   rC   tarball_cwdrR   rY   r    r^   rX   robust_temp_dirrq   rs   r   ContextDecoratorr   r2   r-   r   <module>r      sh   "     	     
    g!    04*"-*" *" *"Z :< ug&
 ]]  " ##Hn6FG "$+/ 0n6 n6bz""J$?$? !):.. !)r-   