U
|
H=®dB ã @ 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 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)Ú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 úND:\z\workplace\VsCode\pyvenv\venv\Lib\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|ddS tS )z<Check whether subclass is considered a subclass of this ABC.r( r, )r r% r! )Úclsr r r r Ú__subclasshook__A s z'AbstractContextManager.__subclasshook__N)
|
Ú__name__Ú
|
__module__Ú __qualname__Ú__doc__r( ÚabcÚabstractmethodr, Ú classmethodr. r r r r r 5 s
|
c @ 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_cmr& r r r Ú
|
refresh_cmL 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 r9 ] s
|
z ContextDecorator._recreate_cmc s t fdd}|S )Nc
|
s* ¡ | |W 5 Q R £ S Q R X d S r )r9 ©ÚargsÚkwds©Úfuncr' r r Úinnerj s
|
z(ContextDecorator.__call__.<locals>.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 | 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||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(<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 r )rB r; ©r? r r ÚhelperÕ s z contextmanager.<locals>.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(<module>.open(<arguments>)) as f:
|
<block>
|
|
is equivalent to this:
|
|
f = <module>.open(<arguments>)
|
try:
|
<block>
|
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 || 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_excÚold_excÚ exc_context©Ú frame_excr r Ú_fix_exception_context@ s z3_make_context_fixer.<locals>._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 Ú<lambda>[ ó z%_make_context_fixer.<locals>.<lambda>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* Úexc_tbr r r rv _ s
|
)Ú InstanceTypec C s t | }|tkr| jS |S r )rE r{ 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.<locals>._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.<locals>._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(|rt| |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 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 Ú_abc_ABCÚABCMetaÚobjectr r% r r rB r r rZ r r r rN rr rv Útypesr{ Ú ImportErrorrE r} r
|
r r r r r r Ú<module> sT ý
|
|
(H"
|
|
q
|
