B `dB @sdZddlZddlZddlZddlmZddlmZddddd d d d d g Zedg7Zej dddkrnej Z ne de fddiZ ddZddZGddde ZGdd d e ZGdddeZddZGddde ZGddde ZGd d d eZGd!d d eZGd"d d e Zej dd#kZer@d$d%Zd&d'Znd(d%Zd)d'Zydd*lmZWnek rzeZ Yn Xd+d,Z Gd-d d e Z!Gd.dde!Z"Gd/ddeZ#dS)0zAcontextlib2 - backports and enhancements to the contextlib moduleN)deque)wrapscontextmanagerclosing nullcontextAbstractContextManagerContextDecorator ExitStackredirect_stdoutredirect_stderrsuppress ContextStack)ABC __slots__cCs4||kr dS||x|jD]}t||qW|S)N)append __bases__ _classic_mro)CresultBrr;/tmp/pip-unpacked-wheel-hu9qqap1/pip/_vendor/contextlib2.pyrs   rcGspy |j}Wn"tk r,tt|g}YnXx<|D]4}x.|D]"}||jkr>|j|dkr^tSPq>WtSq4WdS)NT)__mro__AttributeErrortupler__dict__NotImplemented)rmethodsmromethodrrrr_check_methods$s    r#c@s2eZdZdZddZejddZeddZ dS) rz,An abstract base class for context managers.cCs|S)z0Return `self` upon entering the runtime context.r)selfrrr __enter__8sz AbstractContextManager.__enter__cCsdS)z9Raise any exception triggered within the runtime context.Nr)r$exc_type exc_value tracebackrrr__exit__<szAbstractContextManager.__exit__cCs|tkrt|ddStS)z.inner)r)r$r:r;r)r:r$r__call__iszContextDecorator.__call__N)r,r-r.r/r7r6r<rrrrrIs c@s0eZdZdZddZddZddZdd Zd S) _GeneratorContextManagerz%Helper for @contextmanager decorator.cCsJ||||_||||_|_|_t|dd}|dkr@t|j}||_dS)Nr/)genr:r8r9getattrtyper/)r$r:r8r9docrrr__init__ts    z!_GeneratorContextManager.__init__cCs||j|j|jS)N) __class__r:r8r9)r$rrrr6sz%_GeneratorContextManager._recreate_cmcCs,y t|jStk r&tdYnXdS)Nzgenerator didn't yield)nextr> StopIteration RuntimeError)r$rrrr%s z"_GeneratorContextManager.__enter__c Cs|dkr6yt|jWntk r*dSXtdn|dkrD|}y|j|||tdWntk r}z||k Sd}~XYn^tk r}z$||krdStr|j|krdSWdd}~XYntd|k rނYnXdS)Nzgenerator didn't stopz#generator didn't stop after throw()F) rDr>rErFthrow_HAVE_EXCEPTION_CHAINING __cause__sysexc_info)r$r@valuer(excrrrr)s,  z!_GeneratorContextManager.__exit__N)r,r-r.r/rBr6r%r)rrrrr=qs r=cstfdd}|S)a@contextmanager decorator. Typical usage: @contextmanager def some_generator(): try: yield finally: This makes this: with some_generator() as : equivalent to this: try: = finally: cs t||S)N)r=)r8r9)r:rrhelperszcontextmanager..helper)r)r:rOr)r:rrsc@s(eZdZdZddZddZddZdS) ra2Context to automatically close something at the end of a block. Code like this: with closing(.open()) as f: is equivalent to this: f = .open() try: finally: f.close() cCs ||_dS)N)thing)r$rPrrrrBszclosing.__init__cCs|jS)N)rP)r$rrrr%szclosing.__enter__cGs|jdS)N)rPclose)r$rLrrrr)szclosing.__exit__N)r,r-r.r/rBr%r)rrrrrsc@s(eZdZdZddZddZddZdS)_RedirectStreamNcCs||_g|_dS)N) _new_target _old_targets)r$ new_targetrrrrBsz_RedirectStream.__init__cCs*|jtt|jtt|j|j|jS)N)rTrr?rK_streamsetattrrS)r$rrrr%sz_RedirectStream.__enter__cCstt|j|jdS)N)rWrKrVrTpop)r$exctypeexcinstexctbrrrr)sz_RedirectStream.__exit__)r,r-r.rVrBr%r)rrrrrRsrRc@seZdZdZdZdS)r aAContext manager for temporarily redirecting stdout to another file. # How to send help() to stderr with redirect_stdout(sys.stderr): help(dir) # How to write help() to a file with open('help.txt', 'w') as f: with redirect_stdout(f): help(pow) stdoutN)r,r-r.r/rVrrrrr s c@seZdZdZdZdS)r zCContext manager for temporarily redirecting stderr to another file.stderrN)r,r-r.r/rVrrrrr sc@s(eZdZdZddZddZddZdS) r a?Context manager to suppress specified exceptions After the exception is suppressed, execution proceeds with the next statement following the with statement. with suppress(FileNotFoundError): os.remove(somefile) # Execution still resumes here if the file was already removed cGs ||_dS)N) _exceptions)r$ exceptionsrrrrB)szsuppress.__init__cCsdS)Nr)r$rrrr%,szsuppress.__enter__cCs|dk ot||jS)N) issubclassr^)r$rYrZr[rrrr)/s zsuppress.__exit__N)r,r-r.r/rBr%r)rrrrr s rcsfdd}|S)Ncs8x,|j}||krdS|dks$|kr&P|}qW||_dS)N) __context__)new_excold_exc exc_context) frame_excrr_fix_exception_context@sz3_make_context_fixer.._fix_exception_contextr)rerfr)rer_make_context_fixer?s rgcCs<y|dj}|dWn tk r6||d_YnXdS)NrG)ra BaseException) exc_details fixed_ctxrrr_reraise_with_existing_contextOs    rkcCsddS)NcSsdS)Nr)rbrcrrr[z%_make_context_fixer..r)rerrrrgZscCs|\}}}tddS)Nz!raise exc_type, exc_value, exc_tb)exec)rir&r'exc_tbrrrrk_s ) InstanceTypecCst|}|tkr|jS|S)N)r@rprC)objZobj_typerrr _get_typeksrrc@sXeZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ dS)r aContext manager for dynamic management of a stack of exit callbacks For example: with ExitStack() as stack: files = [stack.enter_context(open(fname)) for fname in filenames] # All opened files will automatically be closed at the end of # the with statement, even if attempts to open files later # in the list raise an exception cCs t|_dS)N)r_exit_callbacks)r$rrrrBszExitStack.__init__cCst|}|j|_t|_|S)z?Preserve the context stack by transferring it to a new instance)r@rsr)r$ new_stackrrrpop_alls zExitStack.pop_allcs"fdd}|_||dS)z:Helper to correctly register callbacks to __exit__ methodscsf|S)Nr)ri)cmcm_exitrr _exit_wrappersz.ExitStack._push_cm_exit.._exit_wrapperN)__self__push)r$rvrwrxr)rvrwr _push_cm_exitszExitStack._push_cm_exitcCsDt|}y |j}Wn tk r2|j|YnX||||S)aRegisters a callback with the standard __exit__ method signature Can suppress exceptions the same way __exit__ methods can. Also accepts any object with an __exit__ method (registering a call to the method instead of the object itself) )rrr)rrsrr{)r$exit_cb_type exit_methodrrrrzs   zExitStack.pushcs$fdd}|_||S)z\Registers an arbitrary callback and arguments. Cannot suppress exceptions. csdS)Nr)r&rNtb)r8callbackr9rrrxsz)ExitStack.callback.._exit_wrapper) __wrapped__rz)r$rr8r9rxr)r8rr9rrs zExitStack.callbackcCs(t|}|j}||}||||S)zEnters the supplied context manager If successful, also pushes its __exit__ method as a callback and returns the result of the __enter__ method. )rrr)r%r{)r$rv_cm_type_exitrrrr enter_contexts   zExitStack.enter_contextcCs|ddddS)z$Immediately unwind the context stackN)r))r$rrrrQszExitStack.closecCs|S)Nr)r$rrrr%szExitStack.__enter__c Gs|ddk }td}t|}d}d}x\|jr|j}y||rPd}d}d}Wq*t}||d|dd}|}Yq*Xq*W|rt||o|S)NrrGFT)NNN)rKrLrgrsrXrk) r$ri received_excrerfsuppressed_exc pending_raisecbnew_exc_detailsrrrr)s(   zExitStack.__exit__N) r,r-r.r/rBrur{rzrrrQr%r)rrrrr ss   cs8eZdZdZfddZddZddZdd ZZS) r z+Backwards compatibility alias for ExitStackcstdttt|dS)Nz*ContextStack has been renamed to ExitStack)r3r4r5superr rB)r$)rCrrrBszContextStack.__init__cCs ||S)N)rz)r$rrrr register_exitszContextStack.register_exitcOs|j|f||S)N)r)r$rr8r9rrrregisterszContextStack.registercCs|S)N)ru)r$rrrpreserveszContextStack.preserve) r,r-r.r/rBrrr __classcell__rr)rCrr s  c@s*eZdZdZd ddZddZddZdS) raMContext manager that does no additional processing. Used as a stand-in for a normal context manager, when a particular block of code is only sometimes used with a normal context manager: cm = optional_cm if condition else nullcontext() with cm: # Perform operation, using optional_cm if condition is True NcCs ||_dS)N) enter_result)r$rrrrrBsznullcontext.__init__cCs|jS)N)r)r$rrrr%sznullcontext.__enter__cGsdS)Nr)r$excinforrrr)sznullcontext.__exit__)N)r,r-r.r/rBr%r)rrrrrs )$r/r0rKr3 collectionsr functoolsr__all__ version_infor_abc_ABCABCMetaobjectrr#rrr=rrrRr r r rIrgrktypesrp ImportErrorr@rrr r rrrrrsL     (H"  q