U
��.eB � @ s� d Z ddlZddlZddlZddlmZ ddlmZ ddddd d
d d d
g Zedg7 Zej dd� dkrnej
Z ne� de
fddi�Z dd� Zdd� ZG dd� de �ZG dd � d e
�ZG dd� de�Zd d� ZG d d� de
�ZG d d � d e
�ZG d d � d e�ZG d!d � d e�ZG d"d
� d
e
�Zej d d#kZe�r@d$d%� Zd&d'� Znd(d%� Zd)d'� Zzdd*l m Z W n e k
�rz e Z Y n
X d+d,� Z G d-d
� d
e
�Z!G d.d� de!�Z"G d/d� de�Z#dS )0zAcontextlib2 - backports and enhancements to the contextlib module� N)�deque��wraps�contextmanager�closing� nullcontext�AbstractContextManager�ContextDecorator� ExitStack�redirect_stdout�redirect_stderr�suppress� ContextStack� )� � �ABC� __slots__� c C s0 | |kr d S |� | � | jD ]}t||� q |S �N)�append� __bases__� _classic_mro)�C�result�Br r �;/usr/lib/python3.8/site-packages/pip/_vendor/contextlib2.pyr s
r c G sv z
| j }W n" tk
r, tt| g ��}Y nX |D ]>}|D ],}||jkr:|j| d krbt S q2q:t S q2dS )NT)�__mro__�AttributeError�tupler �__dict__�NotImplemented)r �methods�mro�methodr r r r �_check_methods$ s
r% c @ s2 e Zd ZdZdd� Zejdd� �Zedd� �Z dS ) r z,An abstract base class for context managers.c C s | S )z0Return `self` upon entering the runtime context.r ��selfr r r � __enter__8 s z AbstractContextManager.__enter__c C s dS )z9Raise any exception triggered within the runtime context.Nr )r' �exc_type� exc_value� tracebackr r r �__exit__< s z AbstractContextManager.__exit__c C s | t krt|dd�S tS )z.innerr )r' r? r@ r r> r �__call__i s zContextDecorator.__call__N)r/ r0 r1 r2 r: r9 rA r r r r r I s c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �_GeneratorContextManagerz%Helper for @contextmanager decorator.c C sJ |||�| _ ||| | _| _| _t|dd �}|d kr@t| �j}|| _d S )Nr2 )�genr? r< r= �getattr�typer2 )r' r? r< r= �docr r r �__init__t s
z!_GeneratorContextManager.__init__c C s | � | j| j| j�S r )� __class__r? r< r= r& r r r r9 � s z%_GeneratorContextManager._recreate_cmc C s. z t | j�W S tk
r( td��Y nX d S )Nzgenerator didn't yield)�nextrC �
StopIteration� RuntimeErrorr& r r r r( � s z"_GeneratorContextManager.__enter__c
C s� |d kr8zt | j� W n tk
r, Y d S X td��n�|d krF|� }z | j�|||� td��W n� tk
r� } z||k W Y �S d }~X Y nj tk
r� } z0||kr�W Y �"dS tr�|j|kr�W Y �
dS � W 5 d }~X Y n t�� d |k r� Y nX d S )Nzgenerator didn't stopz#generator didn't stop after throw()F� ) rI rC rJ rK �throw�_HAVE_EXCEPTION_CHAINING� __cause__�sys�exc_info)r' rE �valuer+ �excr r r r, � s,
z!_GeneratorContextManager.__exit__N)r/ r0 r1 r2 rG r9 r( r, r r r r rB q s
rB c s t � �� fdd��}|S )a� @contextmanager decorator.
Typical usage:
@contextmanager
def some_generator():
try:
yield
finally:
This makes this:
with some_generator() as :
equivalent to this:
try:
=
finally:
c s t � | |�S r )rB r; �r? r r �helper� s z contextmanager..helperr )r? rU r rT r r � s c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) r a2 Context to automatically close something at the end of a block.
Code like this:
with closing(.open()) as f:
is equivalent to this:
f = .open()
try:
finally:
f.close()
c C s
|| _ d S r ��thing)r' rW r r r rG � s zclosing.__init__c C s | j S r rV r& r r r r( � s zclosing.__enter__c G s | j �� d S r )rW �close)r' rQ r r r r, � s zclosing.__exit__N�r/ r0 r1 r2 rG 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 rG � s z_RedirectStream.__init__c C s* | j �tt| j�� tt| j| j� | jS r )r\ r rD rP �_stream�setattrr[ r& r r r r( � s z_RedirectStream.__enter__c C s t t| j| j�� � d S r )r_ rP r^ r\ �pop�r' �exctype�excinst�exctbr r r r, s z_RedirectStream.__exit__)r/ r0 r1 r^ rG r( r, r r r r rZ � s rZ c @ s e Zd ZdZdZdS )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/ r0 r1 r2 r^ r r r r r s c @ s e Zd ZdZdZdS )r zCContext manager for temporarily redirecting stderr to another file.�stderrNrf r r r r r s c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) 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 r )� _exceptions)r' �
exceptionsr r r rG ) s zsuppress.__init__c C s d S r r r& r r r r( , s zsuppress.__enter__c C s |d k ot || j�S r )�
issubclassrh ra r r r r, / s
zsuppress.__exit__NrY r r r r r
s
r c s � fdd�}|S )Nc 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 z3_make_context_fixer.._fix_exception_contextr )rp rq r ro r �_make_context_fixer? s
rr c C s< z| d j }| d �W n tk
r6 || d _ � Y nX d S )NrL )rk �
BaseException)� exc_details� fixed_ctxr r r � _reraise_with_existing_contextO s
rv c C s dd� S )Nc S s d S r r )rl rm r r r �[ � z%_make_context_fixer..r ro r r r rr Z s c C s | \}}}t d� d S )Nz!raise exc_type, exc_value, exc_tb)�exec)rt r) r* Zexc_tbr r r rv _ s
)� InstanceTypec C s t | �}|tkr| jS |S r )rE rz rH )�objZobj_typer r r � _get_typek s r| c @ sX e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dS )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 r )r �_exit_callbacksr& r r r rG s zExitStack.__init__c C s t | �� }| j|_t� | _|S )z?Preserve the context stack by transferring it to a new instance)rE r} r )r' � new_stackr r r �pop_all� s
zExitStack.pop_allc s" � �fdd�}� |_ | �|� dS )z:Helper to correctly register callbacks to __exit__ methodsc s �� f| �� S r r )rt ��cm�cm_exitr r �
_exit_wrapper� s z.ExitStack._push_cm_exit.._exit_wrapperN)�__self__�push)r' r� r� r� r r� r �
_push_cm_exit� s zExitStack._push_cm_exitc C sD t |�}z
|j}W n tk
r2 | j�|� Y nX | �||� |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, r r} r r� )r' �exit�_cb_type� exit_methodr r r r� � s
zExitStack.pushc s$ � ��fdd�}�|_ | �|� �S )z\Registers an arbitrary callback and arguments.
Cannot suppress exceptions.
c s �� �� d S r r )r) rS �tb�r< �callbackr= r r r� � s z)ExitStack.callback.._exit_wrapper)� __wrapped__r� )r' r� r< r= r� r r� r r� � s
zExitStack.callbackc C s( t |�}|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( r� )r' r� �_cm_type�_exitr r r r �
enter_context� s
zExitStack.enter_contextc C s | � ddd� dS )z$Immediately unwind the context stackN)r, r& r r r rX � s zExitStack.closec C s | S r r r& r r r r( � s zExitStack.__enter__c G s� |d d k }t �� d }t|�}d}d}| jr�| j�� }z||� rNd}d}d}W q( t �� }||d |d � d}|}Y q(X q(|r�t|� |o�|S )Nr rL FT)NNN)rP rQ rr r} r` rv ) r' rt � received_excrp rq �suppressed_exc�
pending_raise�cb�new_exc_detailsr r r r, � s(
zExitStack.__exit__N)
r/ r0 r1 r2 rG r r� r� r� r� rX r( r, r r r r r
s s
c s8 e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Z� ZS )
r z+Backwards compatibility alias for ExitStackc s t �dt� tt| ��� d S )Nz*ContextStack has been renamed to ExitStack)r6 r7 r8 �superr rG r& �rH r r rG � s �zContextStack.__init__c C s
| � |�S r )r� )r' r� r r r �
register_exit� s zContextStack.register_exitc O s | j |f|�|�S r )r� )r' r� r< r= r r r �register� s zContextStack.registerc C s | � � S r )r r& r r r �preserve� s zContextStack.preserve) r/ r0 r1 r2 rG r� r� r� �
__classcell__r r r� r r � s
c @ s* e Zd ZdZd dd�Zdd� Zdd� ZdS )
r aM Context manager that does no additional processing.
Used as a stand-in for a normal context manager, when a particular
block of code is only sometimes used with a normal context manager:
cm = optional_cm if condition else nullcontext()
with cm:
# Perform operation, using optional_cm if condition is True
Nc C s
|| _ d S r �� enter_result)r' r� r r r rG � 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__)NrY r r r r r � s
)$r2 r3 rP r6 � collectionsr � functoolsr �__all__� version_infor Z_abc_ABC�ABCMeta�objectr r% r r rB r r rZ r r r
rN rr rv �typesrz � ImportErrorrE r| r
r r r r r r � sT �
(H"
q