
    Ag                     f   d dl Z d dlZd dlZd dlmZmZmZ d dlZd dlmZ d dl	m
Z
 d dlmZ  ej                  e      Z G d d e j                   dd	      ej"                        Z G d
 de      Z G d dej*                        Z G d de      Zdej0                  dee   dej4                  fdZy)    N)CallableOptionalType)_common)cygrpc)MetadataTypec                       e Zd Zy)_AuthMetadataContextN)__name__
__module____qualname__     J/var/www/openai/venv/lib/python3.12/site-packages/grpc/_plugin_wrapping.pyr
   r
      s     	r   r
   AuthMetadataContext)service_urlmethod_namec                       e Zd Zd Zy)_CallbackStatec                 R    t        j                         | _        d| _        d | _        y )NF)	threadingLocklockcalled	exception)selfs    r   __init__z_CallbackState.__init__*   s    NN$	r   N)r   r   r   r   r   r   r   r   r   )   s    r   r   c                   N    e Zd ZU eed<   eed<   dedefdZdedee	e
      fdZy	)
_AuthMetadataPluginCallback_state	_callbackstatecallbackc                      || _         || _        y N)r    r!   )r   r"   r#   s      r   r   z$_AuthMetadataPluginCallback.__init__4   s    !r   metadataerrorc           	      4   | j                   j                  5  | j                   j                  3| j                   j                  rt	        d      d| j                   _        n.t	        dj                  | j                   j                              	 d d d        |,| j                  |t        j                  j                  d        y | j                  d t        j                  j                  t        j                  t        |                   y # 1 sw Y   xY w)Nz2AuthMetadataPluginCallback invoked more than once!Tz1AuthMetadataPluginCallback raised exception "{}"!)r    r   r   r   RuntimeErrorformatr!   r   
StatusCodeokinternalr   encodestr)r   r&   r'   s      r   __call__z$_AuthMetadataPluginCallback.__call__8   s     [[{{$$,;;%%&L  *.DKK&"GNN--  '  =NN8V%6%6%9%94@NNf''00'..U2L# s   A8DDN)r   r   r   r   __annotations__r   r   r   r   r   BaseExceptionr0   r   r   r   r   r   0   s@    "n " "$-5d=6I-Jr   r   c                   `    e Zd ZU ej                  ed<   dej                  fdZdededefdZ	y)	_Plugin_metadata_pluginmetadata_pluginc                 r    || _         d | _        	 dd l}|j                         | _        y # t        $ r Y y w xY w)Nr   )r5   _stored_ctxcontextvarscopy_contextImportError)r   r6   r9   s      r   r   z_Plugin.__init__T   s?     /		
  +779D 		s   * 	66r   r   r#   c           
      &   t        t        j                  |      t        j                  |            }t               }	 | j	                  |t        ||             y # t        $ r}t        j                  d| j                         |j                  5  ||_        |j                  r	 d d d        Y d }~y 	 d d d        n# 1 sw Y   nxY w |d t        j                  j                  t        j                  t        |                   Y d }~y d }~ww xY w)Nz1AuthMetadataPluginCallback "%s" raised exception!)r
   r   decoder   r5   r   	Exception_LOGGERr   r   r   r   r+   r-   r.   r/   )r   r   r   r#   contextcallback_stater   s          r   r0   z_Plugin.__call__c   s    &NN;')D
 ()	!!4^XN  	C%%  $$+4(!(( %$( %$$ f''00'..Y2P 	s6   A 	D$,DB<%D3	D<C	ADDN)
r   r   r   grpcAuthMetadataPluginr1   r   r/   r   r0   r   r   r   r4   r4   Q   s:    ---(?(? C c X r   r4   r6   namereturnc                     |	 | j                   }n|}t        j                  t        j                  t        |       t        j                  |                  S # t        $ r | j                  j                   }Y fw xY wr%   )
r   AttributeError	__class__rB   CallCredentialsr   MetadataPluginCallCredentialsr4   r   r.   )r6   rD   effective_names      r    metadata_plugin_call_credentialsrL   z   sy     |	@,55N ,,O$gnn^&D	
 	  	@,66??N	@s   A A:9A:)collectionsloggingr   typingr   r   r   rB   r   grpc._cythonr   grpc._typingr   	getLoggerr   r?   
namedtupler   r
   objectr   AuthMetadataPluginCallbackr   r4   rC   r/   rI   rL   r   r   r   <module>rV      s       + +    %
'

H
%
	K	
 	
	V $"A"A B&f &R,,4<SM	r   