��Yf�0 � @ s d Z 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)z4Utilities for with-statement contexts. See PEP 343.� N)�deque)�wraps�contextmanager�closing�ContextDecorator� ExitStack�redirect_stdout�redirect_stderr�suppressc @ s. e Z d Z d Z d d � Z d d � Z d S)r zJA base class or mixin that enables context managers to work as decorators.c C s | S)a6 Return 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.
� )�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 )�args�kwds)�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__�__doc__r r r r r r
r s c @ sF e Z d Z d Z d d � Z d d � Z d d � Z d d � Z d
S) �_GeneratorContextManagerz%Helper for @contextmanager decorator.c C si | | | � | _ | | | | _ | _ | _ t | d d � } | d k r\ t | � j } | | _ d S)Nr )�genr r r �getattr�typer )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__�sys�exc_info)r r �value� traceback�excr r r
�__exit__? s.
z!_GeneratorContextManager.__exit__N)r r r r r r r" r+ r r r r
r " s
r c s" t � � � f d d � � } | 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>
c 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 @ s: e Z d Z d Z d d � Z d d � Z d d � Z d S) r a2 Context 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()
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
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& �_stream�setattrr0 )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 �exctype�excinst�exctbr 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 Z d S)r aA Context 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 r4 r r r r
r � s c @ s e Z d Z d Z d Z d S)r zCContext manager for temporarily redirecting stderr to another file.�stderrN)r r r r r4 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) r
a? 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
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
r
� s c @ s� e Z d 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 a� 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
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)z?Preserve the context stack by transferring it to a new instance)r r? r )r � new_stackr r r
�pop_all� s zExitStack.pop_allc s/ � � f d d � } � | _ | j | � d S)z:Helper to correctly register callbacks to __exit__ methodsc s
� � | � S)Nr )� exc_details)�cm�cm_exitr r
�
_exit_wrapper s z.ExitStack._push_cm_exit.<locals>._exit_wrapperN)�__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)a Registers a callback with the standard __exit__ method signature
Can suppress exceptions the same way __exit__ methods can.
Also accepts any object with an __exit__ method (registering a call
to the method instead of the object itself)
)r r+ �AttributeErrorr? r3 rH )r �exit�_cb_type� exit_methodr r r
rG
s
zExitStack.pushc s2 � � � f d d � } � | _ | j | � � S)z\Registers an arbitrary callback and arguments.
Cannot suppress exceptions.
c 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)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 r+ r" rH )r rC �_cm_type�_exit�resultr r r
�
enter_context+ s
zExitStack.enter_contextc C s | j d d d � d S)z$Immediately unwind the context stackN)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_exc�old_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_raise�cb�new_exc_details� fixed_ctxr )rY r
r+ ? s2
zExitStack.__exit__N)
r r r r r rA rH rG rO rT r. r" r+ r r r r
r � s
)r r& � collectionsr � functoolsr �__all__�objectr r r r r/ r r r
r r r r r
�<module> s I "