U
e5d�a � @ sf d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ ddddd d
d d d
dddg Z G dd � d e j
�Z G dd
� d
e j
�Z G dd � d e
�ZG dd� d�ZG dd� dee e�ZG dd� dee �Zdd� Zdd� ZG d d� de �ZG d d � d e �ZG d d� de�ZG d d� de�ZG d!d� de �ZG d"d#� d#�ZG d$d
� d
ee �ZG d%d � d ee �ZG d&d� de �Z dS )'� N)�deque��wraps��
MethodType�asynccontextmanager�contextmanager�closing� nullcontext�AbstractContextManager�AbstractAsyncContextManager�AsyncExitStack�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.8/contextlib.py� __enter__ s z AbstractContextManager.__enter__c C s d S r r �r �exc_type� exc_value� tracebackr r r �__exit__ s z AbstractContextManager.__exit__c C s | t krt�|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� Zejdd� �Zedd� �ZdS )r c � s | S r r r r r r �
__aenter__' s z&AbstractAsyncContextManager.__aenter__c � s d S r r r r r r � __aexit__+ s z%AbstractAsyncContextManager.__aexit__c C s | t krt�|dd�S tS )Nr+ r, )r r r r r! r r r r$ 0 s
�z,AbstractAsyncContextManager.__subclasshook__N) r% r& r' r+ r( r) r, r* r$ r r r r r # s
c @ s e Zd Zdd� Zdd� ZdS )r c C s | S r r r r r r � _recreate_cm; s
z ContextDecorator._recreate_cmc s t � �� �fdd��}|S )Nc
s* �� � � � | |�W 5 Q R � S Q R X d S r )r- ��args�kwds��funcr r r �innerH s
z(ContextDecorator.__call__..innerr )r r2 r3 r r1 r �__call__G s zContextDecorator.__call__N)r% r& r' r- r4 r r r r r 8 s c @ s e Zd Zdd� ZdS )� _GeneratorContextManagerBasec C sJ |||�| _ ||| | _| _| _t|dd �}|d kr@t| �j}|| _d S )N�__doc__)�genr2 r/ r0 �getattr�typer6 )r r2 r/ r0 �docr r r �__init__R s
z%_GeneratorContextManagerBase.__init__N)r% r& r' r; r r r r r5 O s r5 c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�_GeneratorContextManagerc C s | � | j| j| j�S r )� __class__r2 r/ r0 r r r r r- f s z%_GeneratorContextManager._recreate_cmc C s< | ` | `| `z t| j�W S tk
r6 td�d �Y nX d S �Nzgenerator didn't yield)r/ r0 r2 �nextr7 �
StopIteration� RuntimeErrorr r r r r l s
z"_GeneratorContextManager.__enter__c
C s |d kr8zt | j� W n tk
r, Y dS X td��n�|d krF|� }z| j�|||� W n� tk
r� } z||k W Y �S d }~X Y nt tk
r� } z4||kr�W Y �&dS |tkr�|j|kr�W Y �
dS � W 5 d }~X Y n$ t�� d |kr�Y dS � Y nX td��d S )NF�generator didn't stop� z#generator didn't stop after throw())r? r7 r@ rA �throw� __cause__�sys�exc_info)r r9 �valuer �excr r r r u s.
z!_GeneratorContextManager.__exit__N)r% r& r' r- r r r r r r r< a s r< c @ s e Zd Zdd� Zdd� ZdS )� _AsyncGeneratorContextManagerc � s6 z| j �� I d H W S tk
r0 td�d �Y nX d S r> )r7 � __anext__�StopAsyncIterationrA r r r r r+ � s z(_AsyncGeneratorContextManager.__aenter__c
� s& |d kr>z| j �� I d H W n tk
r2 Y d S X td��n�|d krL|� }z"| j �|||�I d H td��W n� tk
r� } z||k W Y �S d }~X Y n� tk
r� } z:||kr�W Y �,dS t|ttf�r�|j|kr�W Y �
dS � W 5 d }~X Y n0 tk
�r } z||k �r� W 5 d }~X Y nX d S )NrB z$generator didn't stop after athrow()F) r7 rK rL rA �athrow�
isinstancer@ rE �
BaseException)r �typrH r rI r r r r, � s.
z'_AsyncGeneratorContextManager.__aexit__N)r% r& r' r+ r, r r r r rJ � s rJ c s t � �� fdd��}|S )Nc s t � | |�S r )r< r. �r2 r r �helper� s z contextmanager..helperr �r2 rR r rQ r r � s c s t � �� fdd��}|S )Nc s t � | |�S r )rJ r. rQ r r rR s z#asynccontextmanager..helperr rS r rQ r r � s c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )r c C s
|| _ d S r ��thing)r rU r r r r; & s zclosing.__init__c C s | j S r rT r r r r r ( s zclosing.__enter__c G s | j �� d S r )rU �close)r rG 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 r )� _new_target� _old_targets)r �
new_targetr r r r; 2 s z_RedirectStream.__init__c C s* | j �tt| j�� tt| j| j� | jS r )rZ �appendr8 rF �_stream�setattrrY r r r r r 7 s z_RedirectStream.__enter__c C s t t| j| j�� � d S r )r^ rF r] rZ �pop�r �exctype�excinst�exctbr r r r < s z_RedirectStream.__exit__)r% r& r' r] r; r r r r r r rX . s rX c @ s e Zd ZdZdS )r �stdoutN�r% r& r' r] r r r r r @ s
c @ s e Zd ZdZdS )r �stderrNre r r r r r P s c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )r c G s
|| _ d S r )� _exceptions)r �
exceptionsr r r r; a s zsuppress.__init__c C s d S r r r r r r r d s zsuppress.__enter__c C s |d k ot || j�S r )�
issubclassrg r` r r r r g s
zsuppress.__exit__NrW r r r r r V s c @ sd e Zd Zedd� �Zedd� �Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
de
_ dd� Z ddd�Z
dS )�_BaseExitStackc C s
t || �S r r ��cm�cm_exitr r r �_create_exit_wrapperw s z#_BaseExitStack._create_exit_wrapperc s � ��fdd�}|S )Nc s �� �� d S r r �r rI �tb�r/ �callbackr0 r r �
_exit_wrapper} s z8_BaseExitStack._create_cb_wrapper.._exit_wrapperr �rr r/ r0 rs r rq r �_create_cb_wrapper{ s z!_BaseExitStack._create_cb_wrapperc C s t � | _d S r )r �_exit_callbacksr r r r r; � s z_BaseExitStack.__init__c C s t | �� }| j|_t� | _|S r )r9 rv r )r � new_stackr r r �pop_all� s
z_BaseExitStack.pop_allc C sB t |�}z
|j}W n tk
r0 | �|� Y nX | �||� |S r )r9 r �AttributeError�_push_exit_callback�
_push_cm_exit�r �exit�_cb_type� exit_methodr r r �push� s
z_BaseExitStack.pushc C s( t |�}|j}|�|�}| �||� |S r )r9 r r r{ �r rl �_cm_type�_exit�resultr r r �
enter_context� s
z _BaseExitStack.enter_contextc O s� t | �dkr| ^}}} nV| s&td��nHd|krZ|�d�}| ^}} dd l}|jdtdd� ntdt | �d ��|j|f| �|�}||_|�|� |S ) N� zBdescriptor 'callback' of '_BaseExitStack' object needs an argumentrr r �4Passing 'callback' as keyword argument is deprecated��
stacklevelz8callback expected at least 1 positional argument, got %drC ) �len� TypeErrorr_ �warnings�warn�DeprecationWarningru � __wrapped__rz �r/ r0 r rr r� rs r r r rr � s&
�
�
z_BaseExitStack.callback�#($self, callback, /, *args, **kwds)c C s | � ||�}| �|d� d S )NT)rn rz �r rl rm rs r r r r{ � s z _BaseExitStack._push_cm_exitTc C s | j �||f� d S r )rv r\ )r rr �is_syncr r r rz � s z"_BaseExitStack._push_exit_callbackN)T)r% r& r' � staticmethodrn ru r; rx r� r� rr �__text_signature__r{ rz r r r r rj t s
rj c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )r c C s | S r r r r r r r � s zExitStack.__enter__c
s� |d d k }t �� d � � fdd�}d}d}| jr�| j�� \}}z||� rVd}d}d}W q, t �� }||d |d � d}|}Y q,X q,|r�z|d j} |d �W n tk
r� | |d _� Y nX |o�|S )Nr rC c s4 | j }||krd S |d ks*|� kr$q*|} q || _ d S r �� __context__��new_exc�old_exc� exc_context�� frame_excr r �_fix_exception_context� s z2ExitStack.__exit__.._fix_exception_contextFT�NNN�rF rG rv r_ r� rO )
r � exc_details� received_excr� �suppressed_exc�
pending_raiser� �cb�new_exc_details� fixed_ctxr r� r r � s2
zExitStack.__exit__c C s | � d d d � d S r )r r r r r rV s zExitStack.closeN)r% r& r' r r rV r r r r r � s 1c @ sb e Zd Zedd� �Zedd� �Zdd� Zdd� Zd d
� Zd e_ d d
� Z
dd� Z dd� Z dd� Z
dS )r
c C s
t || �S r r rk r r r �_create_async_exit_wrapper& s z)AsyncExitStack._create_async_exit_wrapperc s � ��fdd�}|S )Nc � s �� ��I d H d S r r ro rq r r rs , s z>AsyncExitStack._create_async_cb_wrapper.._exit_wrapperr rt r rq r �_create_async_cb_wrapper* s z'AsyncExitStack._create_async_cb_wrapperc � s. t |�}|j}|�|�I d H }| �||� |S r )r9 r, r+ �_push_async_cm_exitr� r r r �enter_async_context0 s
z"AsyncExitStack.enter_async_contextc C sD t |�}z
|j}W n tk
r2 | �|d� Y nX | �||� |S �NF)r9 r, ry rz r� r| r r r �push_async_exit< s
z AsyncExitStack.push_async_exitc O s� t | �dkr| ^}}} nV| s&td��nHd|krZ|�d�}| ^}} dd l}|jdtdd� ntdt | �d ��|j|f| �|�}||_|�|d � |S )
Nr� zMdescriptor 'push_async_callback' of 'AsyncExitStack' object needs an argumentrr r r� r� zCpush_async_callback expected at least 1 positional argument, got %drC F) r� r� r_ r� r� r� r� r� rz r� r r r �push_async_callbackN s&
�
� z"AsyncExitStack.push_async_callbackr� c � s | � d d d �I d H d S r )r, r r r r �aclosek s zAsyncExitStack.aclosec C s | � ||�}| �|d� d S r� )r� rz r� r r r r� o s z"AsyncExitStack._push_async_cm_exitc � s | S r r r r r r r+ u s zAsyncExitStack.__aenter__c � s� |d d k }t �� d � � fdd�}d}d}| jr�| j�� \}}z0|rP||� }n||� I d H }|rnd}d}d}W q, t �� } || d |d � d}| }Y q,X q,|r�z|d j}
|d �W n tk
r� |
|d _� Y nX |o�|S )Nr rC c s4 | j }||krd S |d ks*|� kr$q*|} q || _ d S r r� r� r� r r r� ~ s z8AsyncExitStack.__aexit__.._fix_exception_contextFTr� r� ) r r� r� r� r� r� r� r� � cb_suppressr� r� r r� r r, x s8
zAsyncExitStack.__aexit__N)r% r& r' r� r� r� r� r� r� r� r� r� r+ r, r r r r r
s
c @ s& e Zd Zddd�Zdd� Zdd� ZdS ) r
Nc C s
|| _ d S r �� enter_result)r r� r r r r; � s znullcontext.__init__c C s | j S r r� r r r r r � s znullcontext.__enter__c G s d S r r )r �excinfor r r r � s znullcontext.__exit__)NrW r r r r r
� s
) r( rF r � collectionsr � functoolsr �typesr �__all__�ABCr r �objectr r5 r< rJ r r r rX r r r rj r r
r
r r r r � sL � �D �.!! `E