B
ReB @ 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' Zydd*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)deque)wrapscontextmanagerclosing nullcontextAbstractContextManagerContextDecorator ExitStackredirect_stdoutredirect_stderrsuppress ContextStack ) ABC __slots__ c C s4 | |kr d S | | x| jD ]}t|| q W |S )N)append __bases__ _classic_mro)CresultBr r /builddir/build/BUILDROOT/alt-python37-pip-20.2.4-6.el8.x86_64/opt/alt/python37/lib/python3.7/site-packages/pip/_vendor/contextlib2.pyr s
r c G sp y
| j }W n" tk
r, tt| g }Y nX x<|D ]4}x.|D ]"}||jkr>|j| d kr^tS P q>W tS q4W dS )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.inner)r )r$ r: r; r )r: r$ r __call__i s zContextDecorator.__call__N)r, r- r. r/ r7 r6 r< 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 )Nr/ )genr: r8 r9 getattrtyper/ )r$ r: r8 r9 docr r r __init__t s
z!_GeneratorContextManager.__init__c C s | | j| j| jS )N) __class__r: r8 r9 )r$ r r r r6 s z%_GeneratorContextManager._recreate_cmc C s, y
t | jS tk
r& tdY nX d S )Nzgenerator didn't yield)nextr>
StopIteration RuntimeError)r$ r r r r% s
z"_GeneratorContextManager.__enter__c
C s |d kr6yt | j W n tk
r* d S X tdn|d krD| }y | j||| tdW n tk
r } z||k S d }~X Y n^ tk
r } z$||krdS tr|j|krdS 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 ) rD r> rE rF throw_HAVE_EXCEPTION_CHAINING __cause__sysexc_info)r$ r@ valuer( excr r r r) s,
z!_GeneratorContextManager.__exit__N)r, r- r. r/ rB r6 r% r) r r r r r= q s
r= 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 )N)r= )r8 r9 )r: r r helper s z contextmanager..helper)r )r: rO 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(.open()) as f:
is equivalent to this:
f = .open()
try:
finally:
f.close()
c C s
|| _ d S )N)thing)r$ rP r r r rB s zclosing.__init__c C s | j S )N)rP )r$ r r r r% s zclosing.__enter__c G s | j d S )N)rP close)r$ rL r r r r) s zclosing.__exit__N)r, r- r. r/ rB r% r) 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 rB s z_RedirectStream.__init__c C s* | j tt| j tt| j| j | jS )N)rT r r? rK _streamsetattrrS )r$ r r r r% s z_RedirectStream.__enter__c C s t t| j| j d S )N)rW rK rV rT pop)r$ exctypeexcinstexctbr r r r) s z_RedirectStream.__exit__)r, r- r. rV rB r% r) r r r r rR s rR 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/ rV 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/ rV 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 rB ) s zsuppress.__init__c C s d S )Nr )r$ r r r r% , s zsuppress.__enter__c C s |d k ot || jS )N)
issubclassr^ )r$ rY rZ r[ r r r r) / s
zsuppress.__exit__N)r, r- r. r/ rB r% r) r r r r r s r c s fdd}|S )Nc s8 x,| j }||krd S |d ks$| kr&P |} qW || _ d S )N) __context__)new_excold_exc exc_context) frame_excr r _fix_exception_context@ s z3_make_context_fixer.._fix_exception_contextr )re rf r )re r _make_context_fixer? s
rg c C s< y| d j }| d W n tk
r6 || d _ Y nX d S )NrG )ra
BaseException) exc_details fixed_ctxr r r _reraise_with_existing_contextO s
rk c C s dd S )Nc S s d S )Nr )rb rc r r r [ z%_make_context_fixer..r )re r r r rg Z s c C s | \}}}t d d S )Nz!raise exc_type, exc_value, exc_tb)exec)ri r& r' exc_tbr r r rk _ s
) InstanceTypec C s t | }|tkr| jS |S )N)r@ rp rC )objZobj_typer r r _get_typek s rr 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 rB s zExitStack.__init__c C s t | }| j|_t | _|S )z?Preserve the context stack by transferring it to a new instance)r@ rs 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 )Nr )ri )cmcm_exitr r
_exit_wrapper s z.ExitStack._push_cm_exit.._exit_wrapperN)__self__push)r$ rv rw rx r )rv rw r
_push_cm_exit 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)
)rr r) r rs r r{ )r$ exit_cb_type exit_methodr r r rz s
zExitStack.pushc s$ fdd}|_ | | S )z\Registers an arbitrary callback and arguments.
Cannot suppress exceptions.
c s d S )Nr )r& rN tb)r8 callbackr9 r r rx s z)ExitStack.callback.._exit_wrapper) __wrapped__rz )r$ r r8 r9 rx r )r8 r r9 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.
)rr r) r% r{ )r$ rv _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 rQ 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 |rt| |o|S )Nr rG FT)NNN)rK rL rg rs rX rk ) r$ ri received_excre rf suppressed_exc
pending_raisecbnew_exc_detailsr r r r) s(
zExitStack.__exit__N)
r, r- r. r/ rB ru r{ rz r r rQ 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)r3 r4 r5 superr
rB )r$ )rC r r rB s zContextStack.__init__c C s
| |S )N)rz )r$ r r r r
register_exit s zContextStack.register_exitc O s | j |f||S )N)r )r$ r r8 r9 r r r register s zContextStack.registerc C s | S )N)ru )r$ r r r preserve s zContextStack.preserve) r, r- r. r/ rB r r r
__classcell__r r )rC 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 )N) enter_result)r$ r r r r rB s znullcontext.__init__c C s | j S )N)r )r$ r r r r% s znullcontext.__enter__c G s d S )Nr )r$ excinfor r r r) s znullcontext.__exit__)N)r, r- r. r/ rB r% r) r r r r r s
)$r/ r0 rK r3 collectionsr functoolsr __all__ version_infor _abc_ABCABCMetaobjectr r# r r r= r r rR r
r r rI rg rk typesrp ImportErrorr@ rr r r
r r r r r sL
(H"
q