3
�
�^j3 c @ 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 �N� ��selfr r �"/usr/lib64/python3.6/contextlib.py� __enter__ � z AbstractContextManager.__enter__c C � d S r
r )r �exc_typeZ exc_value� tracebackr r r �__exit__ s z AbstractContextManager.__exit__c C s | t krtj|dd�S tS )Nr r )r �_collections_abcZ_check_methodsZNotImplemented)ZclsZCr r r �__subclasshook__ s z'AbstractContextManager.__subclasshook__N) �__name__�
__module__� __qualname__r �abcZabstractmethodr Z classmethodr r r r r r
s c @ s e Zd Zdd� Zdd� ZdS )r c C r r
r r r r r � _recreate_cm$ �
z ContextDecorator._recreate_cmc s t � �� �fdd��}|S )Nc
s �j � � � | |�S Q R X d S r
)r ��args�kwds��funcr r r �inner1 s
z(ContextDecorator.__call__.<locals>.innerr )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" Zdocr r r �__init__; s
z!_GeneratorContextManager.__init__c C s | j | j| j| j�S r
)Z __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� RuntimeErrorr 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. r/ ZthrowZ __cause__�sys�exc_info)r r+ Zvaluer �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 r
)r' r �r$ r r �helper� r z contextmanager.<locals>.helperr )r$ r5 r r4 r r � s c @ �$ e Zd Zdd� Zdd� Zdd� ZdS )r c C �
|| _ d S r
��thing)r r9 r r r r, � � zclosing.__init__c C s | j S r
r8 r r r r r � r: zclosing.__enter__c G s | j j� d S r
)r9 �close)r r2 r r r r � r: 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 r
)� _new_target� _old_targets)r Z
new_targetr r r r, � s z_RedirectStream.__init__c C s* | j jtt| j�� tt| j| j� | jS r
)r? �appendr* r1 �_stream�setattrr> r r r r r � s z_RedirectStream.__enter__c C s t t| j| jj� � d S r
)rB r1 rA r? �pop�r ZexctypeZexcinstZexctbr r r r � r: z_RedirectStream.__exit__)r r r rA r, r r r r r r r= � s r= c @ � e Zd ZdZdS )r ZstdoutN�r r r rA r r r r r � s
c @ rE )r
ZstderrNrF r r r r r
� s c @ r6 )r c G r7 r
)� _exceptions)r Z
exceptionsr r r r, � r: zsuppress.__init__c C r r
r r r r r r � r: zsuppress.__enter__c C s |d k ot || j�S r
)Z
issubclassrG rD r r r r � r zsuppress.__exit__Nr< 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 r
)r �_exit_callbacksr r r r r, r: zExitStack.__init__c C s t | �� }| j|_t� | _|S r
)r+ rH r )r Z new_stackr r r �pop_all s
zExitStack.pop_allc s" � �fdd�}� |_ | j|� d S )Nc s �� f| �� S r
r )� exc_details��cm�cm_exitr r �
_exit_wrapper r: z.ExitStack._push_cm_exit.<locals>._exit_wrapper)Z__self__�push)r rL rM rN r rK r �
_push_cm_exit s zExitStack._push_cm_exitc C sD t |�}y
|j}W n tk
r2 | jj|� Y nX | j||� |S r
)r+ r ZAttributeErrorrH r@ rP )r ZexitZ_cb_typeZ exit_methodr r r rO s
zExitStack.pushc s$ � ��fdd�}�|_ | j|� �S )Nc s �� �� d S r
r )r r3 Ztb�r! �callbackr" r r rN 9 r: z)ExitStack.callback.<locals>._exit_wrapper)Z __wrapped__rO )r rR r! r" rN r rQ r rR 4 s
zExitStack.callbackc C s( t |�}|j}|j|�}| j||� |S r
)r+ r r rP )r rL Z_cm_typeZ_exitZresultr r r �
enter_contextA s
zExitStack.enter_contextc C s | j d d d � d S r
)r r r r r r; N r 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 r0 c s8 x,| j }||krd S |d ks$|� kr&P |} qW || _ d S r
)� __context__)Znew_excZold_excZ exc_context�Z frame_excr r �_fix_exception_contextX s z2ExitStack.__exit__.<locals>._fix_exception_contextFT)NNN)r1 r2 rH rC rT Z
BaseException) r rJ Z received_excrV Zsuppressed_excZ
pending_raiseZcbZnew_exc_detailsZ fixed_ctxr rU r r R s2
zExitStack.__exit__N) r r r r, rI rP rO rR rS r; r r r r r r s
)r r1 r Z collectionsr Z functoolsr Z__all__ZABCr Zobjectr r' r r r= r r
r r r r r r Z<module> s" I"