U pda @sjdZddlZddlZddlZddlmZddlmZddlm Z dddd d d d d ddddg Z Gdd d ej Z Gdd d ej Z Gdd d eZGdddZGdddee eZGdddee ZddZddZGddde ZGddde ZGd ddeZGd!ddeZGd"dde ZGd#d$d$ZGd%ddee ZGd&d d ee ZGd'd d e ZdS)(z4Utilities for with-statement contexts. See PEP 343.N)dequewraps MethodTypeasynccontextmanagercontextmanagerclosing nullcontextAbstractContextManagerAbstractAsyncContextManagerAsyncExitStackContextDecorator ExitStackredirect_stdoutredirect_stderrsuppressc@s2eZdZdZddZejddZeddZ dS) r z,An abstract base class for context managers.cCs|Sz0Return `self` upon entering the runtime context.selfrr /usr/lib/python3.8/contextlib.py __enter__sz AbstractContextManager.__enter__cCsdSz9Raise any exception triggered within the runtime context.Nrrexc_type exc_value tracebackrrr__exit__szAbstractContextManager.__exit__cCs|tkrt|ddStS)Nrr)r _collections_abc_check_methodsNotImplementedclsCrrr__subclasshook__sz'AbstractContextManager.__subclasshook__N) __name__ __module__ __qualname____doc__rabcabstractmethodr classmethodr%rrrrr s  c@s2eZdZdZddZejddZeddZ dS) r z9An abstract base class for asynchronous context managers.cs|Srrrrrr __aenter__'sz&AbstractAsyncContextManager.__aenter__csdSrrrrrr __aexit__+sz%AbstractAsyncContextManager.__aexit__cCs|tkrt|ddStS)Nr-r.)r rr r!r"rrrr%0s z,AbstractAsyncContextManager.__subclasshook__N) r&r'r(r)r-r*r+r.r,r%rrrrr #s  c@s eZdZdZddZddZdS)rzJA base class or mixin that enables context managers to work as decorators.cCs|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. rrrrr _recreate_cm;s zContextDecorator._recreate_cmcstfdd}|S)Nc s*||W5QRSQRXdSN)r/argskwdsfuncrrrinnerHs z(ContextDecorator.__call__..innerr)rr5r6rr4r__call__GszContextDecorator.__call__N)r&r'r(r)r/r7rrrrr8s c@seZdZdZddZdS)_GeneratorContextManagerBasezBShared functionality for @contextmanager and @asynccontextmanager.cCsJ||||_||||_|_|_t|dd}|dkr@t|j}||_dS)Nr))genr5r2r3getattrtyper))rr5r2r3docrrr__init__Rs    z%_GeneratorContextManagerBase.__init__N)r&r'r(r)r=rrrrr8Osr8c@s(eZdZdZddZddZddZdS) _GeneratorContextManagerz%Helper for @contextmanager decorator.cCs||j|j|jSr0) __class__r5r2r3rrrrr/fsz%_GeneratorContextManager._recreate_cmcCs<|`|`|`z t|jWStk r6tddYnXdSNzgenerator didn't yield)r2r3r5nextr9 StopIteration RuntimeErrorrrrrrls   z"_GeneratorContextManager.__enter__c Cs|dkr8zt|jWntk r,YdSXtdn|dkrF|}z|j|||Wntk r}z||k WYSd}~XYnttk r}z4||krWY&dS|tkr|j|krWY dSW5d}~XYn$td|krYdSYnXtddS)NFgenerator didn't stopz#generator didn't stop after throw())rAr9rBrCthrow __cause__sysexc_info)rr;valuerexcrrrrus.    z!_GeneratorContextManager.__exit__N)r&r'r(r)r/rrrrrrr>as r>c@s eZdZdZddZddZdS)_AsyncGeneratorContextManagerz Helper for @asynccontextmanager.cs6z|jIdHWStk r0tddYnXdSr@)r9 __anext__StopAsyncIterationrCrrrrr-sz(_AsyncGeneratorContextManager.__aenter__c s&|dkr>z|jIdHWntk r2YdSXtdn|dkrL|}z"|j|||IdHtdWntk r}z||k WYSd}~XYntk r}z:||krWY,dSt|ttfr|j|krWY dSW5d}~XYn0tk r }z||k rW5d}~XYnXdS)NrDz$generator didn't stop after athrow()F) r9rMrNrCathrow isinstancerBrG BaseException)rtyprJrrKrrrr.s.      z'_AsyncGeneratorContextManager.__aexit__N)r&r'r(r)r-r.rrrrrLsrLcstfdd}|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||Sr0)r>r1r5rrhelperszcontextmanager..helperrr5rTrrSrrscstfdd}|S)a@asynccontextmanager decorator. Typical usage: @asynccontextmanager async def some_async_generator(): try: yield finally: This makes this: async with some_async_generator() as : equivalent to this: try: = finally: cs t||Sr0)rLr1rSrrrTsz#asynccontextmanager..helperrrUrrSrrsc@s(eZdZdZddZddZddZdS) r a2Context 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 ||_dSr0thing)rrWrrrr=&szclosing.__init__cCs|jSr0rVrrrrr(szclosing.__enter__cGs|jdSr0)rWclose)rrIrrrr*szclosing.__exit__Nr&r'r(r)r=rrrrrrr sc@s(eZdZdZddZddZddZdS)_RedirectStreamNcCs||_g|_dSr0) _new_target _old_targets)r new_targetrrrr=2sz_RedirectStream.__init__cCs*|jtt|jtt|j|j|jSr0)r\appendr:rH_streamsetattrr[rrrrr7sz_RedirectStream.__enter__cCstt|j|jdSr0)r`rHr_r\poprexctypeexcinstexctbrrrr<sz_RedirectStream.__exit__)r&r'r(r_r=rrrrrrrZ.srZc@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) stdoutNr&r'r(r)r_rrrrr@s c@seZdZdZdZdS)rzCContext manager for temporarily redirecting stderr to another file.stderrNrgrrrrrPsc@s(eZdZdZddZddZddZdS) ra?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 ||_dSr0) _exceptions)r exceptionsrrrr=aszsuppress.__init__cCsdSr0rrrrrrdszsuppress.__enter__cCs|dk ot||jSr0) issubclassrirbrrrrgs zsuppress.__exit__NrYrrrrrVs c@sheZdZdZeddZeddZddZdd Zd d Z d d Z ddZ de _ ddZ dddZdS)_BaseExitStackz.A base class for ExitStack and AsyncExitStack.cCs t||Sr0rcmcm_exitrrr_create_exit_wrapperwsz#_BaseExitStack._create_exit_wrappercsfdd}|S)NcsdSr0rrrKtbr2callbackr3rr _exit_wrapper}sz8_BaseExitStack._create_cb_wrapper.._exit_wrapperrrtr2r3rurrsr_create_cb_wrapper{sz!_BaseExitStack._create_cb_wrappercCs t|_dSr0)r_exit_callbacksrrrrr=sz_BaseExitStack.__init__cCst|}|j|_t|_|S)z@Preserve the context stack by transferring it to a new instance.)r;rxr)r new_stackrrrpop_alls z_BaseExitStack.pop_allcCsBt|}z |j}Wntk r0||YnX||||S)aRegisters a callback with the standard __exit__ method signature. Can suppress exceptions the same way __exit__ method can. Also accepts any object with an __exit__ method (registering a call to the method instead of the object itself). )r;rAttributeError_push_exit_callback _push_cm_exitrexit_cb_type exit_methodrrrpushs   z_BaseExitStack.pushcCs(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. )r;rrr}rrn_cm_type_exitresultrrr enter_contexts   z_BaseExitStack.enter_contextcOst|dkr|^}}}nV|s&tdnHd|krZ|d}|^}}ddl}|jdtddntdt|d |j|f||}||_|||S) z\Registers an arbitrary callback and arguments. Cannot suppress exceptions. zBdescriptor 'callback' of '_BaseExitStack' object needs an argumentrtrN4Passing 'callback' as keyword argument is deprecated stacklevelz8callback expected at least 1 positional argument, got %drE) len TypeErrorrawarningswarnDeprecationWarningrw __wrapped__r|r2r3rrtrrurrrrts&      z_BaseExitStack.callback#($self, callback, /, *args, **kwds)cCs|||}||ddS)z;Helper to correctly register callbacks to __exit__ methods.TN)rpr|rrnrorurrrr}s z_BaseExitStack._push_cm_exitTcCs|j||fdSr0)rxr^)rrtis_syncrrrr|sz"_BaseExitStack._push_exit_callbackN)T)r&r'r(r) staticmethodrprwr=rzrrrt__text_signature__r}r|rrrrrlts  rlc@s(eZdZdZddZddZddZdS) 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|Sr0rrrrrrszExitStack.__enter__c s|ddk }tdfdd}d}d}|jr|j\}}|sHtz||r^d}d}d}Wq,t}||d|dd}|}Yq,Xq,|rz|dj} |dWn tk r| |d_YnX|o|S)NrrEcs4|j}||krdS|dks*|kr$q*|}q||_dSr0 __context__new_excold_exc exc_context frame_excrr_fix_exception_contextsz2ExitStack.__exit__.._fix_exception_contextFTNNN)rHrIrxraAssertionErrorrrQ) r exc_details received_excrsuppressed_exc pending_raisercbnew_exc_details fixed_ctxrrrrs4       zExitStack.__exit__cCs|ddddSz%Immediately unwind the context stack.N)rrrrrrXszExitStack.closeN)r&r'r(r)rrrXrrrrrs 1c@sfeZdZdZeddZeddZddZdd Zd d Z d e _ d dZ ddZ ddZ ddZdS)r aAsync context manager for dynamic management of a stack of exit callbacks. For example: async with AsyncExitStack() as stack: connections = [await stack.enter_async_context(get_connection()) for i in range(5)] # All opened connections will automatically be released at the # end of the async with statement, even if attempts to open a # connection later in the list raise an exception. cCs t||Sr0rrmrrr_create_async_exit_wrapper&sz)AsyncExitStack._create_async_exit_wrappercsfdd}|S)NcsIdHdSr0rrqrsrrru,sz>AsyncExitStack._create_async_cb_wrapper.._exit_wrapperrrvrrsr_create_async_cb_wrapper*sz'AsyncExitStack._create_async_cb_wrappercs.t|}|j}||IdH}||||S)zEnters the supplied async context manager. If successful, also pushes its __aexit__ method as a callback and returns the result of the __aenter__ method. N)r;r.r-_push_async_cm_exitrrrrenter_async_context0s  z"AsyncExitStack.enter_async_contextcCsDt|}z |j}Wn tk r2||dYnX||||S)a#Registers a coroutine function with the standard __aexit__ method signature. Can suppress exceptions the same way __aexit__ method can. Also accepts any object with an __aexit__ method (registering a call to the method instead of the object itself). F)r;r.r{r|rr~rrrpush_async_exit<s  zAsyncExitStack.push_async_exitcOst|dkr|^}}}nV|s&tdnHd|krZ|d}|^}}ddl}|jdtddntdt|d |j|f||}||_||d |S) zfRegisters an arbitrary coroutine function and arguments. Cannot suppress exceptions. rzMdescriptor 'push_async_callback' of 'AsyncExitStack' object needs an argumentrtrNrrzCpush_async_callback expected at least 1 positional argument, got %drEF) rrrarrrrrr|rrrrpush_async_callbackNs&      z"AsyncExitStack.push_async_callbackrcs|dddIdHdSr)r.rrrraclosekszAsyncExitStack.aclosecCs|||}||ddS)zLHelper to correctly register coroutine function to __aexit__ method.FN)rr|rrrrros z"AsyncExitStack._push_async_cm_exitcs|Sr0rrrrrr-uszAsyncExitStack.__aenter__c s|ddk }tdfdd}d}d}|jr|j\}}z0|rP||}n||IdH}|rnd}d}d}Wq,t} || d|dd}| }Yq,Xq,|rz|dj} |dWn tk r| |d_YnX|o|S)NrrEcs4|j}||krdS|dks*|kr$q*|}q||_dSr0rrrrrr~sz8AsyncExitStack.__aexit__.._fix_exception_contextFTr)rHrIrxrarrQ) rrrrrrrr cb_suppressrrrrrr.xs8        zAsyncExitStack.__aexit__N)r&r'r(r)rrrrrrrrrr-r.rrrrr s    c@s*eZdZdZd ddZddZddZdS) r aOContext 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 ||_dSr0 enter_result)rrrrrr=sznullcontext.__init__cCs|jSr0rrrrrrsznullcontext.__enter__cGsdSr0r)rexcinforrrrsznullcontext.__exit__)NrYrrrrr s )r)r*rHr collectionsr functoolsrtypesr__all__ABCr r objectrr8r>rLrrr rZrrrrlrr r rrrrsN    D .!!`E