
    Ngi              	          d Z ddlZddlmZ ddlmZ ddlmZ ddlmZ ddlm	Z	 ddl
mZ ej        j        ej        j        j        ej        fej        j        ej        j        j        ej        fej        j        ej        j        j        ej        fej        j        ej        j        j        ej        fiZd	 Zd
 Zd Zd Z G d dej                  Z  G d de	j!        ej"                  Z#d Z$d Z%d Z&d Z'd Z(d Z) G d dej*                  Z+ G d dej,                  Z- G d dej.                  Z/ G d dej0                  Z1 G d d ej2                  Z3 G d! d"ej4                  Z5d# Z6d$ Z7dS )%zCTranslates gRPC's client-side API into gRPC's client-side Beta API.    N)_common)	_metadata)
interfaces)cardinality)future)facec                 ,    | dn| }||S  ||          S )N r
   )metadatametadata_transformernon_none_metadatas      Y/var/www/html/ai-engine/env/lib/python3.11/site-packages/grpc/beta/_client_adaptations.py_effective_metadatar   .   s/    &.H#  ##$5666    c                     | d n| j         S N)credentials)grpc_call_optionss    r   _credentialsr   6   s    $,442C2OOr   c                 :   |                                  }t                              |          }|t          j        j        j        n|d         }t          j        ||                                 |                                 || 	                                          S )Nr   )
code6_STATUS_CODE_TO_ABORTION_KIND_AND_ABORTION_ERROR_CLASSgetr   AbortionKindLOCAL_FAILUREinitial_metadatatrailing_metadatadetails)rpc_error_callr   pair
error_kinds       r   	_abortionr#   :   s      DAEEdKKD59\#11tAwJ=''))((**    r   c                    |                                  }t                              |          }|t          j        n|d         } ||                                 |                                 ||                                           S )N   )r   r   r   r   AbortionErrorr   r   r   )r    r   r!   exception_classs       r   _abortion_errorr(   G   s      DAEEdKKD,0Ld((d1gO?''))((**  	  r   c                       e Zd Zd ZdS )_InvocationProtocolContextc                     d S r   r
   selfs    r    disable_next_request_compressionz;_InvocationProtocolContext.disable_next_request_compressionT   s    r   N)__name__
__module____qualname__r.   r
   r   r   r*   r*   S   s#            r   r*   c                       e Zd Zd Zd Zd Zd Zd ZddZddZ	dd	Z
d
 Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd ZdS )_Rendezvousc                 0    || _         || _        || _        d S r   )_future	_iterator_call)r-   response_futureresponse_iteratorcalls       r   __init__z_Rendezvous.__init__Y   s    &*


r   c                 4    | j                                         S r   )r7   cancelr,   s    r   r=   z_Rendezvous.cancel^   s    z  """r   c                 4    | j                                         S r   )r5   	cancelledr,   s    r   r?   z_Rendezvous.cancelleda   s    |%%'''r   c                 4    | j                                         S r   )r5   runningr,   s    r   rA   z_Rendezvous.runningd   s    |##%%%r   c                 4    | j                                         S r   )r5   doner,   s    r   rC   z_Rendezvous.doneg   s    |  """r   Nc                 
   	 | j                             |          S # t          j        $ r}t	          |          d }~wt          j        $ r t          j                    t          j        $ r t          j	                    w xY wN)timeout)
r5   resultgrpcRpcErrorr(   FutureTimeoutErrorr   TimeoutErrorFutureCancelledErrorCancelledErrorr-   rF   r    s      r   rG   z_Rendezvous.resultj   s    	*<&&w&777} 	2 	2 	2!.111& 	( 	( 	(%'''( 	* 	* 	*')))	*s    B;ABc                     	 | j                             |          }|d S t          |          S # t          j        $ r t          j                    t          j        $ r t          j                    w xY wrE   )	r5   	exceptionr(   rH   rJ   r   rK   rL   rM   rN   s      r   rP   z_Rendezvous.exceptiont   s    		*!\33G3DDN%t&~666& 	( 	( 	(%'''( 	* 	* 	*')))	*s   0 0 AA4c                     	 | j                             |          S # t          j        $ r t	          j                    t          j        $ r t	          j                    w xY wrE   )r5   	tracebackrH   rJ   r   rK   rL   rM   )r-   rF   s     r   rR   z_Rendezvous.traceback   sq    	*<))'):::& 	( 	( 	(%'''( 	* 	* 	*')))	*s
    AA!c                 F      j                              fd           d S )Nc                                S r   r
   )ignored_callbackfnr-   s    r   <lambda>z/_Rendezvous.add_done_callback.<locals>.<lambda>   s    4 r   )r5   add_done_callback)r-   rV   s   ``r   rX   z_Rendezvous.add_done_callback   s-    &&'H'H'H'H'HIIIIIr   c                     | S r   r
   r,   s    r   __iter__z_Rendezvous.__iter__       r   c                 x    	 t          | j                  S # t          j        $ r}t	          |          d }~ww xY wr   )nextr6   rH   rI   r(   )r-   r    s     r   _nextz_Rendezvous._next   sF    	2'''} 	2 	2 	2!.111	2s    949c                 *    |                                  S r   r^   r,   s    r   __next__z_Rendezvous.__next__       zz||r   c                 *    |                                  S r   r`   r,   s    r   r]   z_Rendezvous.next   rb   r   c                 4    | j                                         S r   )r7   	is_activer,   s    r   re   z_Rendezvous.is_active   s    z##%%%r   c                 4    | j                                         S r   )r7   time_remainingr,   s    r   rg   z_Rendezvous.time_remaining   s    z((***r   c                 b      fd} j                             |          }|rd n	 |            S )Nc                                                       t          j        j        ur  t	          j                             d S d S r   )r   rH   
StatusCodeOKr#   r7   )abortion_callbackr-   s   r   done_callbackz8_Rendezvous.add_abortion_callback.<locals>.done_callback   sD    yy{{$/"444!!)DJ"7"788888 54r   )r7   add_callback)r-   rl   rm   
registereds   ``  r   add_abortion_callbackz!_Rendezvous.add_abortion_callback   sN    	9 	9 	9 	9 	9 	9 Z,,];;
!6tt}}6r   c                     t                      S r   )r*   r,   s    r   protocol_contextz_Rendezvous.protocol_context   s    )+++r   c                 X    t          j        | j                                                  S r   )r   betar7   r   r,   s    r   r   z_Rendezvous.initial_metadata   s     ~dj99;;<<<r   c                 X    t          j        | j                                                  S r   )r   rt   r7   terminal_metadatar,   s    r   rv   z_Rendezvous.terminal_metadata   s     ~dj::<<===r   c                 4    | j                                         S r   )r7   r   r,   s    r   r   z_Rendezvous.code   s    z   r   c                 4    | j                                         S r   )r7   r   r,   s    r   r   z_Rendezvous.details   s    z!!###r   r   )r/   r0   r1   r;   r=   r?   rA   rC   rG   rP   rR   rX   rZ   r^   ra   r]   re   rg   rp   rr   r   rv   r   r   r
   r   r   r3   r3   X   sa         
# # #( ( (& & &# # #* * * *
* 
* 
* 
** * * *J J J  2 2 2    & & &+ + +7 7 7, , ,= = => > >! ! !$ $ $ $ $r   r3   c                    	 |                      t          j        ||          |	|
          }t          ||          }|rN|                    ||t          j        |          t          |                    \  }}|t          d d |          fS  |||t          j        |          t          |                    S # t          j
        $ r}t          |          d }~ww xY wN)request_serializerresponse_deserializer)rF   r   r   )unary_unaryr   fully_qualified_methodr   	with_callr   unbetar   r3   rH   rI   r(   )channelgroupmethodrF   r   protocol_optionsr   r   requestr{   r|   multi_callableeffective_metadataresponser:   r    s                   r   _blocking_unary_unaryr      s   . ,,*5&991"7 - 
 

 1;OPP 	+55")*<==()9::	 6  NHd [tT::::!>")*<==()9::	    = . . .n---.   B
B; -B; ;C
CCc
                 
   |                      t          j        ||          ||	          }
t          ||          }|
                    ||t          j        |          t          |                    }t          |d |          S rz   )	r}   r   r~   r   r   r   r   r   r3   )r   r   r   rF   r   r   r   r   r{   r|   r   r   r8   s                r   _future_unary_unaryr      s     ((&uf55-3 )  N
 -X7KLL$++!"455 !122	 ,  O o>>>r   c
                     |                      t          j        ||          ||	          }
t          ||          } |
||t	          j        |          t          |                    }t          d ||          S rz   )unary_streamr   r~   r   r   r   r   r3   )r   r   r   rF   r   r   r   r   r{   r|   r   r   r9   s                r   _unary_streamr      s     ))&uf55-3 *  N
 -X7KLL&!"455 !122	   t.0ABBBr   c                    	 |                      t          j        ||          |	|
          }t          ||          }|rN|                    ||t          j        |          t          |                    \  }}|t          d d |          fS  |||t          j        |          t          |                    S # t          j
        $ r}t          |          d }~ww xY wrz   )stream_unaryr   r~   r   r   r   r   r   r3   rH   rI   r(   )r   r   r   rF   r   r   r   r   request_iteratorr{   r|   r   r   r   r:   r    s                   r   _blocking_stream_unaryr     s   . --*5&991"7 . 
 

 1;OPP 	+55 ")*<==()9::	 6  NHd [tT::::!> ")*<==()9::	    = . . .n---.r   c
                 
   |                      t          j        ||          ||	          }
t          ||          }|
                    ||t          j        |          t          |                    }t          |d |          S rz   )	r   r   r~   r   r   r   r   r   r3   )r   r   r   rF   r   r   r   r   r{   r|   r   r   r8   s                r   _future_stream_unaryr   <  s     ))&uf55-3 *  N
 -X7KLL$++!"455 !122	 ,  O o>>>r   c
                     |                      t          j        ||          ||	          }
t          ||          } |
||t	          j        |          t          |                    }t          d ||          S rz   )stream_streamr   r~   r   r   r   r   r3   )r   r   r   rF   r   r   r   r   r{   r|   r   r   r9   s                r   _stream_streamr   W  s     **&uf55-3 +  N
 -X7KLL&!"455 !122	   t.0ABBBr   c                   6    e Zd Zd Z	 	 	 ddZddZ	 	 ddZdS )	_UnaryUnaryMultiCallablec                 Z    || _         || _        || _        || _        || _        || _        d S r   _channel_group_method_metadata_transformer_request_serializer_response_deserializerr-   r   r   r   r   r{   r|   s          r   r;   z!_UnaryUnaryMultiCallable.__init__s  7      %9"#5 &;###r   NFc                 p    t          | j        | j        | j        ||||| j        || j        | j                  S r   )r   r   r   r   r   r   r   )r-   r   rF   r   r   r   s         r   __call__z!_UnaryUnaryMultiCallable.__call__  sF     %MKL&$'
 
 	
r   c                 n    t          | j        | j        | j        |||| j        || j        | j        
  
        S r   )r   r   r   r   r   r   r   r-   r   rF   r   r   s        r   r   z_UnaryUnaryMultiCallable.future  sA    "MKL&$'
 
 	
r   c                     t                      r   NotImplementedErrorr-   r   receiverrl   rF   r   r   s          r   eventz_UnaryUnaryMultiCallable.event       "###r   NFNNNr/   r0   r1   r;   r   r   r   r
   r   r   r   r   r  sr        < < <( 
 
 
 
,
 
 
 
( 	$ 	$ 	$ 	$ 	$ 	$r   r   c                   (    e Zd Zd ZddZ	 	 ddZdS )_UnaryStreamMultiCallablec                 Z    || _         || _        || _        || _        || _        || _        d S r   r   r   s          r   r;   z"_UnaryStreamMultiCallable.__init__  r   r   Nc                 n    t          | j        | j        | j        |||| j        || j        | j        
  
        S r   )r   r   r   r   r   r   r   r   s        r   r   z"_UnaryStreamMultiCallable.__call__  sA    MKL&$'
 
 	
r   c                     t                      r   r   r   s          r   r   z_UnaryStreamMultiCallable.event  r   r   r   r/   r0   r1   r;   r   r   r
   r   r   r   r     sS        < < < 
 
 
 
( 	$ 	$ 	$ 	$ 	$ 	$r   r   c                   8    e Zd Zd Z	 	 	 ddZ	 ddZ	 	 ddZdS )	_StreamUnaryMultiCallablec                 Z    || _         || _        || _        || _        || _        || _        d S r   r   r   s          r   r;   z"_StreamUnaryMultiCallable.__init__  r   r   NFc                 p    t          | j        | j        | j        ||||| j        || j        | j                  S r   )r   r   r   r   r   r   r   )r-   r   rF   r   r   r   s         r   r   z"_StreamUnaryMultiCallable.__call__  sF     &MKL&$'
 
 	
r   c                 n    t          | j        | j        | j        |||| j        || j        | j        
  
        S r   )r   r   r   r   r   r   r   r-   r   rF   r   r   s        r   r   z _StreamUnaryMultiCallable.future  sC     $MKL&$'
 
 	
r   c                     t                      r   r   r-   r   rl   rF   r   r   s         r   r   z_StreamUnaryMultiCallable.event       "###r   r   r   r   r
   r   r   r   r     sy        < < <( 
 
 
 
. JN
 
 
 
* $ $ $ $ $ $r   r   c                   *    e Zd Zd Z	 ddZ	 	 ddZdS )_StreamStreamMultiCallablec                 Z    || _         || _        || _        || _        || _        || _        d S r   r   r   s          r   r;   z#_StreamStreamMultiCallable.__init__!  r   r   Nc                 n    t          | j        | j        | j        |||| j        || j        | j        
  
        S r   )r   r   r   r   r   r   r   r   s        r   r   z#_StreamStreamMultiCallable.__call__1  sC     MKL&$'
 
 	
r   c                     t                      r   r   r   s         r   r   z _StreamStreamMultiCallable.eventA  r   r   r   r   r
   r   r   r   r      sZ        < < <" JN
 
 
 
* $ $ $ $ $ $r   r   c                       e Zd 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
Z	 	 ddZ	 	 ddZd Zd Zd Zd Zd Zd ZdS )_GenericStubc                 F    || _         || _        |pi | _        |pi | _        d S r   )r   r   _request_serializers_response_deserializers)r-   r   r   request_serializersresponse_deserializerss        r   r;   z_GenericStub.__init__M  s3      %9"$7$=2!'='C$$$r   Nc                     | j                             ||f          }| j                            ||f          }	t          | j        ||||||| j        |||	          S r   )r   r   r   r   r   r   )
r-   r   r   r   rF   r   r   r   r{   r|   s
             r   blocking_unary_unaryz!_GenericStub.blocking_unary_unaryY  s     "6::
 
 !% < @ @!
 !
 %M&!
 
 	
r   c                     | j                             ||f          }| j                            ||f          }t          | j        |||||| j        |||
  
        S r   )r   r   r   r   r   r   	r-   r   r   r   rF   r   r   r{   r|   s	            r   future_unary_unaryz_GenericStub.future_unary_unary}  s     "6::
 
 !% < @ @!
 !
 #M&!
 
 	
r   c                     | j                             ||f          }| j                            ||f          }t          | j        |||||| j        |||
  
        S r   )r   r   r   r   r   r   r   s	            r   inline_unary_streamz _GenericStub.inline_unary_stream  s     "6::
 
 !% < @ @!
 !
 M&!
 
 	
r   c                     | j                             ||f          }| j                            ||f          }	t          | j        ||||||| j        |||	          S r   )r   r   r   r   r   r   )
r-   r   r   r   rF   r   r   r   r{   r|   s
             r   blocking_stream_unaryz"_GenericStub.blocking_stream_unary  s     "6::
 
 !% < @ @!
 !
 &M&!
 
 	
r   c                     | j                             ||f          }| j                            ||f          }t          | j        |||||| j        |||
  
        S r   )r   r   r   r   r   r   	r-   r   r   r   rF   r   r   r{   r|   s	            r   future_stream_unaryz _GenericStub.future_stream_unary  s     "6::
 
 !% < @ @!
 !
 $M&!
 
 	
r   c                     | j                             ||f          }| j                            ||f          }t          | j        |||||| j        |||
  
        S r   )r   r   r   r   r   r   r   s	            r   inline_stream_streamz!_GenericStub.inline_stream_stream  s     "6::
 
 !% < @ @!
 !
 M&!
 
 	
r   c	                     t                      r   r   	r-   r   r   r   r   rl   rF   r   r   s	            r   event_unary_unaryz_GenericStub.event_unary_unary)       "###r   c	                     t                      r   r   r   s	            r   event_unary_streamz_GenericStub.event_unary_stream6  r   r   c                     t                      r   r   r-   r   r   r   rl   rF   r   r   s           r   event_stream_unaryz_GenericStub.event_stream_unaryC       "###r   c                     t                      r   r   r   s           r   event_stream_streamz _GenericStub.event_stream_streamO  r   r   c                     | j                             ||f          }| j                            ||f          }t          | j        ||| j        ||          S r   )r   r   r   r   r   r   r-   r   r   r{   r|   s        r   r}   z_GenericStub.unary_unary[  sw    !6::
 
 !% < @ @!
 !
 (M&!
 
 	
r   c                     | j                             ||f          }| j                            ||f          }t          | j        ||| j        ||          S r   )r   r   r   r   r   r   r   s        r   r   z_GenericStub.unary_streamq  w    !6::
 
 !% < @ @!
 !
 )M&!
 
 	
r   c                     | j                             ||f          }| j                            ||f          }t          | j        ||| j        ||          S r   )r   r   r   r   r   r   r   s        r   r   z_GenericStub.stream_unary  r   r   c                     | j                             ||f          }| j                            ||f          }t          | j        ||| j        ||          S r   )r   r   r   r   r   r   r   s        r   r   z_GenericStub.stream_stream  sw    !6::
 
 !% < @ @!
 !
 *M&!
 
 	
r   c                     | S r   r
   r,   s    r   	__enter__z_GenericStub.__enter__  r[   r   c                     dS NFr
   r-   exc_typeexc_valexc_tbs       r   __exit__z_GenericStub.__exit__      ur   )NNNr   )r/   r0   r1   r;   r   r   r   r   r   r   r   r   r   r   r}   r   r   r   r   r   r
   r   r   r   r   L  s       
D 
D 
D$ "
 "
 "
 "
T  
  
  
  
P  
  
  
  
P "
 "
 "
 "
T  
  
  
  
P  
  
  
  
T $ $ $ $* $ $ $ $( 
$ 
$ 
$ 
$& 
$ 
$ 
$ 
$
 
 
,
 
 
,
 
 
,
 
 
,      r   r   c                   &    e Zd Zd Zd Zd Zd ZdS )_DynamicStubc                 0    || _         || _        || _        d S r   )_generic_stubr   _cardinalities)r-   backing_generic_stubr   cardinalitiess       r   r;   z_DynamicStub.__init__  s    1+r   c                    | j                             |          }|t          j        j        u r | j                            | j        |          S |t          j        j        u r | j        	                    | j        |          S |t          j        j
        u r | j                            | j        |          S |t          j        j        u r | j                            | j        |          S t          d|z            )Nz*_DynamicStub object has no attribute "%s"!)r   r   r   CardinalityUNARY_UNARYr   r}   r   UNARY_STREAMr   STREAM_UNARYr   STREAM_STREAMr   AttributeError)r-   attrmethod_cardinalitys      r   __getattr__z_DynamicStub.__getattr__  s    !044T::!8!DDD%11$+tDDD;#:#GGG%224;EEE;#:#GGG%224;EEE;#:#HHH%33DKFFF <tC  r   c                     | S r   r
   r,   s    r   r   z_DynamicStub.__enter__  r[   r   c                     dS r   r
   r   s       r   r   z_DynamicStub.__exit__  r   r   N)r/   r0   r1   r;   r  r   r   r
   r   r   r   r     sP        , , ,
        r   r   c                 &    t          | |||          S r   )r   )r   hostr   r   r   s        r   generic_stubr    s"     	  r   c                 D    t          t          | |||          ||          S r   )r   r   )r   servicer   r  r   r   r   s          r   dynamic_stubr    s7      "		
 	
 		 	 	r   )8__doc__rH   r   	grpc.betar   r   grpc.framework.commonr   grpc.framework.foundationr   grpc.framework.interfaces.facer   rj   	CANCELLEDr   r   CancellationErrorUNKNOWNREMOTE_FAILURERemoteErrorDEADLINE_EXCEEDEDEXPIREDExpirationErrorUNIMPLEMENTEDr   
LocalErrorr   r   r   r#   r(   GRPCInvocationContextr*   FutureCallr3   r   r   r   r   r   r   UnaryUnaryMultiCallabler   UnaryStreamMultiCallabler   StreamUnaryMultiCallabler   StreamStreamMultiCallabler   GenericStubr   DynamicStubr   r  r  r
   r   r   <module>r)     sT   J I                          - - - - - - , , , , , , / / / / / /
 	O$  	O) 	O%"( 	O!($: 6(7 7 7P P P
 
 
	 	 	    !A   
]$ ]$ ]$ ]$ ]$&- ]$ ]$ ]$@$. $. $.N? ? ?6C C C6$. $. $.N? ? ?6C C C6>$ >$ >$ >$ >$t; >$ >$ >$B($ ($ ($ ($ ($ = ($ ($ ($V?$ ?$ ?$ ?$ ?$ = ?$ ?$ ?$D)$ )$ )$ )$ )$!? )$ )$ )$Xk k k k k4# k k k\    4#   8      r   