o
6��fe � @ s| d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZm Z g d�Z
G dd� de j �Z G dd � d e j �Z
G d
d � d e�ZG d d
� d
e�ZG dd� d�ZG dd� dee e�ZG dd� dee
e�Zdd� Zdd� ZG dd� de �ZG dd� de
�ZG d d � d e �ZG d d � d e�ZG d d!� d!e�ZG 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)�deque��wraps)�
MethodType� GenericAlias)
�asynccontextmanager�contextmanager�closing� nullcontext�AbstractContextManager�AbstractAsyncContextManager�AsyncExitStack�ContextDecorator� ExitStack�redirect_stdout�redirect_stderr�suppress�aclosingc @ �8 e Zd Z ee�Zdd� 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�|dd�S tS )Nr r )r �_collections_abc�_check_methods�NotImplemented��cls�Cr r r �__subclasshook__ s z'AbstractContextManager.__subclasshook__N) �__name__�
__module__� __qualname__� classmethodr �__class_getitem__r �abc�abstractmethodr 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 ��args�kwds��funcr r r �innerL s
$�z(ContextDecorator.__call__.<locals>.innerr �r r; r< r r: r �__call__K � zContextDecorator.__call__N�r) 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 �0�z-AsyncContextDecorator.__call__.<locals>.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 �getattr�typerC )r r; r8 r9 �docr r r �__init__f s
z%_GeneratorContextManagerBase.__init__c C s | � | j| j| j�S 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| j�W S ty td�d �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|t�rf|j|u rfW Y d }~dS � d }~w ty� } z ||urw� W Y d }~dS d }~ww )NF�generator didn't stopz#generator didn't stop after throw())rL rD rM rN �throw�
isinstance� __cause__�
BaseException�r �typ�valuer �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| j�I d H W S ty td�d �w rK )r8 r9 r; �anextrD �StopAsyncIterationrN r r r r r1 � s �
�z(_AsyncGeneratorContextManager.__aenter__c
� s �|d u r z
t | j�I 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|ttf�ro|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.<locals>.helperr �r; r` r r_ r r � � r c r] )Nc r^ r )rY r7 r_ r r r` 8 ra z#asynccontextmanager.<locals>.helperr rb r r_ r r rc r c @ �& e Zd Z dd� Zdd� Zdd� ZdS )r c C �
|| _ d S r ��thing�r 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 �close�r �exc_infor r r r S s zclosing.__exit__N�r) 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�_stream�setattrrt r r r r r z s z_RedirectStream.__enter__c C s t t| j| j�� � d S r )rz rx ry ru �pop�r �exctype�excinst�exctbr 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 �stdoutN�r) 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 || j�S 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 ddd�Z dS )�_BaseExitStackc C �
t || �S r �r ��cm�cm_exitr r r �_create_exit_wrapper� �
z#_BaseExitStack._create_exit_wrapperc � � ��fdd�}|S )Nc s �� i ��� d S r r �r rW �tb�r8 �callbackr9 r r �
_exit_wrapper� s z8_BaseExitStack._create_cb_wrapper.<locals>._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_exit�r �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�_exit�resultr 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_exc�old_exc� exc_context�� frame_excr r �_fix_exception_context � �
z2ExitStack.__exit__.<locals>._fix_exception_contextFT�NNN�rx rn r� r{ r� rS )
r � exc_details� received_excr� �suppressed_exc�
pending_raiser� �cb�new_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.<locals>._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__.<locals>._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 dd�Zdd� 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 �<module> sH
�
@�A!! PE