3
\j3 � @ s� d dl Z d dlZd dlZd dlmZ d dlmZ dddddd d
d gZG d d� de j�Z G d
d� de
�Z G dd� de e �Z dd� Z
G dd� de �ZG dd� de �ZG dd � d e�ZG dd
� d
e�ZG dd � d e �ZG dd� de �ZdS )� N)�deque)�wraps�contextmanager�closing�AbstractContextManager�ContextDecorator� ExitStack�redirect_stdout�redirect_stderr�suppressc @ 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|dd�S tS )Nr r )r �_collections_abc�_check_methods�NotImplemented)�cls�Cr r r �__subclasshook__ s z'AbstractContextManager.__subclasshook__N) �__name__�
__module__� __qualname__r �abc�abstractmethodr � 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 )�args�kwds)�funcr
r r �inner1 s
z(ContextDecorator.__call__.<locals>.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" �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 I s z%_GeneratorContextManager._recreate_cmc C s. y
t | j�S tk
r( td�d �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 td��n�|d krD|� }y| jj|||� W n� tk
r| } z||k S d }~X nf tk
r� } z*||kr�dS |tkr�|j|kr�dS � W Y d d }~X n" tj� d |kr�dS � Y nX td��d S )NFzgenerator didn't stop� z#generator didn't stop after throw())r. r( r/ r0 �throw� __cause__�sys�exc_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.<locals>.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 �_stream�setattrr< )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
�exctype�excinst�exctbr 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 || j�S )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)�cm�cm_exitr r �
_exit_wrapper s z.ExitStack._push_cm_exit.<locals>._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.<locals>._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�_exit�resultr 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 |r�y|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_exc�old_exc� exc_context)� frame_excr r �_fix_exception_contextX s z2ExitStack.__exit__.<locals>._fix_exception_contextFT)NNN)r4 r5 rK rB r` �
BaseException) r
rN � received_excre �suppressed_exc�
pending_raise�cb�new_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 �<module> s" I"