{@ @
@ r N) r
r r r3 r4 r r5 r: r6 r7 r< r2 r
r r r r 2 ss CC# L11 [ r r c $ e Zd ZdZd Zd Zd ZdS )r zJA base class or mixin that enables context managers to work as decorators.c ^ t j 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)warningswarnDeprecationWarning _recreate_cmr! s r
refresh_cmzContextDecorator.refresh_cmL s0
J( * * * """r c | 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! s r rC z ContextDecorator._recreate_cm] s r c @ t fd }|S )Nc p 5 | i |cd d d S # 1 swxY w Y d S NrC argskwdsfuncr" s r innerz(ContextDecorator.__call__..innerj s ""$$
+
+tT*T**
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+s +//r r" rM rN s `` r __call__zContextDecorator.__call__i s9 t + + + + +
+ r N)r
r r r3 rD rC rP r
r r r r I sG PP# # #"
r r c e Zd ZdZd Zd ZdS )AsyncContextDecoratorzPA base class or mixin that enables async context managers to work as decorators.c | S )z-Return a recreated instance of self.
r
r! s r rC z"AsyncContextDecorator._recreate_cmt s r c @ t fd }|S )Nc K 4 d {V | i | d {V cd d d d {V S # 1 d {V swxY w Y d S rH rI rJ s r rN z-AsyncContextDecorator.__call__..innerz s ((**
1
1
1
1
1
1
1
1 !T4 04 0 0000000
1
1
1
1
1
1
1
1
1
1
1
1
1
1
1
1
1
1
1
1
1
1
1
1
1
1
1
1
1
1s ?
A A r rO s `` r rP z AsyncContextDecorator.__call__y s9 t 1 1 1 1 1
1 r N)r
r r r3 rC rP r
r r rR rR q s8 VV
r rR c e Zd ZdZd ZdS ) _GeneratorContextManagerBasezBShared functionality for @contextmanager and @asynccontextmanager.c ||i || _ |||c| _ | _ | _ t |dd }|t | j }|| _ d S )Nr3 )genrM rK rL getattrtyper3 )r" rM rK rL docs r __init__z%_GeneratorContextManagerBase.__init__ s] 4&&&*.d' 49didIt,, ;t**$C r N)r
r r r3 r] r
r r rW rW s) LL r rW c $ e Zd ZdZd Zd Zd ZdS )_GeneratorContextManagerz%Helper for @contextmanager decorator.c N | | j | j | j S rH __class__rM rK rL r! s r rC z%_GeneratorContextManager._recreate_cm ~~diDI>>>r c t | ` | `| ` t | j S # t
$ r t
d d w xY wNzgenerator didn't yield)rK rL rM nextrY
StopIteration RuntimeErrorr! s r r# z"_GeneratorContextManager.__enter__ sU
Ity$) C>> ! C C C 788d B Cs 7c |5 t | j t d # t $ r Y dS w xY w|
| } | j ||| nj# t $ r}||ucY d }~S d }~wt $ r(}||u rY d }~dS |t u r|j |u rY d }~dS d }~w t
j d |u rY dS xY wt d )Ngenerator didn't stopF z#generator didn't stop after throw())rf rY rh rg throw __cause__sysexc_info)r" r[ valuer) excs r r* z!_GeneratorContextManager.__exit__ sC <
<TX ##:;;; !
uu
}
tUI6666
(
(
( %'''''''
%<< 55555 =((S]e-C-C 55555
<>>!$-- 55 DEE EsH '
55 A" "
C ,A5/C 5
C B% B%$B%% C C N)r
r r r3 rC r# r* r
r r r_ r_ sQ 0/? ? ? C C C-F -F -F -F -Fr r_ c $ e Zd ZdZd Zd Zd ZdS ) _AsyncGeneratorContextManagerz Helper for @asynccontextmanager.c N | | j | j | j S rH ra r! s r rC z*_AsyncGeneratorContextManager._recreate_cm rc r c K | j d {V S # t $ r t d d w xY wre )rY __anext__StopAsyncIterationrh r! s r r: z(_AsyncGeneratorContextManager.__aenter__ sb C ++-------- -! C C C 788d B Cs # >c K |@ | j d {V t d # t $ r Y d S w xY w|
| } | j ||| d {V t d # t $ r}||ucY d }~S d }~wt $ r;}||u rY d }~dS t |t t f r|j |u rY d }~dS d }~wt $ r}||ur Y d }~d S d }~ww xY w)Nrj z$generator didn't stop after athrow()F) rY rv rh rw athrow
isinstancerg rm
BaseException)r" typrp r) rq s r r< z'_AsyncGeneratorContextManager.__aexit__ s ;
<h((********* ##:;;; &
}
hooc5)<<<<<<<<<"#IJJJ%
(
(
(%'''''''
%<< 55555 em5G%HII %}-- $uuuuu
e## $#####
sQ 4
AA1B
C6
BC6
C6#C-%CC
C6&C11C6N)r
r r r3 rC r: r< r
r r rs rs sL +*? ? ? C C C ! ! ! ! !r rs c < t fd }|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 & t | | S rH )r_ rK rL rM s r helperz contextmanager..helper' s 'dD999r r rM r s ` r r r s3 6 4[[: : : : [: Mr c < t fd }|S )a @asynccontextmanager decorator.
Typical usage:
@asynccontextmanager
async def some_async_generator():
try:
yield
finally:
This makes this:
async with some_async_generator() as :
equivalent to this:
try:
=
finally:
c & t | | S rH )rs r s r r z#asynccontextmanager..helperH s ,T4>>>r r r s ` r r r - s3 6 4[[? ? ? ? [? Mr c $ e Zd ZdZd Zd Zd 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 || _ d S rH thingr" r s r r] zclosing.__init___
r c | j S rH r r! s r r# zclosing.__enter__a s
zr c 8 | j d S rH )r closer" ro s r r* zclosing.__exit__c s
r Nr
r r r3 r] r# r* r
r r r r N sK r r c $ e Zd ZdZd Zd Zd ZdS )r a Async context manager for safely finalizing an asynchronously cleaned-up
resource such as an async generator, calling its ``aclose()`` method.
Code like this:
async with aclosing(.fetch()) as agen:
is equivalent to this:
agen = .fetch()
try:
finally:
await agen.aclose()
c || _ d S rH r r s r r] zaclosing.__init__y r r c K | j S rH r r! s r r: zaclosing.__aenter__{ s zr c H K | j d {V d S rH )r acloser s r r< zaclosing.__aexit__} s2 j !!!!!!!!!!!r N)r
r r r3 r] r: r< r
r r r r g sK " " " " " "r r c $ e Zd ZdZd Zd Zd ZdS )_RedirectStreamNc " || _ g | _ d S rH ) _new_target _old_targets)r"
new_targets r r] z_RedirectStream.__init__ s % r c | j t t | j t t | j | j | j S rH )r appendrZ rn _streamsetattrr r! s r r# z_RedirectStream.__enter__ sE dl!;!;<<<T\4#3444 r c j t t | j | j d S rH )r rn r r popr" exctypeexcinstexctbs r r* z_RedirectStream.__exit__ s+ T\4#4#8#8#:#:;;;;;r )r
r r r r] r# r* r
r r r r sF G
< < < < ._exit_wrapper s Hd #d # # # # #r r
rK rL r r s `` @r _create_cb_wrapperz!_BaseExitStack._create_cb_wrapper s9 4 $ $ $ $ $ $ $ r c , t | _ d S rH )r _exit_callbacksr! s r r] z_BaseExitStack.__init__ s $ww r c r t | }| j |_ t | _ |S )z@Preserve the context stack by transferring it to a new instance.)r[ r r )r" new_stacks r pop_allz_BaseExitStack.pop_all s0 DJJLL $($8 ! $ww r c t | } |j }| || n%# t $ r | | Y nw xY w|S )a Registers a callback with the standard __exit__ method signature.
Can suppress exceptions the same way __exit__ method can.
Also accepts any object with an __exit__ method (registering a call
to the method instead of the object itself).
)r[ r*
_push_cm_exitAttributeError_push_exit_callbackr" exit_cb_type exit_methods r pushz_BaseExitStack.push su :: 2"+K
t[ 1 1 1 1 + + + $ $T * * * * * +
s / AAc 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_exitresults r
enter_contextz _BaseExitStack.enter_context sE 88!##B'' 2u%%%
r c , | ^}}} n_# t $ rR}t | d d }d | }t | dd}~ww xY w |j |g| R i |}||_ | | |S )z\Registers an arbitrary callback and arguments.
Cannot suppress exceptions.
( "Not enough positional arguments {}N)
ValueErrorstr partitionformat TypeErrorr __wrapped__r rK rL r" r rq exc_detailsmsgr s r r z_BaseExitStack.callback s +$( !D(TT + + + c((,,S11!4K6==kJJCC..d * + 0/H4HHH4HH
%-
! ///
A$A
A A$c ^ | || }| |d dS )z;Helper to correctly register callbacks to __exit__ methods.TN)r r r" r r r s r r z _BaseExitStack._push_cm_exit s3 11"g>>
55555r Tc > | j ||f d S rH )r r )r" r is_syncs r r z"_BaseExitStack._push_exit_callback s$ ##Wh$788888r N)T)r
r r r3 staticmethodr r r] r r r r r r r
r r r r s 88' ' \' \ ' ' ' ( (6 6 6
9 9 9 9 9 9r r c $ e Zd ZdZd Zd Zd ZdS )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 | S rH r
r! s r r# zExitStack.__enter__+ s r c
|d d u}t j d
fd}d}d}| j rj| j \ }}|sJ || rd}d}d}n6# t j } ||d |d d}|}Y nxY w| j j|r2 |d j } |d # t
$ r | |d _ w xY w|o|S )Nr rk c F | j }||u rd S ||u rn|} || _ d S rH __context__new_excold_exc exc_context frame_excs r _fix_exception_contextz2ExitStack.__exit__.._fix_exception_context4 M
& % 1 '))F &+*B*B%
& #*G r FTNNNrn ro r r r r{ ) r" r received_excr suppressed_exc
pending_raiser cbnew_exc_details fixed_ctxr s @r r* zExitStack.__exit__. sA "1~T1 LNN1% * * * * *
"
. .2244KGRNN7
.2{# 5%)N$)M"4K
."%,..&&q'9;q>JJJ $
- "
.
(N 6 !!n$
-6 A*
..s A" "1B"B7 7Cc 4 | ddd dS z%Immediately unwind the context stack.N)r* r! s r r zExitStack.close_ s
dD$'''''r N)r
r r r3 r# r* r r
r r r r sL // // //b( ( ( ( (r r c h e Zd ZdZed Zed Zd Zd Zd Z d Z
d Z d Z d
Z
d S ) r a Async context manager for dynamic management of a stack of exit
callbacks.
For example:
async with AsyncExitStack() as stack:
connections = [await stack.enter_async_context(get_connection())
for i in range(5)]
# All opened connections will automatically be released at the
# end of the async with statement, even if attempts to open a
# connection later in the list raise an exception.
c " t || S rH r r s r _create_async_exit_wrapperz)AsyncExitStack._create_async_exit_wrapperr r r c " ^ fd}|S )Nc ( K i d {V d S rH r
r s r r z>AsyncExitStack._create_async_cb_wrapper.._exit_wrapperz s4 (D)D)) ) ) ) ) ) ) ) ) )r r
r s `` @r _create_async_cb_wrapperz'AsyncExitStack._create_async_cb_wrapperv s9 4 * * * * * * * r c K t | }|j }| | d{V }| || |S )zEnters the supplied async context manager.
If successful, also pushes its __aexit__ method as a callback and
returns the result of the __aenter__ method.
N)r[ r< r: _push_async_cm_exitr s r enter_async_contextz"AsyncExitStack.enter_async_context~ s[ 88" **2........ U+++
r c t | } |j }| || n&# t $ r | |d Y nw xY w|S )a# Registers a coroutine function with the standard __aexit__ method
signature.
Can suppress exceptions the same way __aexit__ method can.
Also accepts any object with an __aexit__ method (registering a call
to the method instead of the object itself).
F)r[ r< r r r r s r push_async_exitz AsyncExitStack.push_async_exit sw :: 8",K
$ $T; 7 7 7 7 2 2 2 $ $T5 1 1 1 1 1 2
s / AAc . | ^}}} n_# t $ rR}t | d d }d | }t | dd}~ww xY w |j |g| R i |}||_ | |d |S )zfRegisters an arbitrary coroutine function and arguments.
Cannot suppress exceptions.
r r r NF)r r r r r r r r r s r push_async_callbackz"AsyncExitStack.push_async_callback s +$( !D(TT + + + c((,,S11!4K6==kJJCC..d * + 65hNNNNNN
%-
! 666r c D K | ddd d{V dS r )r< r! s r r zAsyncExitStack.aclose s4 nnT4...........r c ^ | || }| |d dS )zLHelper to correctly register coroutine function to __aexit__
method.FN)r r r s r r z"AsyncExitStack._push_async_cm_exit s5 77GDD
66666r c
K | S rH r
r! s r r: zAsyncExitStack.__aenter__ s r c K |d d u}t j d fd}d}d}| j rv| j \ }} |r || }n || d {V }|rd}d}d}n6# t j } || d |d d}| }Y nxY w| j v|r2 |d j }
|d # t
$ r |
|d _ w xY w|o|S )Nr rk c F | j }||u rd S ||u rn|} || _ d S rH r r s r r z8AsyncExitStack.__aexit__.._fix_exception_context r r FTr r ) r" r r r r r r r cb_suppressr r r s @r r< zAsyncExitStack.__aexit__ si "1~T1 LNN1% * * * * *
" . .2244KGR
. 9"$"k"2KK(*K(8"8"8"8"8"8"8K 5%)N$)M"4K
."%,..&&q'9;q>JJJ $
- # " .$
(N 6 !!n$
-6 A*
..s A0 01B#0C C N)r
r r r3 r r r r r r r r r: r< r
r r r r e s
' ' \' \
$ (/ / /7 7 7 3/ 3/ 3/ 3/ 3/r r c 2 e Zd ZdZddZd Zd Zd Zd ZdS ) r aO 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 || _ d S rH enter_result)r" r s r r] znullcontext.__init__ s (r c | j S rH r r! s r r# znullcontext.__enter__ s r c d S rH r
r" excinfos r r* znullcontext.__exit__ r r c K | j S rH r r! s r r: znullcontext.__aenter__ s r c
K d S rH r
r s r r< znullcontext.__aexit__
s r rH ) r
r r r3 r] r# r* r: r< r
r r r r sn ) ) ) )! ! !
! ! !
r r c 4 e Zd ZdZ fdZd Zd Zd Z xZS )r z+Backwards compatibility alias for ExitStackc t j dt t t | d S )Nz*ContextStack has been renamed to ExitStack)r@ rA rB superr r] )r" rb s r r] zContextStack.__init__ s?
B( * * *
lD!!**,,,,,r c , | | S rH )r )r" r s r
register_exitzContextStack.register_exit s yy"""r c $ | j |g|R i |S rH )r )r" r rK rL s r registerzContextStack.register s$ t}X5555555r c * | S rH )r r! s r preservezContextStack.preserve s ||~~ r ) r
r r r3 r] r r" r$
__classcell__)rb s @r r r sk 55- - - - -
# # #6 6 6 r )$r3 r6 rn r@ r, collectionsr functoolsr typesr r ImportError__all__ABCr r objectr rR rW r_ rs r r r r r r r r r r r r r r
r r r- sb G G
"""""""
I I I N SW * #' .% % % % %v % % %P
F
$AF AF AF AF AF ; 5 /AF AF AFH2 2 2 2 2$@$?$92 2 2j B B $ 2" " " " "* " " "4< < < < <, < < <$
o
o M M M M M% M M M