U
.eB @ s d Z ddlZddlZddlZddlmZ ddlmZ ddddd d
d d d
g Zedg7 Zej dd dkrnej
Z ne de
fddiZ dd Zdd ZG dd de ZG dd d e
ZG dd deZd d ZG d d de
ZG d d d e
ZG d d d eZG d!d d eZG d"d
d
e
Zej d d#kZer@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 deZ#dS )0zAcontextlib2 - backports and enhancements to the contextlib module N)dequewrapscontextmanagerclosing nullcontextAbstractContextManagerContextDecorator ExitStackredirect_stdoutredirect_stderrsuppress ContextStack ) ABC __slots__ c C s0 | |kr d S | | | jD ]}t|| q |S N)append __bases__ _classic_mro)CresultBr 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__AttributeErrortupler __dict__NotImplemented)r methodsmromethodr 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|ddS 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= getattrtyper2 )r' r? r< r= docr r r __init__t s
z!_GeneratorContextManager.__init__c C s | | j| j| jS r ) __class__r? r< r= r& r r r r9 s z%_GeneratorContextManager._recreate_cmc C s. z t | jW S tk
r( tdY 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 tdn|d krF| }z | j||| tdW n tk
r } z||k W Y S d }~X Y nj tk
r } z0||krW Y "dS tr|j|krW 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__sysexc_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__Nr/ 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 _streamsetattrr[ r& r r r r( s z_RedirectStream.__enter__c C s t t| j| j d S r )r_ rP r^ r\ popr' exctypeexcinstexctbr 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)
stdoutNr/ 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 || jS 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_excold_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 cmcm_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 tbr< 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 )zEnters 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(|rt| |o|S )Nr rL FT)NNN)rP rQ rr r} r` rv ) r' rt received_excrp rq suppressed_exc
pending_raisecbnew_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 fddZdd 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 ddZdd 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_ABCABCMetaobjectr 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