]0 @ s d d l Z d d l m Z d d l m Z d d d d d d d
g Z Gd d d e Z Gd d
d
e Z d d Z Gd d d e Z
Gd d d Z Gd d d e Z Gd d d e Z
Gd d
d
Z Gd d d e Z d S) N)deque)wrapscontextmanagerclosingContextDecorator ExitStackredirect_stdoutredirect_stderrsuppressc @ s( e Z d Z d d Z d d Z d S)r c C s | S)N )selfr r //opt/alt/python35/lib64/python3.5/contextlib.py _recreate_cm s
z ContextDecorator._recreate_cmc s% t f d d } | S)Nc
s% j | | SWd QRXd S)N)r )argskwds)funcr r r
inner s
z(ContextDecorator.__call__.<locals>.inner)r )r r r r )r r r
__call__ s !zContextDecorator.__call__N)__name__
__module__ __qualname__r r r r r r
r s c @ s@ e Z d Z d d Z d d Z d d Z d d Z d S)
_GeneratorContextManagerc C si | | | | _ | | | | _ | _ | _ t | d d } | d k r\ 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 | j S)N) __class__r r r )r r r r
r 3 s z%_GeneratorContextManager._recreate_cmc C s9 y t | j SWn! t k
r4 t d d Yn Xd S)Nzgenerator didn't yield)nextr
StopIteration RuntimeError)r r r r
__enter__9 s
z"_GeneratorContextManager.__enter__c C s0 | d k rE y t | j Wn t k
r5 d SYq,Xt d n | d k rZ | } y | j j | | | Wn t k
r } z | | k SWYd d } ~ Xn{ t k
r } z7 | | k r d S| t k r | j | k r d S WYd d } ~ Xn% t j d | k rd S Yn Xt d d S)NFzgenerator didn't stop z#generator didn't stop after throw())r r r r! throw __cause__sysexc_info)r r value tracebackexcr r r
__exit__? s.
z!_GeneratorContextManager.__exit__N)r r r r r r" r+ r r r r
r " s r c s" t f d d } | S)Nc s t | | S)N)r )r r )r r r
helper s z contextmanager.<locals>.helper)r )r r, r )r r
r k s c @ s4 e Z d Z d d Z d d Z d d Z d S)r c C s
| | _ d S)N)thing)r r- r r r
r s zclosing.__init__c C s | j S)N)r- )r r r r
r" s zclosing.__enter__c G s | j j d S)N)r- close)r r' r r r
r+ s zclosing.__exit__N)r r r r r" r+ r r r r
r s c @ s: e Z d Z d Z d d Z d d Z d d Z d S)_RedirectStreamNc C s | | _ g | _ d S)N) _new_target _old_targets)r
new_targetr r r
r s z_RedirectStream.__init__c C s9 | j j t t | j t t | j | j | j S)N)r1 appendr r& _streamsetattrr0 )r r r r
r" s z_RedirectStream.__enter__c C s t t | j | j j d S)N)r5 r& r4 r1 pop)r exctypeexcinstexctbr r r
r+ s z_RedirectStream.__exit__)r r r r4 r r" r+ r r r r
r/ s r/ c @ s e Z d Z d Z d S)r stdoutN)r r r r4 r r r r
r s
c @ s e Z d Z d Z d S)r stderrN)r r r r4 r r r r
r s c @ s4 e Z d Z d d Z d d Z d d Z d S)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 o t | | j S)N)
issubclassr< )r r7 r8 r9 r r r
r+ s
zsuppress.__exit__N)r r r r r" r+ r r r r
r
s c @ s| e Z d Z d d Z d d Z d d Z d d Z d d
Z d d Z d
d Z d d Z
d d Z d S)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 r? r )r new_stackr r r
pop_all s zExitStack.pop_allc s/ f d d } | _ | j | d S)Nc s
| S)Nr ) exc_details)cmcm_exitr r
_exit_wrapper s z.ExitStack._push_cm_exit.<locals>._exit_wrapper)__self__push)r rC rD rE r )rC rD r
_push_cm_exit s zExitStack._push_cm_exitc C sR t | } y
| j } Wn" t k
r= | j j | Yn X| j | | | S)N)r r+ AttributeErrorr? r3 rH )r exit_cb_type exit_methodr r r
rG
s
zExitStack.pushc s2 f d d } | _ | j | S)Nc s d S)Nr )exc_typer* tb)r callbackr r r
rE # s z)ExitStack.callback.<locals>._exit_wrapper) __wrapped__rG )r rO r r rE r )r rO r r
rO s
zExitStack.callbackc C s8 t | } | j } | j | } | j | | | S)N)r r+ r" rH )r rC _cm_type_exitresultr r r
enter_context+ s
zExitStack.enter_contextc C s | j d d d d S)N)r+ )r r r r
r. 8 s zExitStack.closec C s | S)Nr )r r r r
r" < s zExitStack.__enter__c s | d d k } t j d f d d } d } d } xv | j r | j j } y" | | rz d } d } d } WqA t j } | | d | d d } | } YqA XqA W| r y | d j } | d Wn" t k
r | | d _ Yn X| o| S)Nr r# c sL x< | j } | | k r d S| d k s4 | k r5 P| } q W| | _ d S)N) __context__)new_excold_exc exc_context) frame_excr r
_fix_exception_contextE s
z2ExitStack.__exit__.<locals>._fix_exception_contextFT)NNN)r& r' r? r6 rU
BaseException) r rB received_excrZ suppressed_exc
pending_raisecbnew_exc_details fixed_ctxr )rY r
r+ ? s2
zExitStack.__exit__N) r r r r rA rH rG rO rT r. r" r+ r r r r
r s
)r& collectionsr functoolsr __all__objectr r r r r/ r r r
r r r r r
<module> s I "