9dZddlZddlZddlmZddlmZgdZedgz ZGddeZ Gd d e Z d Z Gd d eZ GddeZ Gdde ZGdde ZGddeZej"ddk\ZerdZdZndZdZ ddlmZdZGddeZGddeZy#e$reZY wxYw) zAcontextlib2 - backports and enhancements to the contextlib moduleN)dequewraps)contextmanagerclosingContextDecorator ExitStackredirect_stdoutredirect_stderrsuppress ContextStackc"eZdZdZdZdZdZy)rzJA base class or mixin that enables context managers to work as decorators.cVtjdt|jS)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_cmselfs 'lib/third_party/contextlib2/__init__.py refresh_cmzContextDecorator.refresh_cms%  J( *  ""c|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. rs rrzContextDecorator._recreate_cm"s  rc2tfd}|S)Nc`j5|i|cdddS#1swYyxYwN)r)argskwdsfuncrs rinnerz(ContextDecorator.__call__..inner/s(""$T*T*%$$s$-r)rr r!s`` r__call__zContextDecorator.__call__.s t +  + rN)__name__ __module__ __qualname____doc__rrr"rrrrrsP#" rrc(eZdZdZdZdZdZdZy)_GeneratorContextManagerz%Helper for @contextmanager decorator.c||i||_|||c|_|_|_t |dd}|t |j }||_y)Nr&)genr rrgetattrtyper&)rr rrdocs r__init__z!_GeneratorContextManager.__init__9sR&&*.d' 49didIt, ;t*$$C rcd|j|j|j|jSr) __class__r rrrs rrz%_GeneratorContextManager._recreate_cmGs#~~diiDII>>rc^ t|jS#t$r tdwxYw)Nzgenerator didn't yield)nextr* StopIteration RuntimeErrorrs r __enter__z"_GeneratorContextManager.__enter__Ms0 9> ! 978 8 9s,c|! t|jtd||} |jj |||td#t$rYywxYw#t$r}||ucYd}~Sd}~wt$r(}||urYd}~yt r|j |urYd}~yd}~wtjd|urYyxYw)Nzgenerator didn't stopz#generator didn't stop after throw()F) r2r*r4r3throw_HAVE_EXCEPTION_CHAINING __cause__sysexc_info)rr,value tracebackexcs r__exit__z!_GeneratorContextManager.__exit__Ss < <TXX##:;;} tUI6"#HII!  ! (%'' %< , 0F  <<>!$E12sFA(A% A"!A"% C.A71C7 CB& B&%B&&CN)r#r$r%r&r.rr5r@rrrr(r(6s/? 9 (rr(c.tfd}|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: ct||Sr)r()rrr s rhelperzcontextmanager..helpers'dD99rr)r rCs` rrr~s 8 4[:: Mrc"eZdZdZdZdZdZy)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() c||_yrthing)rrGs rr.zclosing.__init__s  rc|jSrrFrs rr5zclosing.__enter__s zzrc8|jjyr)rGclose)rr<s rr@zclosing.__exit__s rNr#r$r%r&r.r5r@rrrrrs rrc"eZdZdZdZdZdZy)_RedirectStreamNc ||_g|_yr) _new_target _old_targets)r new_targets rr.z_RedirectStream.__init__s%rc|jjtt|jt t|j|j |j Sr)rPappendr+r;_streamsetattrrOrs rr5z_RedirectStream.__enter__sC   dll!;<T\\4#3#34rcjtt|j|jj yr)rUr;rTrPpoprexctypeexcinstexctbs rr@z_RedirectStream.__exit__s!T\\4#4#4#8#8#:;r)r#r$r%rTr.r5r@rrrrMrMsG  ._fix_exception_contextsB%11 ')&+*B#*G &rr)rorps` r_make_context_fixerrqs *&%rc\ |dj}|d#t$r |d_wxYw)Nr7)rk BaseException) exc_details fixed_ctxs r_reraise_with_existing_contextrvs@ $A22Ia.  )2KN &  s+cdS)Ncyrr)rlrms rz%_make_context_fixer..srr)ros rrqrqs,,rc&|\}}}tdy)Nz!raise exc_type, exc_value, exc_tb)exec)rtexc_type exc_valueexc_tbs rrvrv"s&1#)V 12r) InstanceTypecDt|}|tur |jS|Sr)r,rr0)objobj_types r _get_typer.s"9 | #== rcFeZdZdZdZdZdZdZdZdZ dZ d Z d Z y ) 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 c"t|_yr)r_exit_callbacksrs rr.zExitStack.__init__As $wrcft|}|j|_t|_|S)z?Preserve the context stack by transferring it to a new instance)r,rr)r new_stacks rpop_allzExitStack.pop_allDs-DJL $($8$8 !$wrcDfd}|_|j|y)z:Helper to correctly register callbacks to __exit__ methodscg|Srr)rtcmcm_exits r _exit_wrapperz.ExitStack._push_cm_exit.._exit_wrapperMs2, , ,rN)__self__push)rrrrs `` r _push_cm_exitzExitStack._push_cm_exitKs -!#  - rct|} |j}|j|||S#t$r|jj |Y|SwxYw)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) )rr@rAttributeErrorrrS)rexit_cb_type exit_methods rrzExitStack.pushRs`T? 2"++K   t[ 1  .  ' ' -  .s -$AAcJfd}|_|j|S)z\Registers an arbitrary callback and arguments. Cannot suppress exceptions. ciyrr)r|r?tbrcallbackrs rrz)ExitStack.callback.._exit_wrapperks d #d #r) __wrapped__r)rrrrrs ``` rrzExitStack.callbackfs#  $%- ! - rczt|}|j}|j|}|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. )rr@r5r)rr_cm_type_exitresults r enter_contextzExitStack.enter_contextss=R=!!##B' 2u% rc*|jdddy)z$Immediately unwind the context stackN)r@rs rrJzExitStack.closes dD$'rc|Srrrs rr5zExitStack.__enter__s rc\|ddu}tjd}t|}d}d}|jr3|jj } ||rd}d}d}|jr3|r t ||xr|S#tj}||d|dd}|}YKxYw)Nrr7FT)NNN)r;r<rqrrWrv) rrt received_excrorpsuppressed_exc pending_raisecbnew_exc_detailss rr@zExitStack.__exit__s"1~T1 LLN1% !4Y!? ""%%))+B .{#%)N$)M"4K ""  *; 7.. ."%,,.&q'9;q>J $ - s  B)B+N) r#r$r%r&r.rrrrrrJr5r@rrrr r 5s4 '!(  (/rr c4eZdZdZfdZdZdZdZxZS)r z+Backwards compatibility alias for ExitStackc^tjdttt|y)Nz*ContextStack has been renamed to ExitStack)rrrsuperr r.)rr0s rr.zContextStack.__init__s" B( * lD*,rc$|j|Sr)r)rrs r register_exitzContextStack.register_exitsyy""rc.|j|g|i|Sr)r)rrrrs rregisterzContextStack.registerst}}X5555rc"|jSr)rrs rpreservezContextStack.preserves||~r) r#r$r%r&r.rrr __classcell__)r0s@rr r s5- #6r)r&r;r collectionsr functoolsr__all__objectrr(rrrMr r r version_infor9rqrvtypesrr ImportErrorr,r r rrrrsG  =N %v%PE/EPDf2++A.!3& - 3 " m/m/`9yIsB33B=<B=