+ h$csb^RIHt^RIt^RIt^RIt^RIt^RIt^RIt^RIt^RI t ^RI t ^RI t ^RI t ^RI t^RIHt] P"R8d^RIHtM^RIt]P(RRl4t]P(RRRll4tR R ltR t]!]]4tR tR t]P(]P83Rl4t]P<!]]!4R7t]P(RR]3RRll4t !RR4t!!RR]PD]PF4t"!RR]PF4t$R#)) annotationsN)Iterator)tarfilec V^8dQhRRRR/#)dirzstr | os.PathLikereturnIterator[str | os.PathLike]Zformat">> tmp_path = getfixture('tmp_path') >>> with pushd(tmp_path): ... assert os.getcwd() == os.fspath(tmp_path) >>> assert os.getcwd() != os.fspath(tmp_path) N)osZgetcwdZchdir)rZorig& r pushdrs= 99;DHHSM  s,A&A A& A##A&cr)r target_dirzstr | os.PathLike | Nonerrr r r r r r *s*"*"-*" *"rc#s"Vf@\PPV4PRR4PRR4p\P!V4\ P PV4p\P!VRR7;_uu_4pVPV\R7RRR4Vx\P!V4R# +'giL,;i \P!T4i;i5i)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.gzZz.tgzzr|*)ZfileobjZmode)pathZfilter)rrZbasenameZreplaceZmkdirurllibZrequestZurlopenrZopenZ extractallstrip_first_componentshutilrmtree)urlrZreqZtfs&& r tarballr)sDWW%%c*229bAII&RTU HHZ"nn$$S) \\#E 2 2b MMz2GM H3 j! 3 2  j!s6AD?C+C4 C+D C( #C++DDcs V^8dQhRRRR/#)rmemberztarfile.TarInforr r r r r r Ws rcsLVPPR^4wq nV#)Z/)ZnameZsplit)rr_s&& r rrWs#[[&&sA.NA{ MrcsHRp\P!V\V44#)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) cs@aaVV3Rlp\P!V4#)c ?s<"S!V/VB;_uu_4pS!V4;_uu_4pVxRRR4RRR4R# +'giL;i +'giR#;i5iNr )argskwargsZsavedZresinnerouters*, r composedZ/_compose..compose_two..composedts@'''5%,,# 3?'',,'''s/A#A< A A# A A A  A#) contextlibcontextmanager)rr r!sff r compose_twoZ_compose..compose_twoss ((22r) functoolsZreduceZreversed)Zcmgrsr$s* r _composer&_s (3   K% 99rcsfVwr4pV\P\P\P39dyVP\P 8XdZ\P !V\P\P,\P,4V!V4R#h)z6 Add support for removing read-only files on Windows. N) rZrmdirZremoveZunlinkerrnoZEACCESZchmodstatZS_IRWXUZS_IRWXGZS_IRWXO)funcrexc_inforZexcs&&& r remove_readonlyr+seIAA "))RYY//CII4M t||dll2T\\AB T  rcs\P!4R8Xd+\P!\P \ R7#\P #)ZWindows)Zonerror)platformZsystemr%partialrrr+r rr robust_removerr.s< ??  ) &--A]]rc#sl"\P!4pVxV!V4R# T!T4i;i5i)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)tempfileZmkdtemp)removertemp_dirrr r1r1s/!Hs4' 4 14)r0Tcr)rbranchz str | Nonequietboolr r r r r r s+/rc#s8"RV9dRMRpV!4;_uu_4pVRW.pVPRV.\V4,4V'd\PMRp\P!WgVR7VxRRR4R# +'giR#;i5i)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 ZgitZhgZclonez--branchN)ZstdoutZstderr)Zextendr4 subprocessZDEVNULLZ check_call)rr2r3Zdest_ctxZexeZrepo_dirZcmdZstreams&&&& r repo_contextr6sr C<%TC xGS+ J'$v,67',##$c@ sBAB< B B  Bcs]tRt^tRtRt]33RltRt] R4t ] R4t ] R4t Rt R tR ]/R ltR tR tR#) ExceptionTrapaA 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 Nc sWnR#r) exceptions)selfr8&&r __init__ZExceptionTrap.__init__s$rc V#rr r9&r __enter__ZExceptionTrap.__enter__ rc s(VP^,#rr*r=r>r typeZExceptionTrap.type}}Qrc s(VP^,#)rBr=r>r valueZExceptionTrap.valuerDrc s(VP^,#)rrBr=r>r tbZExceptionTrap.tbrDrc srV^,pT;'d\W P4pV'dWnV#rA)Z issubclassr8r*)r9r*rCZmatchess&* r __exit__ZExceptionTrap.__exit__s.{<<:dOO< $Mrc s,\VP4#r)r4rCr=r>r __bool__ZExceptionTrap.__bool__sDIIr_testc sNaaa\P!S4VVV3Rl4pV#)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 cs<\SP4;_uu_4pS!V/VBRRR4S!X4# +'giL;ir)r7r8)rrZtraprJr)r9s*, r wrapperZ%ExceptionTrap.raises..wrappers9t//4d%f%0; 0/s 8 A )r%Zwraps)r9r)rJrKsffd r raisesZExceptionTrap.raisess'&      rc sDVPV\PR7#)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 )rJ)rLoperatorZnot_)r9r)r:r passesZExceptionTrap.passes"s&{{4x}}{55r)r*r8)NNN)__name__ __module__ __qualname____firstlineno____doc__r*Z Exceptionr;r?ZpropertyrCrFrGrHrIr4rLrN__static_attributes__r rr r7r7suB H#,,%      D66rr7cs]tRtRtRtRtR#)suppressi8z A version of contextlib.suppress with decorator support. >>> @suppress(KeyError) ... def key_error(): ... {}[''] >>> key_error() r N)rOrPrQrRrSrTr rr rUrU8srrUcs0]tRtRtRtRRltRtRtRtR#) on_interruptiCa 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 sWnW nR#ractioncode)r9rXrYs""&r r;Zon_interrupt.__init__Xs   rc r<rr r=r>r r?Zon_interrupt.__enter__\r@rc sV\JgVPR8XdR#VPR8Xd\VP4VhVPR8H#)ZignoreNerrorrU)ZKeyboardInterruptrXZ SystemExitrY)r9ZexctypeZexcinstZexctbs&&&&r rHZon_interrupt.__exit___sE + +t{{h/F  [[G #TYY'W 4{{j((rrWN)rZrE) rOrPrQrRrSr;r?rHrTr rr rVrVCs()rrV)ii r)%Z __future__rr"r'r%rMrr,rr(r5Zsysr/Zurllib.requestrZtypingrZ version_infoZ backportsrr#rrrr&Z tarball_cwdr+r.rr1r-Zrobust_temp_dirr6r7rUZContextDecoratorrVr rr r[s3"    g!     *"*"Z:<ug&   ]]"##Hn6FG "$0n6n6bz""J$?$?!):..!)r