o
0ie @ s| d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZm Z g dZ
G dd de j Z G dd d e j Z
G d
d d eZG d d
d
eZG dd dZG dd dee eZG dd dee
eZdd Zdd ZG dd de ZG dd de
ZG d d d e ZG d d d eZG d d! d!eZG d"d# d#e ZG d$d% d%Z G d&d' d'e e Z G d(d) d)e e
Z G d*d+ d+e e
Z dS ), N)dequewraps)
MethodType GenericAlias)
asynccontextmanagercontextmanagerclosing nullcontextAbstractContextManagerAbstractAsyncContextManagerAsyncExitStackContextDecorator ExitStackredirect_stdoutredirect_stderrsuppressaclosingc @ 8 e Zd Z eeZdd Zejdd Z edd Z
dS )r c C | S N selfr r 1/opt/alt/python310/lib64/python3.10/contextlib.py __enter__ s z AbstractContextManager.__enter__c C s d S r r r exc_type exc_value tracebackr r r __exit__ s z AbstractContextManager.__exit__c C | t u r t|ddS tS )Nr r )r _collections_abc_check_methodsNotImplementedclsCr r r __subclasshook__ s z'AbstractContextManager.__subclasshook__N) __name__
__module__ __qualname__ classmethodr __class_getitem__r abcabstractmethodr r( r r r r r
r c @ r )r c s | S r r r r r r
__aenter__+ s z&AbstractAsyncContextManager.__aenter__c s d S r r r r r r __aexit__/ s z%AbstractAsyncContextManager.__aexit__c C r! )Nr1 r2 )r r" r# r$ r% r r r r( 4 s
z,AbstractAsyncContextManager.__subclasshook__N) r) r* r+ r, r r- r1 r. r/ r2 r( r r r r r % r0 r c @ e Zd Z dd Zdd ZdS )r c C r r r r r r r _recreate_cm? s z ContextDecorator._recreate_cmc t fdd}|S )Nc s: | i |W d S 1 sw Y d S r r4 argskwdsfuncr r r innerL s
$z(ContextDecorator.__call__..innerr r r; r< r r: r __call__K zContextDecorator.__call__Nr) r* r+ r4 r> r r r r r < s r c @ r3 )AsyncContextDecoratorc C r r r r r r r r4 V s z"AsyncContextDecorator._recreate_cmc r5 )Nc sV 4 I d H | i |I d H W d I d H S 1 I d H s$w Y d S r r6 r7 r: r r r< \ s 0z-AsyncContextDecorator.__call__..innerr r= r r: r r> [ r? z AsyncContextDecorator.__call__Nr@ r r r r rA S s rA c @ r3 ) _GeneratorContextManagerBasec C sN ||i || _ |||| _| _| _t|dd }|d u r"t| j}|| _d S )N__doc__)genr; r8 r9 getattrtyperC )r r; r8 r9 docr r r __init__f s
z%_GeneratorContextManagerBase.__init__c C s | | j| j| jS r ) __class__r; r8 r9 r r r r r4 t s z)_GeneratorContextManagerBase._recreate_cmN)r) r* r+ rH r4 r r r r rB c s rB c @ r3 )_GeneratorContextManagerc C s2 | ` | `| `zt| jW S ty tdd w Nzgenerator didn't yield)r8 r9 r; nextrD
StopIteration RuntimeErrorr r r r r s
z"_GeneratorContextManager.__enter__c
C s |d u rz
t | j W td ty Y dS w |d u r | }z
| j||| W td tyC } z
||uW Y d }~S d }~w tyk } z ||u rUW Y d }~dS t|trf|j|u rfW Y d }~dS d }~w ty } z ||urw W Y d }~dS d }~ww )NFgenerator didn't stopz#generator didn't stop after throw())rL rD rM rN throw
isinstance __cause__
BaseExceptionr typvaluer excr r r r s<
z!_GeneratorContextManager.__exit__N)r) r* r+ r r r r r r rJ { rJ c @ r3 ) _AsyncGeneratorContextManagerc s: | ` | `| `z t| jI d H W S ty tdd w rK )r8 r9 r; anextrD StopAsyncIterationrN r r r r r1 s
z(_AsyncGeneratorContextManager.__aenter__c
s |d u r z
t | jI d H W td ty Y dS w |d u r$| }z| j|||I d H W td tyJ } z
||uW Y d }~S d }~w tyt } z ||u r\W Y d }~dS t|ttfro|j|u roW Y d }~dS d }~w ty } z ||ur W Y d }~dS d }~ww )NFrO z$generator didn't stop after athrow()) rZ rD r[ rN athrowrQ rM rR rS rT r r r r2 s>
z'_AsyncGeneratorContextManager.__aexit__N)r) r* r+ r1 r2 r r r r rY rX rY c t fdd}|S )Nc t | |S r )rJ r7 r; r r helper z contextmanager..helperr r; r` r r_ r r r c r] )Nc r^ r )rY r7 r_ r r r` 8 ra z#asynccontextmanager..helperr rb r r_ r r rc r c @ & e Zd Z dd Zdd Zdd ZdS )r c C
|| _ d S r thingr rg r r r rH O
zclosing.__init__c C | j S r rf r r r r r Q zclosing.__enter__c G s | j d S r )rg closer exc_infor r r r S s zclosing.__exit__Nr) r* r+ rH r r r r r r r > s
r c @ rd )r c C re r rf rh r r r rH i ri zaclosing.__init__c | j S r rf r r r r r1 k zaclosing.__aenter__c s | j I d H d S r )rg acloserm r r r r2 m s zaclosing.__aexit__N)r) r* r+ rH r1 r2 r r r r r W s
r c @ s( e Zd ZdZdd Zdd Zdd ZdS )_RedirectStreamNc C s || _ g | _d S r ) _new_target _old_targets)r
new_targetr r r rH u s
z_RedirectStream.__init__c C s* | j tt| j tt| j| j | jS r )ru appendrE sys_streamsetattrrt r r r r r z s z_RedirectStream.__enter__c C s t t| j| j d S r )rz rx ry ru popr exctypeexcinstexctbr r r r s z_RedirectStream.__exit__)r) r* r+ ry rH r r r r r r rs q s
rs c @ e Zd Z dZdS )r stdoutNr) r* r+ ry r r r r r s r c @ r )r stderrNr r r r r r s r c @ rd )r c G re r ) _exceptions)r
exceptionsr r r rH ri zsuppress.__init__c C d S r r r r r r r zsuppress.__enter__c C s |d uo t || jS r )
issubclassr r| r r r r s
zsuppress.__exit__Nro r r r r r s
r c @ s` e Zd Z edd Zedd Zdd Zdd Zd d
Zd d Z d
d Z
dd Z dddZ dS )_BaseExitStackc C
t || S r r cmcm_exitr r r _create_exit_wrapper
z#_BaseExitStack._create_exit_wrapperc fdd}|S )Nc s i d S r r r rW tbr8 callbackr9 r r
_exit_wrapper s z8_BaseExitStack._create_cb_wrapper.._exit_wrapperr r r8 r9 r r r r _create_cb_wrapper z!_BaseExitStack._create_cb_wrapperc C s t | _d S r )r _exit_callbacksr r r r rH s z_BaseExitStack.__init__c C s t | }| j|_t | _|S r )rF r r )r new_stackr r r pop_all s
z_BaseExitStack.pop_allc C sD t |}z|j}W n ty | | Y |S w | || |S r )rF r AttributeError_push_exit_callback
_push_cm_exitr exit_cb_type exit_methodr r r push s
z_BaseExitStack.pushc C s* t |}|j}||}| || |S r )rF r r r r r _cm_type_exitresultr r r
enter_context s
z _BaseExitStack.enter_contextc O s. | j |g|R i |}||_| | |S r )r __wrapped__r r r r8 r9 r r r r r s
z_BaseExitStack.callbackc C | ||}| |d d S )NT)r r r r r r r r r r s z _BaseExitStack._push_cm_exitTc C s | j ||f d S r )r rw )r r is_syncr r r r s z"_BaseExitStack._push_exit_callbackN)T)
r) r* r+ staticmethodr r rH r r r r r r r r r r r s
r c @ rd )r c C s | S r r r r r r r r zExitStack.__enter__c
s |d d u}t d fdd}d}d}| jrE| j \}}z || r+d}d}d}W n t }||d |d d}|}Y | js|r^z |d j} |d ty] | |d _ w |oa|S )Nr c 6 | j }|d u s ||u rd S | u rn|} q|| _ d S r __context__new_excold_exc exc_context frame_excr r _fix_exception_context
z2ExitStack.__exit__.._fix_exception_contextFTNNNrx rn r r{ r rS )
r exc_details received_excr suppressed_exc
pending_raiser cbnew_exc_details fixed_ctxr r r r s8
zExitStack.__exit__c C s | d d d d S r )r r r r r rl F s zExitStack.closeN)r) r* r+ r r rl r r r r r s
1r c @ s^ e Zd Z edd Zedd Zdd Zdd Zd d
Zd d Z d
d Z
dd Z dd Z dS )r
c C r r r r r r r _create_async_exit_wrapperY r z)AsyncExitStack._create_async_exit_wrapperc r )Nc s i I d H d S r r r r r r r _ s z>AsyncExitStack._create_async_cb_wrapper.._exit_wrapperr r r r r _create_async_cb_wrapper] r z'AsyncExitStack._create_async_cb_wrapperc s2 t |}|j}||I d H }| || |S r )rF r2 r1 _push_async_cm_exitr r r r enter_async_contextc s z"AsyncExitStack.enter_async_contextc C sF t |}z|j}W n ty | |d Y |S w | || |S NF)rF r2 r r r r r r r push_async_exito s
z AsyncExitStack.push_async_exitc O s0 | j |g|R i |}||_| |d |S r )r r r r r r r push_async_callback s
z"AsyncExitStack.push_async_callbackc s | d d d I d H d S r )r2 r r r r rr s zAsyncExitStack.aclosec C r r )r r r r r r r s z"AsyncExitStack._push_async_cm_exitc s | S r r r r r r r1 zAsyncExitStack.__aenter__c s |d d u}t d fdd}d}d}| jrR| j \}}z|r)|| }n|| I d H }|r8d}d}d}W n t } || d |d d}| }Y | js|rkz |d j}
|d tyj |
|d _ w |on|S )Nr r c r r r r r r r r r z8AsyncExitStack.__aexit__.._fix_exception_contextFTr r ) r r r r r r r r cb_suppressr r r r r r2 s@
zAsyncExitStack.__aexit__N)
r) r* r+ r r r r r r rr r r1 r2 r r r r r
L s
r
c @ s8 e Zd Z d ddZdd Zdd Zdd Zd
d ZdS )
r
Nc C re r enter_result)r r r r r rH ri znullcontext.__init__c C rj r r r r r r r rk znullcontext.__enter__c G r r r r excinfor r r r r znullcontext.__exit__c rp r r r r r r r1 rq znullcontext.__aenter__c s d S r r r r r r r2 r znullcontext.__aexit__r )r) r* r+ rH r r r1 r2 r r r r r
s
r
) r. rx r" collectionsr functoolsr typesr r __all__ABCr r objectr rA rB rJ rY r r r r rs r r r r r r
r
r r r r sH
@A!! PE