3
\j3 @ s d dl Z d dlZd dlZd dlmZ d dlmZ dddddd d
d gZG d d de jZ G d
d de
Z G dd de e Z dd Z
G dd de ZG dd de ZG dd d eZG dd
d
eZG dd d e ZG dd de ZdS ) N)deque)wrapscontextmanagerclosingAbstractContextManagerContextDecorator ExitStackredirect_stdoutredirect_stderrsuppressc @ s. e Zd Zdd Zejdd Zedd ZdS )r c C s | S )N )selfr r "/usr/lib64/python3.6/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 krtj|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 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 j | |S Q R X d S )N)r )argskwds)funcr
r r inner1 s
z(ContextDecorator.__call__..inner)r )r
r# r$ r )r# r
r __call__0 s zContextDecorator.__call__N)r r r r r% r r r r r ! s c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
_GeneratorContextManagerc C sJ |||| _ ||| | _| _| _t|dd }|d kr@t| j}|| _d S )N__doc__)genr# r! r" getattrtyper' )r
r# r! r" docr r r __init__; s
z!_GeneratorContextManager.__init__c C s | j | j| j| jS )N) __class__r# r! r" )r
r r r r I s z%_GeneratorContextManager._recreate_cmc C s. y
t | jS tk
r( tdd Y nX d S )Nzgenerator didn't yield)nextr(
StopIteration RuntimeError)r
r r r r O s
z"_GeneratorContextManager.__enter__c C s |d kr6yt | j W n tk
r* dS X tdn|d krD| }y| jj||| W n tk
r| } z||k S d }~X nf tk
r } z*||krdS |tkr|j|krdS W Y d d }~X n" tj d |krdS Y nX tdd S )NFzgenerator didn't stop z#generator didn't stop after throw())r. r( r/ r0 throw __cause__sysexc_info)r
r* valuer excr r r r U s.
z!_GeneratorContextManager.__exit__N)r r r r, r r r r r r r r& 8 s r& c s t fdd}|S )Nc s t | |S )N)r& )r! r" )r# r r helper s z contextmanager..helper)r )r# r8 r )r# r r s c @ s$ e Zd Zdd Zdd Zdd ZdS )r c C s
|| _ d S )N)thing)r
r9 r r r r, s zclosing.__init__c C s | j S )N)r9 )r
r r r r s zclosing.__enter__c G s | j j d S )N)r9 close)r
r5 r r r r s zclosing.__exit__N)r 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 )N) _new_target _old_targets)r
new_targetr r r r, s z_RedirectStream.__init__c C s* | j jtt| j tt| j| j | jS )N)r= appendr) r4 _streamsetattrr< )r
r r r r s z_RedirectStream.__enter__c C s t t| j| jj d S )N)rA r4 r@ r= pop)r
exctypeexcinstexctbr r r r s z_RedirectStream.__exit__)r r r r@ r, r r r r r r r; s r; c @ s e Zd ZdZdS )r stdoutN)r r r r@ r r r r r s
c @ s e Zd ZdZdS )r
stderrN)r r r r@ r r r r r
s c @ s$ e Zd Zdd Zdd Zdd ZdS )r c G s
|| _ d S )N) _exceptions)r
exceptionsr r r r, s zsuppress.__init__c C s d S )Nr )r
r r r r s zsuppress.__enter__c C s |d k ot || jS )N)
issubclassrH )r
rC rD rE r r r r s
zsuppress.__exit__N)r r r r, r r r r r r r s c @ sL e Zd Zdd Zdd Zdd Zdd Zd d
Zd d Zd
d Z dd Z
dS )r c C s t | _d S )N)r _exit_callbacks)r
r r r r, s zExitStack.__init__c C s t | }| j|_t | _|S )N)r* rK r )r
new_stackr r r pop_all s
zExitStack.pop_allc s" fdd} |_ | j| d S )Nc s f| S )Nr ) exc_details)cmcm_exitr r
_exit_wrapper s z.ExitStack._push_cm_exit.._exit_wrapper)__self__push)r
rO rP rQ r )rO rP r
_push_cm_exit s zExitStack._push_cm_exitc C sD t |}y
|j}W n tk
r2 | jj| Y nX | j|| |S )N)r* r AttributeErrorrK r? rT )r
exit_cb_type exit_methodr r r rS s
zExitStack.pushc s$ fdd}|_ | j| S )Nc s d S )Nr )r r7 tb)r! callbackr" r r rQ 9 s z)ExitStack.callback.._exit_wrapper) __wrapped__rS )r
rZ r! r" rQ r )r! rZ r" r rZ 4 s
zExitStack.callbackc C s( t |}|j}|j|}| j|| |S )N)r* r r rT )r
rO _cm_type_exitresultr r r
enter_contextA s
zExitStack.enter_contextc C s | j d d d d S )N)r )r
r r r r: N s zExitStack.closec s |d d k }t j d fdd}d}d}x\| jr| jj }y|| rTd}d}d}W q. t j }||d |d d}|}Y q.X q.W |ry|d j}|d W n tk
r ||d _ Y nX |o|S )Nr r1 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_contextX s z2ExitStack.__exit__.._fix_exception_contextFT)NNN)r4 r5 rK rB r`
BaseException) r
rN received_excre suppressed_exc
pending_raisecbnew_exc_details fixed_ctxr )rd r r R s2
zExitStack.__exit__N) r r r r, rM rT rS rZ r_ r: r r r r r r s
)r r4 r collectionsr functoolsr __all__ABCr objectr r& r r r; r r
r r r r r r s" I"