← Back
Editing: contextlib.cpython-35.pyc
��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 "
Save File
Cancel