U
e5da @ sf d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ ddddd d
d d d
dddg Z G dd d e j
Z G dd
d
e j
Z G dd d e
ZG dd dZG dd dee eZG dd dee Zdd Zdd ZG d d de ZG d d d e ZG d d deZG d d deZG d!d de ZG d"d# d#ZG d$d
d
ee ZG d%d d ee ZG d&d de Z dS )' N)dequewraps
MethodTypeasynccontextmanagercontextmanagerclosing nullcontextAbstractContextManagerAbstractAsyncContextManagerAsyncExitStackContextDecorator ExitStackredirect_stdoutredirect_stderrsuppressc @ s. e Zd Zdd Zejdd Zedd ZdS )r c C s | S N selfr r "/usr/lib64/python3.8/contextlib.py __enter__ s z AbstractContextManager.__enter__c C s d S r r r exc_type exc_value tracebackr r r __exit__ s z AbstractContextManager.__exit__c C s | t krt|ddS tS )Nr r )r _collections_abc_check_methodsNotImplementedclsCr r r __subclasshook__ s z'AbstractContextManager.__subclasshook__N) __name__
__module__ __qualname__r abcabstractmethodr classmethodr$ r r r r r s
c @ s. e Zd Zdd Zejdd Zedd ZdS )r c s | S r r r r r r
__aenter__' s z&AbstractAsyncContextManager.__aenter__c s d S r r r r r r __aexit__+ s z%AbstractAsyncContextManager.__aexit__c C s | t krt|ddS tS )Nr+ r, )r r r r r! r r r r$ 0 s
z,AbstractAsyncContextManager.__subclasshook__N) r% r& r' r+ r( r) r, r* r$ r r r r r # s
c @ s e Zd Zdd Zdd ZdS )r c C s | S r r r r r r _recreate_cm; s
z ContextDecorator._recreate_cmc s t fdd}|S )Nc
s* | |W 5 Q R S Q R X d S r )r- argskwdsfuncr r r innerH s
z(ContextDecorator.__call__..innerr )r r2 r3 r r1 r __call__G s zContextDecorator.__call__N)r% r& r' r- r4 r r r r r 8 s c @ s e Zd Zdd ZdS ) _GeneratorContextManagerBasec C sJ |||| _ ||| | _| _| _t|dd }|d kr@t| j}|| _d S )N__doc__)genr2 r/ r0 getattrtyper6 )r r2 r/ r0 docr r r __init__R s
z%_GeneratorContextManagerBase.__init__N)r% r& r' r; r r r r r5 O s r5 c @ s$ e Zd Zdd Zdd Zdd ZdS )_GeneratorContextManagerc C s | | j| j| jS r ) __class__r2 r/ r0 r r r r r- f s z%_GeneratorContextManager._recreate_cmc C s< | ` | `| `z t| jW S tk
r6 tdd Y nX d S Nzgenerator didn't yield)r/ r0 r2 nextr7
StopIteration RuntimeErrorr r r r r l s
z"_GeneratorContextManager.__enter__c
C s |d kr8zt | j W n tk
r, Y dS X tdn|d krF| }z| j||| W n tk
r } z||k W Y S d }~X Y nt tk
r } z4||krW Y &dS |tkr|j|krW Y
dS W 5 d }~X Y n$ t d |krY dS Y nX tdd S )NFgenerator didn't stop z#generator didn't stop after throw())r? r7 r@ rA throw __cause__sysexc_info)r r9 valuer excr r r r u s.
z!_GeneratorContextManager.__exit__N)r% r& r' r- r r r r r r r< a s r< c @ s e Zd Zdd Zdd ZdS ) _AsyncGeneratorContextManagerc s6 z| j I d H W S tk
r0 tdd Y nX d S r> )r7 __anext__StopAsyncIterationrA r r r r r+ s z(_AsyncGeneratorContextManager.__aenter__c
s& |d kr>z| j I d H W n tk
r2 Y d S X tdn|d krL| }z"| j |||I d H tdW n tk
r } z||k W Y S d }~X Y n tk
r } z:||krW Y ,dS t|ttfr|j|krW Y
dS W 5 d }~X Y n0 tk
r } z||k r W 5 d }~X Y nX d S )NrB z$generator didn't stop after athrow()F) r7 rK rL rA athrow
isinstancer@ rE
BaseException)r typrH r rI r r r r, s.
z'_AsyncGeneratorContextManager.__aexit__N)r% r& r' r+ r, r r r r rJ s rJ c s t fdd}|S )Nc s t | |S r )r< r. r2 r r helper s z contextmanager..helperr r2 rR r rQ r r s c s t fdd}|S )Nc s t | |S r )rJ r. rQ r r rR s z#asynccontextmanager..helperr rS r rQ r r s c @ s$ e Zd Zdd Zdd Zdd ZdS )r c C s
|| _ d S r thing)r rU r r r r; & s zclosing.__init__c C s | j S r rT r r r r r ( s zclosing.__enter__c G s | j d S r )rU close)r rG r r r r * s zclosing.__exit__Nr% r& r' r; r r r r r r r s c @ s( e Zd ZdZdd Zdd Zdd ZdS )_RedirectStreamNc C s || _ g | _d S r ) _new_target _old_targets)r
new_targetr r r r; 2 s z_RedirectStream.__init__c C s* | j tt| j tt| j| j | jS r )rZ appendr8 rF _streamsetattrrY r r r r r 7 s z_RedirectStream.__enter__c C s t t| j| j d S r )r^ rF r] rZ popr exctypeexcinstexctbr r r r < s z_RedirectStream.__exit__)r% r& r' r] r; r r r r r r rX . s rX c @ s e Zd ZdZdS )r stdoutNr% r& r' r] r r r r r @ s
c @ s e Zd ZdZdS )r stderrNre r r r r r P s c @ s$ e Zd Zdd Zdd Zdd ZdS )r c G s
|| _ d S r ) _exceptions)r
exceptionsr r r r; a s zsuppress.__init__c C s d S r r r r r r r d s zsuppress.__enter__c C s |d k ot || jS r )
issubclassrg r` r r r r g s
zsuppress.__exit__NrW r r r r r V s c @ sd e Zd Zedd Zedd Zdd Zdd Zd d
Zd d Z d
d Z
de
_ dd Z dddZ
dS )_BaseExitStackc C s
t || S r r cmcm_exitr r r _create_exit_wrapperw s z#_BaseExitStack._create_exit_wrapperc s fdd}|S )Nc s d S r r r rI tbr/ callbackr0 r r
_exit_wrapper} s z8_BaseExitStack._create_cb_wrapper.._exit_wrapperr rr r/ r0 rs r rq r _create_cb_wrapper{ s z!_BaseExitStack._create_cb_wrapperc C s t | _d S r )r _exit_callbacksr r r r r; s z_BaseExitStack.__init__c C s t | }| j|_t | _|S r )r9 rv r )r new_stackr r r pop_all s
z_BaseExitStack.pop_allc C sB t |}z
|j}W n tk
r0 | | Y nX | || |S r )r9 r AttributeError_push_exit_callback
_push_cm_exitr exit_cb_type exit_methodr r r push s
z_BaseExitStack.pushc C s( t |}|j}||}| || |S r )r9 r r r{ r rl _cm_type_exitresultr r r
enter_context s
z _BaseExitStack.enter_contextc O s t | dkr| ^}}} nV| s&tdnHd|krZ|d}| ^}} dd l}|jdtdd ntdt | d |j|f| |}||_|| |S ) N zBdescriptor 'callback' of '_BaseExitStack' object needs an argumentrr r 4Passing 'callback' as keyword argument is deprecated
stacklevelz8callback expected at least 1 positional argument, got %drC ) len TypeErrorr_ warningswarnDeprecationWarningru __wrapped__rz r/ r0 r rr r rs r r r rr s&
z_BaseExitStack.callback#($self, callback, /, *args, **kwds)c C s | ||}| |d d S )NT)rn rz r rl rm rs r r r r{ s z _BaseExitStack._push_cm_exitTc C s | j ||f d S r )rv r\ )r rr is_syncr r r rz s z"_BaseExitStack._push_exit_callbackN)T)r% r& r' staticmethodrn ru r; rx r r rr __text_signature__r{ rz r r r r rj t s
rj c @ s$ e Zd Zdd Zdd Zdd ZdS )r c C s | S r r r r r r r s zExitStack.__enter__c
s |d d k }t d fdd}d}d}| jr| j \}}z|| rVd}d}d}W q, t }||d |d d}|}Y q,X q,|rz|d j} |d W n tk
r | |d _ Y nX |o|S )Nr rC c s4 | j }||krd S |d ks*| kr$q*|} q || _ d S r __context__new_excold_exc exc_context frame_excr r _fix_exception_context s z2ExitStack.__exit__.._fix_exception_contextFTNNNrF rG rv r_ r rO )
r exc_details received_excr suppressed_exc
pending_raiser cbnew_exc_details fixed_ctxr r r r s2
zExitStack.__exit__c C s | d d d d S r )r r r r r rV s zExitStack.closeN)r% r& r' r r rV r r r r r s 1c @ sb e Zd Zedd Zedd Zdd Zdd Zd d
Zd e_ d d
Z
dd Z dd Z dd Z
dS )r
c C s
t || S r r rk r r r _create_async_exit_wrapper& s z)AsyncExitStack._create_async_exit_wrapperc s fdd}|S )Nc s I d H d S r r ro rq r r rs , s z>AsyncExitStack._create_async_cb_wrapper.._exit_wrapperr rt r rq r _create_async_cb_wrapper* s z'AsyncExitStack._create_async_cb_wrapperc s. t |}|j}||I d H }| || |S r )r9 r, r+ _push_async_cm_exitr r r r enter_async_context0 s
z"AsyncExitStack.enter_async_contextc C sD t |}z
|j}W n tk
r2 | |d Y nX | || |S NF)r9 r, ry rz r r| r r r push_async_exit< s
z AsyncExitStack.push_async_exitc O s t | dkr| ^}}} nV| s&tdnHd|krZ|d}| ^}} dd l}|jdtdd ntdt | d |j|f| |}||_||d |S )
Nr zMdescriptor 'push_async_callback' of 'AsyncExitStack' object needs an argumentrr r r r zCpush_async_callback expected at least 1 positional argument, got %drC F) r r r_ r r r r r rz r r r r push_async_callbackN s&
z"AsyncExitStack.push_async_callbackr c s | d d d I d H d S r )r, r r r r aclosek s zAsyncExitStack.aclosec C s | ||}| |d d S r )r rz r r r r r o s z"AsyncExitStack._push_async_cm_exitc s | S r r r r r r r+ u s zAsyncExitStack.__aenter__c s |d d k }t d fdd}d}d}| jr| j \}}z0|rP|| }n|| I d H }|rnd}d}d}W q, t } || d |d d}| }Y q,X q,|rz|d j}
|d W n tk
r |
|d _ Y nX |o|S )Nr rC c s4 | j }||krd S |d ks*| kr$q*|} q || _ d S r r r r r r r ~ s z8AsyncExitStack.__aexit__.._fix_exception_contextFTr r ) r r r r r r r r cb_suppressr r r r r r, x s8
zAsyncExitStack.__aexit__N)r% r& r' r r r r r r r r r r+ r, r r r r r
s
c @ s& e Zd ZdddZdd Zdd ZdS ) r
Nc C s
|| _ d S r enter_result)r r r r r r; s znullcontext.__init__c C s | j S r r r r r r r s znullcontext.__enter__c G s d S r r )r excinfor r r r s znullcontext.__exit__)NrW r r r r r
s
) r( rF r collectionsr functoolsr typesr __all__ABCr r objectr r5 r< rJ r r r rX r r r rj r r
r
r r r r sL D .!! `E