o
    b<                     @   s   d Z ddlZddlZddlZddlZddlZddlmZ ddlZee	Z
dZG dd deddejZG d	d
 d
ejZG dd dejZdd ZdS )z#Internal utilities for gRPC Python.    N)_commonz6Exception calling connectivity future "done" callback!c                   @   s   e Zd ZdS )RpcMethodHandlerN)__name__
__module____qualname__ r   r   /lib/python3.10/site-packages/grpc/_utilities.pyr      s    r   Z_RpcMethodHandler)Zrequest_streamingZresponse_streamingZrequest_deserializerZresponse_serializerZunary_unaryZunary_streamZstream_unaryZstream_streamc                   @   s$   e Zd Zdd Zdd Zdd ZdS )DictionaryGenericHandlerc                    s$    | _  fddt|D | _d S )Nc                    s   i | ]\}}t  ||qS r   )r   Zfully_qualified_method).0methodZmethod_handlerservicer   r   
<dictcomp>1   s    z5DictionaryGenericHandler.__init__.<locals>.<dictcomp>)_namesixZ	iteritems_method_handlers)selfr   Zmethod_handlersr   r   r   __init__/   s   
z!DictionaryGenericHandler.__init__c                 C   s   | j S N)r   r   r   r   r   service_name6   s   z%DictionaryGenericHandler.service_namec                 C   s   | j |jS r   )r   getr   )r   Zhandler_call_detailsr   r   r   r   9      z DictionaryGenericHandler.serviceN)r   r   r   r   r   r   r   r   r   r   r	   -   s    r	   c                   @   sz   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	dddZ
dddZdddZdd Zdd Zdd ZdS )_ChannelReadyFuturec                 C   s&   t  | _|| _d| _d| _g | _d S )NF)	threadingZ	Condition
_condition_channel_matured
_cancelled_done_callbacks)r   channelr   r   r   r   ?   s
   

z_ChannelReadyFuture.__init__c                 C   s   |d u rd nt   | }| j5 	 | jrt | jr$	 W d    d S |d u r.| j  n|t    }|dk r<t | jj|d q1 sGw   Y  d S )NTr   )timeout)timer   r   grpcZFutureCancelledErrorr   waitZFutureTimeoutError)r   r!   ZuntilZ	remainingr   r   r   _blockG   s    z_ChannelReadyFuture._blockc              	   C   s   | j 2 | js%|tjju r%d| _| j| j | j 	  t
| j}d | _n		 W d    d S W d    n1 s8w   Y  |D ]}z||  W q? tyU   tt Y q?w d S )NT)r   r   r#   ZChannelConnectivityZREADYr   r   unsubscribe_update
notify_alltupler   	Exception_LOGGER	exception$_DONE_CALLBACK_EXCEPTION_LOG_MESSAGE)r   Zconnectivitydone_callbacksdone_callbackr   r   r   r'   Y   s(   


z_ChannelReadyFuture._updatec              	   C   s   | j , | jsd| _| j| j | j   t| j}d | _n		 W d    dS W d    n1 s2w   Y  |D ]}z||  W q9 t	yO   t
t Y q9w dS )NTF)r   r   r   r   r&   r'   r(   r)   r   r*   r+   r,   r-   )r   r.   r/   r   r   r   cancelk   s$   


z_ChannelReadyFuture.cancelc                 C   s0   | j  | jW  d    S 1 sw   Y  d S r   )r   r   r   r   r   r   	cancelled~   s   $z_ChannelReadyFuture.cancelledc                 C   s:   | j  | j o| j W  d    S 1 sw   Y  d S r   r   r   r   r   r   r   r   running   s   $z_ChannelReadyFuture.runningc                 C   s6   | j  | jp	| jW  d    S 1 sw   Y  d S r   r2   r   r   r   r   done   s   
$z_ChannelReadyFuture.doneNc                 C      |  | d S r   r%   r   r!   r   r   r   result   r   z_ChannelReadyFuture.resultc                 C   r5   r   r6   r7   r   r   r   r,      r   z_ChannelReadyFuture.exceptionc                 C   r5   r   r6   r7   r   r   r   	traceback   r   z_ChannelReadyFuture.tracebackc                 C   s\   | j  | js| js| j| 	 W d    d S W d    n1 s#w   Y  ||  d S r   )r   r   r   r   append)r   fnr   r   r   add_done_callback   s    z%_ChannelReadyFuture.add_done_callbackc                 C   s>   | j  | jj| jdd W d    d S 1 sw   Y  d S )NT)Ztry_to_connect)r   r   Z	subscriber'   r   r   r   r   start   s   "z_ChannelReadyFuture.startc                 C   sf   | j & | js| js!| j| j W d    d S W d    d S W d    d S 1 s,w   Y  d S r   )r   r   r   r   r&   r'   r   r   r   r   __del__   s   "z_ChannelReadyFuture.__del__r   )r   r   r   r   r%   r'   r0   r1   r3   r4   r8   r,   r9   r<   r=   r>   r   r   r   r   r   =   s    


r   c                 C   s   t | }|  |S r   )r   r=   )r    Zready_futurer   r   r   channel_ready_future   s   r?   )__doc__collectionsZloggingr   r"   r#   r   r   Z	getLoggerr   r+   r-   
namedtupler   ZServiceRpcHandlerr	   ZFuturer   r?   r   r   r   r   <module>   s$   


	h