B X9@sLdZddlZddlZddlmZddlmZddddd d d gZed g7ZGd ddeZ Gddde Z ddZ GdddeZ GdddeZ Gdd d e ZGdd d e ZGdd d eZejddkZerddZddZnddZddZyddlmZWnek reZYn Xdd ZGd!ddeZGd"d d eZdS)#zAcontextlib2 - backports and enhancements to the contextlib moduleN)deque)wrapscontextmanagerclosingContextDecorator ExitStackredirect_stdoutredirect_stderrsuppress ContextStackc@s(eZdZdZddZddZddZdS) rzJA base class or mixin that enables context managers to work as decorators.cCstdt|S)aReturns the context manager used to actually wrap the call to the decorated function. The default implementation just returns *self*. Overriding this method allows otherwise one-shot context managers like _GeneratorContextManager to support use as decorators via implicit recreation. DEPRECATED: refresh_cm was never added to the standard library's ContextDecorator API z2refresh_cm was never added to the standard library)warningswarnDeprecationWarning _recreate_cm)selfr*lib/python3.7/site-packages/contextlib2.py refresh_cms zContextDecorator.refresh_cmcCs|S)a6Return a recreated instance of self. Allows an otherwise one-shot context manager like _GeneratorContextManager to support use as a decorator via implicit recreation. This is a private interface just for _GeneratorContextManager. See issue #11647 for details. r)rrrrr"s zContextDecorator._recreate_cmcstfdd}|S)Nc s ||SQRXdS)N)r)argskwds)funcrrrinner/s z(ContextDecorator.__call__..inner)r)rrrr)rrr__call__.szContextDecorator.__call__N)__name__ __module__ __qualname____doc__rrrrrrrrs c@s0eZdZdZddZddZddZdd Zd S) _GeneratorContextManagerz%Helper for @contextmanager decorator.cCsJ||||_||||_|_|_t|dd}|dkr@t|j}||_dS)Nr)genrrrgetattrtyper)rrrrdocrrr__init__9s    z!_GeneratorContextManager.__init__cCs||j|j|jS)N) __class__rrr)rrrrrGsz%_GeneratorContextManager._recreate_cmcCs,y t|jStk r&tdYnXdS)Nzgenerator didn't yield)nextr StopIteration RuntimeError)rrrr __enter__Ms 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) r$rr%r&throw_HAVE_EXCEPTION_CHAINING __cause__sysexc_info)rr value tracebackexcrrr__exit__Ss,  z!_GeneratorContextManager.__exit__N)rrrrr"rr'r1rrrrr6s rcstfdd}|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)rr)rrrhelperszcontextmanager..helper)r)rr2r)rrr~sc@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)rr3rrrr"szclosing.__init__cCs|jS)N)r3)rrrrr'szclosing.__enter__cGs|jdS)N)r3close)rr-rrrr1szclosing.__exit__N)rrrrr"r'r1rrrrrsc@s(eZdZdZddZddZddZdS)_RedirectStreamNcCs||_g|_dS)N) _new_target _old_targets)r new_targetrrrr"sz_RedirectStream.__init__cCs*|jtt|jtt|j|j|jS)N)r7appendrr,_streamsetattrr6)rrrrr'sz_RedirectStream.__enter__cCstt|j|jdS)N)r;r,r:r7pop)rexctypeexcinstexctbrrrr1sz_RedirectStream.__exit__)rrrr:r"r'r1rrrrr5sr5c@seZdZdZdZdS)raAContext 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)rrrrr:rrrrrs c@seZdZdZdZdS)r zCContext manager for temporarily redirecting stderr to another file.stderrN)rrrrr:rrrrr 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 exceptionsrrrr"szsuppress.__init__cCsdS)Nr)rrrrr'szsuppress.__enter__cCs|dk ot||jS)N) issubclassrB)rr=r>r?rrrr1s zsuppress.__exit__N)rrrrr"r'r1rrrrr s csfdd}|S)Ncs8x,|j}||krdS|dks$|kr&P|}qW||_dS)N) __context__)new_excold_exc exc_context) frame_excrr_fix_exception_contextsz3_make_context_fixer.._fix_exception_contextr)rJrKr)rJr_make_context_fixers rLcCs<y|dj}|dWn tk r6||d_YnXdS)Nr()rF BaseException) exc_details fixed_ctxrrr_reraise_with_existing_contexts    rPcCsddS)NcSsdS)Nr)rGrHrrrsz%_make_context_fixer..r)rJrrrrLscCs|\}}}tddS)Nz!raise exc_type, exc_value, exc_tb)exec)rNexc_type exc_valueZexc_tbrrrrP"s ) InstanceTypecCst|}|tkr|jS|S)N)r rUr#)objZobj_typerrr _get_type.srWc@sXeZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ dS)raContext 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)rrrrr"AszExitStack.__init__cCst|}|j|_t|_|S)z?Preserve the context stack by transferring it to a new instance)r rXr)r new_stackrrrpop_allDs zExitStack.pop_allcs"fdd}|_||dS)z:Helper to correctly register callbacks to __exit__ methodscsf|S)Nr)rN)cmcm_exitrr _exit_wrapperMsz.ExitStack._push_cm_exit.._exit_wrapperN)__self__push)rr[r\r]r)r[r\r _push_cm_exitKszExitStack._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) )rWr1AttributeErrorrXr9r`)rexit_cb_type exit_methodrrrr_Rs   zExitStack.pushcs$fdd}|_||S)z\Registers an arbitrary callback and arguments. Cannot suppress exceptions. csdS)Nr)rSr0tb)rcallbackrrrr]ksz)ExitStack.callback.._exit_wrapper) __wrapped__r_)rrfrrr]r)rrfrrrffs 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. )rWr1r'r`)rr[_cm_type_exitresultrrr enter_contextss   zExitStack.enter_contextcCs|ddddS)z$Immediately unwind the context stackN)r1)rrrrr4szExitStack.closecCs|S)Nr)rrrrr'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)Nrr(FT)NNN)r,r-rLrXr<rP) rrN received_excrJrKsuppressed_exc pending_raisecbnew_exc_detailsrrrr1s(   zExitStack.__exit__N) rrrrr"rZr`r_rfrkr4r'r1rrrrr5s   cs8eZdZdZfddZddZddZdd ZZS) r z+Backwards compatibility alias for ExitStackcstdttt|dS)Nz*ContextStack has been renamed to ExitStack)r r rsuperr r")r)r#rrr"szContextStack.__init__cCs ||S)N)r_)rrfrrr register_exitszContextStack.register_exitcOs|j|f||S)N)rf)rrfrrrrrregisterszContextStack.registercCs|S)N)rZ)rrrrpreserveszContextStack.preserve) rrrrr"rrrsrt __classcell__rr)r#rr s  )rr,r collectionsr functoolsr__all__objectrrrrr5rr r version_infor*rLrPtypesrU ImportErrorr rWrr rrrrs8    (H"  p