3 ] d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z d dl Z d dl
mZ ej dk rd dlmZ nd dlZej d d
Zej d d d
Zd!dZd Z eee Zej d Zd Zej ej fd Zej ddefd Z d Z G d d Z G d dej ej Z G d d ej Z!dS )" ) annotationsN)Iterator) )tarfiledirstr | os.PathLikereturnIterator[str | os.PathLike]c # K t j }t j | | V t j | dS # t j | w xY w)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)osgetcwdchdir)r origs s/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib/python3.11/site-packages/setuptools/_vendor/jaraco/context.pypushdr sQ 9;;DHSMMM
s A A
target_dirstr | os.PathLike | Nonec # K |Gt j | dd dd }t j | t
j | }t j |d 5 }|
|t ddd n # 1 swxY w Y |V t j
| dS # t j
| w xY w)a
Get a tarball, extract it, yield, then clean up.
>>> 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)
Nz.tar.gz z.tgzzr|*)fileobjmode)pathfilter)r
r basenamereplacemkdirurllibrequesturlopenr open
extractallstrip_first_componentshutilrmtree)urlr reqtfs r tarballr) ' s6 W%%c**229bAAII&RTUU
HZ"n$$S))
\#E
2
2
2 Ib MMz2GM H H H I I I I I I I I I I I I I I I
j!!!!!
j!!!!s0 !5C$ B?3 C$ ?CC$ CC$ $C:membertarfile.TarInfoc L | j dd \ }| _ | S )N/ )namesplit)r* r _s r r# r# G s& [&&sA..NAv{ M c L d }t j |t | S )a
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 : fd}t j | S )Nc ? K | i |5 } | 5 }|V d d d n # 1 swxY w Y d d d d S # 1 swxY w Y d S N )argskwargssavedresinnerouters r composedz/_compose..compose_two..composedd s '''
5%%,,
#
s, A) A- A- AAA)
contextlibcontextmanager)r< r= r> s `` r compose_twoz _compose..compose_twoc s5 (222r2 ) functoolsreducereversed)cmgrsrA s r _composerF O s, (3 3 3 K% 9 99r2 c / K t j dt d | dt }t | i |5 } || 5 }|V d d d n # 1 swxY w Y d d d d S # 1 swxY w Y d S )NzBtarball_context is deprecated. Use tarball or tarball_cwd instead.
stacklevelr )warningswarnDeprecationWarningpopr r) )r8 r9 pushd_ctxtballr s r tarball_contextrQ p s ML
7E**I $ !& ! ! UIIe,<,< s6 A;A# A;#A' 'A;*A' +A;;A?A?c t j dt d | dd }t ddd }| |d S )
a
Given a URL or filename, infer the compression code for tar.
>>> infer_compression('http://foo/bar.tar.gz')
'z'
>>> infer_compression('http://foo/bar.tgz')
'z'
>>> infer_compression('file.bz')
'j'
>>> infer_compression('file.xz')
'J'
z3infer_compression is deprecated with no replacementrH rI NzjJ)gzbzxz)rK rL rM dictget)r& compression_indicatormappings r infer_compressionr^ | s_
M= Hccc***G ;;,c 2 22r2 c # p K t j } |V | | dS # | | w xY w)a`
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)
... _ = pathlib.Path(the_dir).joinpath('somefile').write_text('contents', encoding='utf-8')
>>> assert not os.path.exists(the_dir)
N)tempfilemkdtemp)removertemp_dirs r rc rc sP !!Hs (
5Tc # K d| v rdnd} | 5 }|d| |g}|r| d|g t t j j d }|r|nd}t j || |V ddd dS # 1 swxY w Y dS )z
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.
githgclonez--branchwN)stdout)extendr! r
r devnull
subprocess
check_call) r& branchquietdest_ctxexerepo_dircmdrk ri s r repo_contextrt s C<<%%TC xGS(+ - JJ
F+ , , ,rw,, !+t c&1111 s A BBBc ` t j dt d t j S )aD
A null context suitable to stand in for a meaningful context.
>>> with null() as value:
... assert value is None
This context is most useful when dealing with two or more code
branches but only some need a context. Wrap the others in a null
context to provide symmetry across all options.
z.null is deprecated. Use contextlib.nullcontextrH rI )rK rL rM r? nullcontextr7 r2 r nullrw s7
M8
! # ##r2 c e Zd ZdZdZeffdZd Zed Z ed Z
ed Z d Z d Z
ed
d Zd Zd
S )
ExceptionTrapa
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
>>> with ExceptionTrap(ValueError) as trap:
... raise Exception()
Traceback (most recent call last):
...
Exception
>>> bool(trap)
False
)NNNc || _ d S r6 )
exceptions)selfr{ s r __init__zExceptionTrap.__init__ s
$r2 c | S r6 r7 r| s r __enter__zExceptionTrap.__enter__ r2 c | j d S Nr exc_infor s r typezExceptionTrap.type }Q r2 c | j d S )Nr. r r s r valuezExceptionTrap.value r r2 c | j d S )NrH r r s r tbzExceptionTrap.tb r r2 c V |d }|ot || j }|r|| _ |S r )
issubclassr{ r )r| r r matchess r __exit__zExceptionTrap.__exit__ s5 {<:dDO<< % $DMr2 c * t | j S r6 )boolr r s r __bool__zExceptionTrap.__bool__ s DI r2 _testc N t j fd }|S )a
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 j 5 } | i | d d d n # 1 swxY w Y | S r6 )ry r{ )r8 r9 trapr funcr| s r wrapperz%ExceptionTrap.raises..wrapper! s t//
&4d%f%%%
&
&
&
&
&
&
&
&
&
&
&
&
&
&
&5;; s +//)rB wraps)r| r r r s ``` r raiseszExceptionTrap.raises sF &
r2 c D | |t j S )a
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 operatornot_)r| r s r passeszExceptionTrap.passes) s & {{4x}{555r2 N)__name__
__module__ __qualname____doc__r Exceptionr} r propertyr r r r r r r r r7 r2 r ry ry s B H#,, % % % % X X X %) 66 6 6 6 6r2 ry c e Zd ZdZdS )suppressz
A version of contextlib.suppress with decorator support.
>>> @suppress(KeyError)
... def key_error():
... {}['']
>>> key_error()
N)r r r r r7 r2 r r r ? s r2 r c & e Zd ZdZddZd Zd ZdS ) on_interrupta
Replace a KeyboardInterrupt with SystemExit(1)
>>> 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)()
errorr. c " || _ || _ d S r6 )actioncode)r| r r s r r} zon_interrupt.__init__] s r2 c | S r6 r7 r s r r zon_interrupt.__enter__a r r2 c |t us | j dk rd S | j dk rt | j || j dk S )Nignorer r )KeyboardInterruptr
SystemExitr )r| exctypeexcinstexctbs r r zon_interrupt.__exit__d sK + + +t{h/F/F F
[G
#
# TY''W 4{j((r2 N)r r. )r r r r r} r r r7 r2 r r r J sP $ ) ) ) ) )r2 r )r r r
r r6 )r r r
r )r* r+ r
r+ )"
__future__r r? rB r r
r$ rl sysr` urllib.requestr rK typingr version_info backportsr r@ r r) r# rF tarball_cwdrQ r^ r% rc rt rw ry r ContextDecoratorr r7 r2 r r sn " " " " " "
g !!!!!!!NNN
04 " " " " "> : : :< hug&& 3 3 32 ] $ ! $$ $ $&n6 n6 n6 n6 n6 n6 n6 n6b z"J$? ) ) ) ) ):. ) ) ) ) )r2