Your IP : 216.73.216.25
U
i�f�a�@sjdZddlZddlZddlZddlmZddlmZddlm Z dddd d
ddd
ddddgZ
Gdd
�d
ej�ZGdd�dej�Z
Gdd
�d
e�ZGdd�d�ZGdd�deee�ZGdd�dee
�Zdd�Zdd�ZGdd�de�ZGdd�de�ZGd d�de�ZGd!d�de�ZGd"d�de�ZGd#d$�d$�ZGd%d�dee�ZGd&d�dee
�ZGd'd �d e�ZdS)(z4Utilities for with-statement contexts. See PEP 343.�N)�deque��wraps��
MethodType�asynccontextmanager�contextmanager�closing�nullcontext�AbstractContextManager�AbstractAsyncContextManager�AsyncExitStack�ContextDecorator� ExitStack�redirect_stdout�redirect_stderr�suppressc@s2eZdZdZdd�Zejdd��Zedd��Z dS) rz,An abstract base class for context managers.cCs|S�z0Return `self` upon entering the runtime context.���selfrr�//opt/alt/python38/lib64/python3.8/contextlib.py� __enter__sz AbstractContextManager.__enter__cCsdS�z9Raise any exception triggered within the runtime context.Nr�r�exc_type� exc_value� tracebackrrr�__exit__szAbstractContextManager.__exit__cCs|tkrt�|dd�StS)Nrr)r�_collections_abc�_check_methods�NotImplemented��cls�Crrr�__subclasshook__sz'AbstractContextManager.__subclasshook__N)
�__name__�
__module__�__qualname__�__doc__r�abc�abstractmethodr�classmethodr%rrrrrs
c@s2eZdZdZdd�Zejdd��Zedd��Z dS) rz9An abstract base class for asynchronous context managers.c�s|Srrrrrr�
__aenter__'sz&AbstractAsyncContextManager.__aenter__c�sdSrrrrrr� __aexit__+sz%AbstractAsyncContextManager.__aexit__cCs|tkrt�|dd�StS)Nr-r.)rrr r!r"rrrr%0s
�z,AbstractAsyncContextManager.__subclasshook__N)
r&r'r(r)r-r*r+r.r,r%rrrrr#s
c@s eZdZdZdd�Zdd�ZdS)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_cmcst����fdd��}|S)Nc
s*�����||�W5QR�SQRXdS�N)r/��args�kwds��funcrrr�innerHs
z(ContextDecorator.__call__.<locals>.innerr)rr5r6rr4r�__call__GszContextDecorator.__call__N)r&r'r(r)r/r7rrrrr8sc@seZdZdZdd�ZdS)�_GeneratorContextManagerBasezBShared functionality for @contextmanager and @asynccontextmanager.cCsJ|||�|_||||_|_|_t|dd�}|dkr@t|�j}||_dS)Nr))�genr5r2r3�getattr�typer))rr5r2r3�docrrr�__init__Rs
z%_GeneratorContextManagerBase.__init__N)r&r'r(r)r=rrrrr8Osr8c@s(eZdZdZdd�Zdd�Zdd�ZdS) �_GeneratorContextManagerz%Helper for @contextmanager decorator.cCs|�|j|j|j�Sr0)� __class__r5r2r3rrrrr/fsz%_GeneratorContextManager._recreate_cmcCs<|`|`|`zt|j�WStk
r6td�d�YnXdS�Nzgenerator didn't yield)r2r3r5�nextr9�
StopIteration�RuntimeErrorrrrrrls
z"_GeneratorContextManager.__enter__c
Cs|dkr8zt|j�Wntk
r,YdSXtd��n�|dkrF|�}z|j�|||�Wn�tk
r�}z||k WY�Sd}~XYnttk
r�}z4||kr�WY�&dS|tkr�|j|kr�WY�
dS�W5d}~XYn$t��d|kr�YdS�YnXtd��dS)NF�generator didn't stop�z#generator didn't stop after throw())rAr9rBrC�throw� __cause__�sys�exc_info)rr;�valuer�excrrrrus.
z!_GeneratorContextManager.__exit__N)r&r'r(r)r/rrrrrrr>as r>c@s eZdZdZdd�Zdd�ZdS)�_AsyncGeneratorContextManagerz Helper for @asynccontextmanager.c�s6z|j��IdHWStk
r0td�d�YnXdSr@)r9� __anext__�StopAsyncIterationrCrrrrr-�sz(_AsyncGeneratorContextManager.__aenter__c
�s&|dkr>z|j��IdHWntk
r2YdSXtd��n�|dkrL|�}z"|j�|||�IdHtd��Wn�tk
r�}z||k WY�Sd}~XYn�tk
r�}z:||kr�WY�,dSt|ttf�r�|j|kr�WY�
dS�W5d}~XYn0tk
�r }z||k �r�W5d}~XYnXdS)NrDz$generator didn't stop after athrow()F) r9rMrNrC�athrow�
isinstancerBrG�
BaseException)r�typrJrrKrrrr.�s.
z'_AsyncGeneratorContextManager.__aexit__N)r&r'r(r)r-r.rrrrrL�srLcst���fdd��}|S)a�@contextmanager decorator.
Typical usage:
@contextmanager
def some_generator(<arguments>):
<setup>
try:
yield <value>
finally:
<cleanup>
This makes this:
with some_generator(<arguments>) as <variable>:
<body>
equivalent to this:
<setup>
try:
<variable> = <value>
<body>
finally:
<cleanup>
cst�||�Sr0)r>r1�r5rr�helper�szcontextmanager.<locals>.helperr�r5rTrrSrr�scst���fdd��}|S)a�@asynccontextmanager decorator.
Typical usage:
@asynccontextmanager
async def some_async_generator(<arguments>):
<setup>
try:
yield <value>
finally:
<cleanup>
This makes this:
async with some_async_generator(<arguments>) as <variable>:
<body>
equivalent to this:
<setup>
try:
<variable> = <value>
<body>
finally:
<cleanup>
cst�||�Sr0)rLr1rSrrrTsz#asynccontextmanager.<locals>.helperrrUrrSrr�sc@s(eZdZdZdd�Zdd�Zdd�ZdS) r a2Context to automatically close something at the end of a block.
Code like this:
with closing(<module>.open(<arguments>)) as f:
<block>
is equivalent to this:
f = <module>.open(<arguments>)
try:
<block>
finally:
f.close()
cCs
||_dSr0��thing)rrWrrrr=&szclosing.__init__cCs|jSr0rVrrrrr(szclosing.__enter__cGs|j��dSr0)rW�close)rrIrrrr*szclosing.__exit__N�r&r'r(r)r=rrrrrrr sc@s(eZdZdZdd�Zdd�Zdd�ZdS)�_RedirectStreamNcCs||_g|_dSr0)�_new_target�_old_targets)r�
new_targetrrrr=2sz_RedirectStream.__init__cCs*|j�tt|j��tt|j|j�|jSr0)r\�appendr:rH�_stream�setattrr[rrrrr7sz_RedirectStream.__enter__cCstt|j|j���dSr0)r`rHr_r\�pop�r�exctype�excinst�exctbrrrr<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)
�stdoutN�r&r'r(r)r_rrrrr@sc@seZdZdZdZdS)rzCContext manager for temporarily redirecting stderr to another file.�stderrNrgrrrrrPsc@s(eZdZdZdd�Zdd�Zdd�ZdS) 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||j�Sr0)�
issubclassrirbrrrrgs
zsuppress.__exit__NrYrrrrrVs
c@sheZdZdZedd��Zedd��Zdd�Zdd �Zd
d�Z dd
�Z
dd�Zde_dd�Z
ddd�ZdS)�_BaseExitStackz.A base class for ExitStack and AsyncExitStack.cCs
t||�Sr0r��cm�cm_exitrrr�_create_exit_wrapperwsz#_BaseExitStack._create_exit_wrappercs���fdd�}|S)Ncs����dSr0r�rrK�tb�r2�callbackr3rr�
_exit_wrapper}sz8_BaseExitStack._create_cb_wrapper.<locals>._exit_wrapperr�rtr2r3rurrsr�_create_cb_wrapper{sz!_BaseExitStack._create_cb_wrappercCst�|_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_stackrrr�pop_all�s
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;r�AttributeError�_push_exit_callback�
_push_cm_exit�r�exit�_cb_type�exit_methodrrr�push�s
z_BaseExitStack.pushcCs(t|�}|j}|�|�}|�||�|S)z�Enters 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�_exit�resultrrr�
enter_context�s
z_BaseExitStack.enter_contextcOs�t|�dkr|^}}}nV|s&td��nHd|krZ|�d�}|^}}ddl}|jdtdd�ntdt|�d ��|j|f|�|�}||_|�|�|S)
z\Registers an arbitrary callback and arguments.
Cannot suppress exceptions.
�zBdescriptor 'callback' of '_BaseExitStack' object needs an argumentrtrN�4Passing 'callback' as keyword argument is deprecated��
stacklevelz8callback expected at least 1 positional argument, got %drE) �len� TypeErrorra�warnings�warn�DeprecationWarningrw�__wrapped__r|�r2r3rrtr�rurrrrt�s&
�
�
z_BaseExitStack.callback�#($self, callback, /, *args, **kwds)cCs|�||�}|�|d�dS)z;Helper to correctly register callbacks to __exit__ methods.TN)rpr|�rrnrorurrrr}�sz_BaseExitStack._push_cm_exitTcCs|j�||f�dSr0)rxr^)rrt�is_syncrrrr|�sz"_BaseExitStack._push_exit_callbackN)T)r&r'r(r)�staticmethodrprwr=rzr�r�rt�__text_signature__r}r|rrrrrlts
rlc@s(eZdZdZdd�Zdd�Zdd�ZdS) ra�Context 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|Sr0rrrrrr�szExitStack.__enter__c
s�|ddk }t��d��fdd�}d}d}|jr�|j��\}}|sHt�z||�r^d}d}d}Wq,t��}||d|d�d}|}Yq,Xq,|r�z|dj} |d�Wn tk
r�| |d_�YnX|o�|S)NrrEcs4|j}||krdS|dks*|�kr$q*|}q||_dSr0��__context__��new_exc�old_exc�exc_context�� frame_excrr�_fix_exception_context�sz2ExitStack.__exit__.<locals>._fix_exception_contextFT�NNN)rHrIrxra�AssertionErrorr�rQ)
r�exc_details�received_excr��suppressed_exc�
pending_raiser��cb�new_exc_details� fixed_ctxrr�rr�s4
zExitStack.__exit__cCs|�ddd�dS�z%Immediately unwind the context stack.N)rrrrrrXszExitStack.closeN)r&r'r(r)rrrXrrrrr�s
1c@sfeZdZdZedd��Zedd��Zdd�Zdd �Zd
d�Z de _
d
d�Zdd�Zdd�Z
dd�ZdS)r
a�Async 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_wrappercs���fdd�}|S)Nc�s����IdHdSr0rrqrsrrru,sz>AsyncExitStack._create_async_cb_wrapper.<locals>._exit_wrapperrrvrrsr�_create_async_cb_wrapper*sz'AsyncExitStack._create_async_cb_wrapperc�s.t|�}|j}|�|�IdH}|�||�|S)z�Enters 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_exitr�rrr�enter_async_context0s
z"AsyncExitStack.enter_async_contextcCsDt|�}z
|j}Wn tk
r2|�|d�YnX|�||�|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|r�r~rrr�push_async_exit<s
zAsyncExitStack.push_async_exitcOs�t|�dkr|^}}}nV|s&td��nHd|krZ|�d�}|^}}ddl}|jdtdd�ntdt|�d ��|j|f|�|�}||_|�|d
�|S)zfRegisters an arbitrary coroutine function and arguments.
Cannot suppress exceptions.
r�zMdescriptor 'push_async_callback' of 'AsyncExitStack' object needs an argumentrtrNr�r�zCpush_async_callback expected at least 1 positional argument, got %drEF) r�r�rar�r�r�r�r�r|r�rrr�push_async_callbackNs&
�
�z"AsyncExitStack.push_async_callbackr�c�s|�ddd�IdHdSr�)r.rrrr�aclosekszAsyncExitStack.aclosecCs|�||�}|�|d�dS)zLHelper to correctly register coroutine function to __aexit__
method.FN)r�r|r�rrrr�osz"AsyncExitStack._push_async_cm_exitc�s|Sr0rrrrrr-uszAsyncExitStack.__aenter__c�s�|ddk }t��d��fdd�}d}d}|jr�|j��\}}z0|rP||�}n||�IdH}|rnd}d}d}Wq,t��} || d|d�d}| }Yq,Xq,|r�z|dj}
|d�Wn tk
r�|
|d_�YnX|o�|S)NrrEcs4|j}||krdS|dks*|�kr$q*|}q||_dSr0r�r�r�rrr�~sz8AsyncExitStack.__aexit__.<locals>._fix_exception_contextFTr�)rHrIrxrar�rQ)rr�r�r�r�r�r�r��cb_suppressr�r�rr�rr.xs8
zAsyncExitStack.__aexit__N)r&r'r(r)r�r�r�r�r�r�r�r�r�r-r.rrrrr
s
c@s*eZdZdZd dd�Zdd�Zdd�ZdS)
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)rr�rrrr=�sznullcontext.__init__cCs|jSr0r�rrrrr�sznullcontext.__enter__cGsdSr0r)r�excinforrrr�sznullcontext.__exit__)NrYrrrrr
�s
)r)r*rHr�collectionsr� functoolsr�typesr�__all__�ABCrr�objectrr8r>rLrrr rZrrrrlrr
r
rrrr�<module>sN��D�.!!`E
?>