B h$@sddlZddlZddlZddlZddlZddlZddlZddlZejddZ ejdde fddZ ddZ ejej fdd Z ejdd e fd d Zejd dZGdddZGdddejejZGdddejZdS)Nc cs.t}t|z |VWdt|XdS)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)dirorigraC:\Users\sanjo\AppData\Local\Qlobot\Launcher\ext_packages\pkg_resources\_vendor\jaraco\context.pypushd s   r c cs|dkr$tj|dddd}|dkr>tjtjdd}n t dt |dj ft zRd }d }d ||f}||j fd t|it || |VWdQRXWd|d j ft XdS)z Get a tarball, extract it, change to that directory, yield, then clean up. `runner` is the function to invoke commands. `pushd` is a context manager for changing the directory. Nz.tar.gzz.tgzT)shellzrunner parameter is deprecatedzmkdir {target_dir}zwget {url} -O -z7tar x{compression} --strip-components=1 -C {target_dir}z | compressionzrm -Rf {target_dir})rpathbasenamereplace functoolspartial subprocess check_callwarningswarnDeprecationWarningformatvarsjoininfer_compression)urlZ target_dirrunnerr getterextractcmdrrrtarball_contexts  r cCs&|dd}tdddd}||dS)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' NzjJ)gzZbzxz)dictget)rZcompression_indicatormappingrrrr:s rccs"t}z |VWd||XdS)aN 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') >>> assert not os.path.exists(the_dir) N)tempfilemkdtemp)Zremovertemp_dirrrrr,Ns  r,Tc csrd|kr dnd}|R}|d||g}|r6|d|gttjjd}|rL|nd}tj||d|VWdQRXdS)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. gitZhgclonez--branchwN)stdout)extendopenrr devnullrr) rbranchquietZdest_ctxZexeZrepo_dirrr3r0rrr repo_contextas  r6ccs dVdS)z A null context suitable to stand in for a meaningful context. >>> with null() as value: ... assert value is None Nrrrrrnulltsr7c@steZdZdZdZeffddZddZeddZ ed d Z ed d Z d dZ ddZ edddZddZdS) 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 )NNNcCs ||_dS)N) exceptions)selfr9rrr__init__szExceptionTrap.__init__cCs|S)Nr)r:rrr __enter__szExceptionTrap.__enter__cCs |jdS)Nr)exc_info)r:rrrtypeszExceptionTrap.typecCs |jdS)N)r=)r:rrrvalueszExceptionTrap.valuecCs |jdS)N)r=)r:rrrtbszExceptionTrap.tbcGs&|d}|ot||j}|r"||_|S)Nr) issubclassr9r=)r:r=r>matchesrrr__exit__s zExceptionTrap.__exit__cCs t|jS)N)boolr>)r:rrr__bool__szExceptionTrap.__bool__)_testcstfdd}|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 s(tj}||WdQRX|S)N)r8r9)argskwargstrap)rHfuncr:rrwrappers z%ExceptionTrap.raises..wrapper)rwraps)r:rLrHrMr)rHrLr:rraisesszExceptionTrap.raisescCs|j|tjdS)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 )rH)rOoperatornot_)r:rLrrrpassesszExceptionTrap.passesN)__name__ __module__ __qualname____doc__r= Exceptionr;r<propertyr>r@rBrErGrFrOrRrrrrr8s    r8c@seZdZdZdS)suppressz A version of contextlib.suppress with decorator support. >>> @suppress(KeyError) ... def key_error(): ... {}[''] >>> key_error() N)rSrTrUrVrrrrrYsrYc@s*eZdZdZd ddZddZdd Zd S) 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?cCs||_||_dS)N)actioncode)r:r\r]rrrr;szon_interrupt.__init__cCs|S)Nr)r:rrrr<szon_interrupt.__enter__cCs6|tk s|jdkrdS|jdkr,t|j||jdkS)Nignorer[rY)KeyboardInterruptr\ SystemExitr])r:exctypeexcinstexctbrrrrEs   zon_interrupt.__exit__N)r[r?)rSrTrUrVr;r<rErrrrrZs  rZ)rr contextlibrr*shutilrPrcontextmanagerr r rrmtreer,r6r7r8rYContextDecoratorrZrrrrs& q