Your IP : 216.73.216.196
B
�s�W�9 � @ sL d Z ddlZddlZddlmZ ddlmZ ddddd d
d gZed g7 ZG d
d� de�Z G dd� de �Z
dd� Z G dd� de�Z G dd� de�Z
G dd � d e
�ZG dd
� d
e
�ZG dd � d e�Zejd dkZer�dd� Zdd� Znd d� Zd d� Zydd lmZ W n ek
�r eZY n
X d d � ZG d!d� de�ZG d"d � d e�ZdS )#zAcontextlib2 - backports and enhancements to the contextlib module� N)�deque)�wraps�contextmanager�closing�ContextDecorator� ExitStack�redirect_stdout�redirect_stderr�suppress� ContextStackc @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) r zJA base class or mixin that enables context managers to work as decorators.c C s t �dt� | �� S )a� Returns the context manager used to actually wrap the call to the
decorated function.
The default implementation just returns *self*.
Overriding this method allows otherwise one-shot context managers
like _GeneratorContextManager to support use as decorators via
implicit recreation.
DEPRECATED: refresh_cm was never added to the standard library's
ContextDecorator API
z2refresh_cm was never added to the standard library)�warnings�warn�DeprecationWarning� _recreate_cm)�self� r �</opt/alt/python37/lib/python3.7/site-packages/contextlib2.py�
refresh_cm s
zContextDecorator.refresh_cmc 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.
r )r r r r r " s
z ContextDecorator._recreate_cmc s t � �� �fdd��}|S )Nc s �� � � � | |�S Q R X d 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 r 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 )Nr )�genr r r �getattr�typer )r r r r �docr r r �__init__9 s
z!_GeneratorContextManager.__init__c C s | � | j| j| j�S )N)� __class__r r r )r r r r r G s z%_GeneratorContextManager._recreate_cmc C s, y
t | j�S tk
r& td��Y nX d S )Nzgenerator didn't yield)�nextr �
StopIteration� RuntimeError)r r r r � __enter__M s
z"_GeneratorContextManager.__enter__c
C s� |d kr6yt | j� W n tk
r* d S X td��n�|d krD|� }y | j�|||� td��W n� tk
r� } z||k S d }~X Y n^ tk
r� } z$||kr�dS tr�|j|kr�dS � W d d }~X Y n t�� d |k rނ Y nX d S )Nzgenerator didn't stopz#generator didn't stop after throw()F� ) r$ r r% r& �throw�_HAVE_EXCEPTION_CHAINING� __cause__�sys�exc_info)r r �value� traceback�excr r r �__exit__S s,
z!_GeneratorContextManager.__exit__N)r r r r r" r r' r1 r r r r r 6 s
r c s t � �� fdd��}|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 r2 r )r 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(<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 r3 r r r r" � s zclosing.__init__c C s | j S )N)r3 )r r r r r' � s zclosing.__enter__c G s | j �� d S )N)r3 �close)r r- r r r r1 � s zclosing.__exit__N)r r r r r" r' r1 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 �tt| j�� tt| j| j� | jS )N)r7 �appendr r, �_stream�setattrr6 )r r r r r' � s z_RedirectStream.__enter__c C s t t| j| j�� � d S )N)r; r, r: r7 �pop)r �exctype�excinst�exctbr r r r1 � s z_RedirectStream.__exit__)r r r r: r" r' r1 r r r r r5 � s r5 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 r r r r: r r r r r � s c @ s e Zd ZdZdZdS )r zCContext manager for temporarily redirecting stderr to another file.�stderrN)r r r r r: 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 )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)�
issubclassrB )r r= r> r? r r r r1 � s
zsuppress.__exit__N)r r r r r" r' r1 r r r r r
� s � c s � fdd�}|S )Nc 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_context s z3_make_context_fixer.<locals>._fix_exception_contextr )rJ rK r )rJ r �_make_context_fixer s
rL c C s< y| d j }| d �W n tk
r6 || d _ � Y nX d S )Nr( )rF �
BaseException)� exc_details� fixed_ctxr r r � _reraise_with_existing_context s
rP c C s dd� S )Nc S s d S )Nr )rG rH r r r �<lambda> � z%_make_context_fixer.<locals>.<lambda>r )rJ r r r rL s c C s | \}}}t d� d S )Nz!raise exc_type, exc_value, exc_tb)�exec)rN �exc_type� exc_value�exc_tbr r r rP " s
)� InstanceTypec C s t | �}|tkr| jS |S )N)r rW r# )�objZobj_typer r r � _get_type. s rY 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 )N)r �_exit_callbacks)r r r r r" A s zExitStack.__init__c C s t | �� }| j|_t� | _|S )z?Preserve the context stack by transferring it to a new instance)r rZ r )r � new_stackr r r �pop_allD s
zExitStack.pop_allc s" � �fdd�}� |_ | �|� dS )z:Helper to correctly register callbacks to __exit__ methodsc s �� f| �� S )Nr )rN )�cm�cm_exitr r �
_exit_wrapperM s z.ExitStack._push_cm_exit.<locals>._exit_wrapperN)�__self__�push)r r] r^ r_ r )r] r^ r �
_push_cm_exitK s zExitStack._push_cm_exitc C sD t |�}y
|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)
)rY r1 �AttributeErrorrZ r9 rb )r �exit�_cb_type� exit_methodr r r ra R s
zExitStack.pushc s$ � ��fdd�}�|_ | �|� �S )z\Registers an arbitrary callback and arguments.
Cannot suppress exceptions.
c s �� �� d S )Nr )rT r0 �tb)r �callbackr r r r_ k s z)ExitStack.callback.<locals>._exit_wrapper)� __wrapped__ra )r rh r r r_ r )r rh r r rh f 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.
)rY r1 r' rb )r r] �_cm_type�_exit�resultr r r �
enter_contexts s
zExitStack.enter_contextc C s | � ddd� dS )z$Immediately unwind the context stackN)r1 )r r r r r4 � s zExitStack.closec C s | S )Nr )r r r r r' � s zExitStack.__enter__c G s� |d d k }t �� d }t|�}d}d}x\| jr�| j�� }y||� rPd}d}d}W q* t �� }||d |d � d}|}Y q*X q*W |r�t|� |o�|S )Nr r( FT)NNN)r, r- rL rZ r< rP ) r rN � received_excrJ rK �suppressed_exc�
pending_raise�cb�new_exc_detailsr r r r1 � s(
zExitStack.__exit__N)
r r r r r" r\ rb ra rh rm r4 r' r1 r r r r r 5 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)r r
r �superr r" )r )r# r r r" � s zContextStack.__init__c C s
| � |�S )N)ra )r rh r r r �
register_exit� s zContextStack.register_exitc O s | j |f|�|�S )N)rh )r rh r r r r r �register� s zContextStack.registerc C s | � � S )N)r\ )r r r r �preserve� s zContextStack.preserve) r r r r r" rt ru rv �
__classcell__r r )r# r r � s
) r r, r � collectionsr � functoolsr �__all__�objectr r r r r5 r r r
� version_infor* rL rP �typesrW � ImportErrorr rY r r r r r r �<module> s8
(H"
p