B
] @ sZ d dl Z d dlZd dlZd dlmZ d dlmZ dddddd d
d d d
ddg ZG dd de jZ G dd d e jZ
G dd d e Z G dd dZ
G dd de
e e ZG dd de
e
Zdd Zdd ZG dd de ZG d d d e ZG d d
d
eZG d d deZG d d de ZG d!d" d"ZG d#d d ee ZG d$d
d
ee
ZG d%d de ZdS )& N)deque)wrapsasynccontextmanagercontextmanagerclosing nullcontextAbstractContextManagerAbstractAsyncContextManagerAsyncExitStackContextDecorator ExitStackredirect_stdoutredirect_stderrsuppressc @ s. e Zd Zdd Zejdd Zedd ZdS )r c C s | S )N )selfr r //opt/alt/python37/lib64/python3.7/contextlib.py __enter__ s z AbstractContextManager.__enter__c C s d S )Nr )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_methodsNotImplemented)clsCr 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 )Nr )r r r r
__aenter__& s z&AbstractAsyncContextManager.__aenter__c s d S )Nr )r 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 r / 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 )Nr )r r r r _recreate_cm: s
z ContextDecorator._recreate_cmc s t fdd}|S )Nc s | |S Q R X d S )N)r& )argskwds)funcr r r innerG s
z(ContextDecorator.__call__..inner)r )r r) r* r )r) r r __call__F s zContextDecorator.__call__N)r r r r& r+ r r r r r 7 s c @ s e Zd Zdd ZdS ) _GeneratorContextManagerBasec C sJ |||| _ ||| | _| _| _t|dd }|d kr@t| j}|| _d S )N__doc__)genr) r' r( getattrtyper- )r r) r' r( docr r r __init__Q s
z%_GeneratorContextManagerBase.__init__N)r r r r2 r r r r r, N s r, c @ s$ e Zd Zdd Zdd Zdd ZdS )_GeneratorContextManagerc C s | | j| j| jS )N) __class__r) r' r( )r r r r r& e s z%_GeneratorContextManager._recreate_cmc C s: | ` | `| `y
t| jS tk
r4 tdd Y nX d S )Nzgenerator didn't yield)r' r( r) nextr.
StopIteration RuntimeError)r r r r r k s
z"_GeneratorContextManager.__enter__c
C s |d kr6yt | j W n tk
r* dS X tdn|d krD| }y| j||| W n tk
r~ } z||k S d }~X Y nf tk
r } z(||krdS |tkr|j|krdS W d d }~X Y n" t d |krdS Y nX tdd S )NFzgenerator didn't stop z#generator didn't stop after throw())r5 r. r6 r7 throw __cause__sysexc_info)r r0 valuer excr r r r t s.
z!_GeneratorContextManager.__exit__N)r r r r& r r r r r r r3 ` s r3 c @ s e Zd Zdd Zdd ZdS ) _AsyncGeneratorContextManagerc s4 y| j I d H S tk
r. tdd Y nX d S )Nzgenerator didn't yield)r. __anext__StopAsyncIterationr7 )r r r r r$ s z(_AsyncGeneratorContextManager.__aenter__c
s |d kr r r r r% s.
z'_AsyncGeneratorContextManager.__aexit__N)r r r r$ r% r r r r r? s r? c s t fdd}|S )Nc s t | |S )N)r3 )r' r( )r) r r helper s z contextmanager..helper)r )r) rF r )r) r r s c s t fdd}|S )Nc s t | |S )N)r? )r' r( )r) r r rF s z#asynccontextmanager..helper)r )r) rF r )r) r r s c @ s$ e Zd Zdd Zdd Zdd ZdS )r c C s
|| _ d S )N)thing)r rG r r r r2 % s zclosing.__init__c C s | j S )N)rG )r r r r r ' s zclosing.__enter__c G s | j d S )N)rG close)r r< r r r r ) s zclosing.__exit__N)r r r r2 r r r r r r r s c @ s( e Zd ZdZdd Zdd Zdd ZdS )_RedirectStreamNc C s || _ g | _d S )N) _new_target _old_targets)r
new_targetr r r r2 1 s z_RedirectStream.__init__c C s* | j tt| j tt| j| j | jS )N)rK appendr/ r; _streamsetattrrJ )r r r r r 6 s z_RedirectStream.__enter__c C s t t| j| j d S )N)rO r; rN rK pop)r exctypeexcinstexctbr r r r ; s z_RedirectStream.__exit__)r r r rN r2 r r r r r r rI - s rI c @ s e Zd ZdZdS )r
stdoutN)r r r rN r r r r r
? s
c @ s e Zd ZdZdS )r stderrN)r r r rN r r r r r O s c @ s$ e Zd Zdd Zdd Zdd ZdS )r c G s
|| _ d S )N) _exceptions)r
exceptionsr r r r2 ` s zsuppress.__init__c C s d S )Nr )r r r r r c s zsuppress.__enter__c C s |d k ot || jS )N)
issubclassrV )r rQ rR rS r r r r f s
zsuppress.__exit__N)r r r r2 r r r r r r r U s c @ s^ e Zd Zedd Zedd Zdd Zdd Zd d
Zd d Z d
d Z
dd Z dddZ dS )_BaseExitStackc s fdd}|S )Nc s | ||S )Nr )r r> tb)cmcm_exitr r
_exit_wrapperx s z:_BaseExitStack._create_exit_wrapper.._exit_wrapperr )r[ r\ r] r )r[ r\ r _create_exit_wrapperv s z#_BaseExitStack._create_exit_wrapperc s ^ fdd}|S )Nc s d S )Nr )r r> rZ )r' callbackr( r r r] s z8_BaseExitStack._create_cb_wrapper.._exit_wrapperr )r' r( r] r )r' r_ r( r _create_cb_wrapper| s z!_BaseExitStack._create_cb_wrapperc C s t | _d S )N)r _exit_callbacks)r r r r r2 s z_BaseExitStack.__init__c C s t | }| j|_t | _|S )N)r0 ra r )r new_stackr r r pop_all s
z_BaseExitStack.pop_allc C sB t |}y
|j}W n tk
r0 | | Y nX | || |S )N)r0 r AttributeError_push_exit_callback
_push_cm_exit)r exit_cb_type exit_methodr r r push s
z_BaseExitStack.pushc C s( t |}|j}||}| || |S )N)r0 r r rf )r r[ _cm_type_exitresultr r r
enter_context s
z _BaseExitStack.enter_contextc O s| t | dkr| ^}}} n>| s&tdn0d|krB|d}| ^}} ntdt | d |j|f| |}||_|| |S )N zBdescriptor 'callback' of '_BaseExitStack' object needs an argumentr_ z8callback expected at least 1 positional argument, got %dr8 )len TypeErrorrP r` __wrapped__re )r' r( r r_ r] r r r r_ s
z_BaseExitStack.callbackc C s" | ||}||_| |d d S )NT)r^ __self__re )r r[ r\ r] r r r rf s z _BaseExitStack._push_cm_exitTc C s | j ||f d S )N)ra rM )r r_ is_syncr r r re s z"_BaseExitStack._push_exit_callbackN)T)
r r r staticmethodr^ r` r2 rc rj rn r_ rf re r r r r rY s s rY c @ s$ e Zd Zdd Zdd Zdd ZdS )r c C s | S )Nr )r r r r r s zExitStack.__enter__c
s |d d k }t d fdd}d}d}x`| jr| j \}}y|| rXd}d}d}W q. t }||d |d d}|}Y q.X q.W |ry|d j} |d W n tk
r | |d _ Y nX |o|S )Nr r8 c s8 x,| j }||krd S |d ks$| kr&P |} qW || _ d S )N) __context__)new_excold_exc exc_context) frame_excr r _fix_exception_context s z2ExitStack.__exit__.._fix_exception_contextFT)NNN)r; r< ra rP rv rD )
r exc_details received_excr{ suppressed_exc
pending_raisert cbnew_exc_details fixed_ctxr )rz r r s2
zExitStack.__exit__c C s | d d d d S )N)r )r r r r rH s zExitStack.closeN)r r r r r rH r r r r r s 1c @ s\ e Zd Zedd Zedd Zdd Zdd Zd d
Zd d Z d
d Z
dd Z dd Z dS )r
c s fdd}|S )Nc s | ||I d H S )Nr )r r> rZ )r[ r\ r r r] ' s z@AsyncExitStack._create_async_exit_wrapper.._exit_wrapperr )r[ r\ r] r )r[ r\ r _create_async_exit_wrapper% s z)AsyncExitStack._create_async_exit_wrapperc s ^ fdd}|S )Nc s I d H d S )Nr )r r> rZ )r' r_ r( r r r] . s z>AsyncExitStack._create_async_cb_wrapper.._exit_wrapperr )r' r( r] r )r' r_ r( r _create_async_cb_wrapper+ s z'AsyncExitStack._create_async_cb_wrapperc s. t |}|j}||I d H }| || |S )N)r0 r% r$ _push_async_cm_exit)r r[ rk rl rm r r r enter_async_context2 s
z"AsyncExitStack.enter_async_contextc C sD t |}y
|j}W n tk
r2 | |d Y nX | || |S )NF)r0 r% rd re r )r rg rh ri r r r push_async_exit> s
z AsyncExitStack.push_async_exitc O s~ t | dkr| ^}}} n>| s&tdn0d|krB|d}| ^}} ntdt | d |j|f| |}||_||d |S )Nro zMdescriptor 'push_async_callback' of 'AsyncExitStack' object needs an argumentr_ zCpush_async_callback expected at least 1 positional argument, got %dr8 F)rp rq rP r rr re )r' r( r r_ r] r r r push_async_callbackP s
z"AsyncExitStack.push_async_callbackc s | d d d I d H d S )N)r% )r r r r aclosei s zAsyncExitStack.aclosec C s" | ||}||_| |d d S )NF)r rs re )r r[ r\ r] r r r r m s z"AsyncExitStack._push_async_cm_exitc s | S )Nr )r r r r r$ t s zAsyncExitStack.__aenter__c s |d d k }t d fdd}d}d}xx| jr| j \}}y0|rR|| }n|| I d H }|rpd}d}d}W q. t } || d |d d}| }Y q.X q.W |ry|d j}
|d W n tk
r |
|d _ Y nX |o|S )Nr r8 c s8 x,| j }||krd S |d ks$| kr&P |} qW || _ d S )N)rv )rw rx ry )rz r r r{ } s z8AsyncExitStack.__aexit__.._fix_exception_contextFT)NNN)r; r< ra rP rv rD ) r r| r} r{ r~ r rt r cb_suppressr r r )rz r r% w s8
zAsyncExitStack.__aexit__N)
r r r ru 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 )N) enter_result)r r r r r r2 s znullcontext.__init__c C s | j S )N)r )r r r r r s znullcontext.__enter__c G s d S )Nr )r excinfor r r r s znullcontext.__exit__)N)r r r r2 r r r r r r r s
)r! r; r collectionsr functoolsr __all__ABCr r objectr r, r3 r? r r r rI r
r r rY r r
r r r r r s:
B -!! `E