ruạṛ
a K��fe � @ s� d Z ddlZddlZddlZddlmZmZmZmZm Z m Z ddl Z ddl m Z ddl mZ ddl mZ ddl mZ dd l mZ G d d � d e�Zeee j ee d �d d�ZG dd� de�dd�e j�Ze je je eeee je e j f d�dd�Z G dd� de j e j e j!�Z"G dd� de j!e j �Z#G dd� de j$�Z%G dd � d e j&�Z'G d d � d e j(�Z)G d d � d e j*�Z+G d!d"� d"e j,�Z-e j,eee e j.e j/e j0e j1f e j,d#�d$d%�Z2dS )&z+Implementation of gRPC Python interceptors.� N)�Any�Callable�Optional�Sequence�Tuple�Union� )�DeserializingFunction)�DoneCallbackType)� MetadataType)�RequestIterableType)�SerializingFunctionc @ sp e Zd ZU eej ed< eej d�dd�Ze e e d�dd�Z e e ej ej d�d d �Ze ej ej d �d d �ZdS )�_ServicePipeline� interceptors�r c C s t |�| _d S �N)�tupler )�selfr � r �u/sparta/input/_build_configuration/image_build+validate/lib/bmcenv/lib64/python3.9/site-packages/grpc/_interceptor.py�__init__! s z_ServicePipeline.__init__)�thunk�index�returnc s � ��fdd�S )Nc s �� �� | �S r �� _intercept_at)�context�r r r r r �<lambda>% � z0_ServicePipeline._continuation.<locals>.<lambda>r )r r r r r r � _continuation$ s z _ServicePipeline._continuation)r r r r c C s@ |t | j�k r4| j| }| �||d �}|�||�S ||�S d S )Nr )�lenr r �intercept_service)r r r r � interceptorr r r r ' s z _ServicePipeline._intercept_at)r r r c C s | � |d|�S )Nr r )r r r r r r �execute1 s z_ServicePipeline.executeN)�__name__� __module__� __qualname__r �grpc�ServerInterceptor�__annotations__r r r �intr �HandlerCallDetails�RpcMethodHandlerr r$ r r r r r s � �r )r r c C s | r t | �S d S r )r r r r r �service_pipeline7 s r. c @ s e Zd ZdS )�_ClientCallDetailsN)r% r& r' r r r r r/ = s r/ )�method�timeout�metadata� credentials�wait_for_ready� compression)� call_details�default_detailsr c C s� z | j }W n ty" |j }Y n0 z | j}W n tyF |j}Y n0 z | j}W n tyj |j}Y n0 z | j}W n ty� |j}Y n0 z | j}W n ty� |j}Y n0 z | j}W n ty� |j}Y n0 ||||||fS r )r0 �AttributeErrorr1 r2 r3 r4 r5 )r6 r7 r0 r1 r2 r3 r4 r5 r r r �_unwrap_client_call_detailsN s> � �� �� �r9 c s\ e Zd ZU eed< ejed< eejd�� fdd� Zee d�dd�Z ee d�d d �Z ee j d�d d �Zee d�d d�Zed�dd�Zed�dd�Zed�dd�Zee d�dd�Zed�dd�Zed�dd�Zd/ee d �d d �Zd0ee ee d �d d!�Zd1ee eej d �d"d#�Zed�d$d%�Z e dd&�d'd(�Z d)d*� Z d+d,� Z d-d.� Z!� Z"S )2�_FailureOutcome� _exception� _traceback)� exception� tracebackc s t t| ��� || _|| _d S r )�superr: r r; r< )r r= r> �� __class__r r r � s z_FailureOutcome.__init__�r c C s d S r r �r r r r �initial_metadata� s z _FailureOutcome.initial_metadatac C s d S r r rC r r r �trailing_metadata� s z!_FailureOutcome.trailing_metadatac C s t jjS r )r( � StatusCode�INTERNALrC r r r �code� s z_FailureOutcome.codec C s dS )Nz+Exception raised while intercepting the RPCr rC r r r �details� s z_FailureOutcome.detailsc C s dS �NFr rC r r r �cancel� s z_FailureOutcome.cancelc C s dS rJ r rC r r r � cancelled� s z_FailureOutcome.cancelledc C s dS rJ r rC r r r � is_active� s z_FailureOutcome.is_activec C s d S r r rC r r r �time_remaining� s z _FailureOutcome.time_remainingc C s dS rJ r rC r r r �running� s z_FailureOutcome.runningc C s dS �NTr rC r r r �done� s z_FailureOutcome.doneN��ignored_timeoutc C s | j �d S r �r; �r rS r r r �result� s z_FailureOutcome.result)rS r c C s | j S r rT rU r r r r= � s z_FailureOutcome.exceptionc C s | j S r )r< rU r r r r> � s z_FailureOutcome.tracebackc C s dS rJ r )r �unused_callbackr r r � add_callback� s z _FailureOutcome.add_callback��fnr c C s || � d S r r �r rZ r r r �add_done_callback� s z!_FailureOutcome.add_done_callbackc C s | S r r rC r r r �__iter__� s z_FailureOutcome.__iter__c C s | j �d S r rT rC r r r �__next__� s z_FailureOutcome.__next__c C s | � � S r )r^ rC r r r �next� s z_FailureOutcome.next)N)N)N)#r% r& r' � Exceptionr* �types� TracebackTyper r r rD rE r( rF rH �strrI �boolrK rL rM �floatrN rO rQ rV r= r> rX r r\ r] r^ r_ � __classcell__r r r@ r r: � s: �� �� r: c @ s. e Zd ZU eed< ejed< eejd�dd�Zee d�dd�Z ee d�d d �Z eej d�d d �Z ee d�d d�Zed�dd�Zee d�dd�Zed�dd�Zed�dd�Zed�dd�Zed�dd�Zed�dd �Zd(ee d �d d �Zd)ee d �d!d"�Zd*ee d �d#d$�Ze d d%�d&d'�Z d S )+� _UnaryOutcome� _response�_call)�response�callc C s || _ || _d S r )rh ri )r rj rk r r r r � s z_UnaryOutcome.__init__rB c C s | j �� S r )ri rD rC r r r rD � s z _UnaryOutcome.initial_metadatac C s | j �� S r )ri rE rC r r r rE � s z _UnaryOutcome.trailing_metadatac C s | j �� S r )ri rH rC r r r rH � s z_UnaryOutcome.codec C s | j �� S r )ri rI rC r r r rI � s z_UnaryOutcome.detailsc C s | j �� S r )ri rM rC r r r rM � s z_UnaryOutcome.is_activec C s | j �� S r )ri rN rC r r r rN � s z _UnaryOutcome.time_remainingc C s | j �� S r )ri rK rC r r r rK � s z_UnaryOutcome.cancelc C s | j �|�S r )ri rX �r �callbackr r r rX � s z_UnaryOutcome.add_callbackc C s dS rJ r rC r r r rL � s z_UnaryOutcome.cancelledc C s dS rJ r rC r r r rO � s z_UnaryOutcome.runningc C s dS rP r rC r r r rQ � s z_UnaryOutcome.doneNrR c C s | j S r )rh rU r r r rV � s z_UnaryOutcome.resultc C s d S r r rU r r r r= � s z_UnaryOutcome.exceptionc C s d S r r rU r r r r> � s z_UnaryOutcome.tracebackrY c C s || � d S r r r[ r r r r\ � s z _UnaryOutcome.add_done_callback)N)N)N) r% r& r' r r* r( �Callr r r rD rE rF rH rc rI rd rM re rN rK rX rL rO rQ rV r= r> r r\ r r r r rg � s$ rg c @ s e Zd ZU eed< eed< ejed< eeejd�dd�Zde e e e e e ej e e e ej e d�d d �Zde e e e e e ej e e e ej ee ejf d�d d �Zde e e e e e ej e e e ej ee ejf d�d d�Zde e e e e e ej e e e ej e d�dd�ZdS )�_UnaryUnaryMultiCallable�_thunk�_method� _interceptor�r r0 r# c C s || _ || _|| _d S r �rp rq rr �r r r0 r# r r r r s z!_UnaryUnaryMultiCallable.__init__N)�requestr1 r2 r3 r4 r5 r c C s | j ||||||d�\}}|S �N�r1 r2 r3 r4 r5 �� _with_call) r rv r1 r2 r3 r4 r5 rj � ignored_callr r r �__call__ s � z!_UnaryUnaryMultiCallable.__call__c s> t �j|||||�� � �fdd�}�j�|� |�}|�� |fS )Nc s� t | � �\}}}}}}z,��|�j||||||d�\}} t|| �W S tjyl } z| W Y d } ~ S d } ~ 0 ty� } z t| t� � d �W Y d } ~ S d } ~ 0 0 d S �Nrx � � r9 rp � with_callrg r( �RpcErrorr` r: �sys�exc_info) � new_detailsrv � new_method� new_timeout� new_metadata�new_credentials�new_wait_for_ready�new_compressionrj rk � rpc_errorr= ��client_call_detailsr r r � continuation1 s, � � z9_UnaryUnaryMultiCallable._with_call.<locals>.continuation)r/ rq rr �intercept_unary_unaryrV ) r rv r1 r2 r3 r4 r5 r� rk r r� r rz s � �z#_UnaryUnaryMultiCallable._with_callc C s | j ||||||d�S rw ry )r rv r1 r2 r3 r4 r5 r r r r� N s �z"_UnaryUnaryMultiCallable.with_callc sp t �j|||||�� � �fdd�}z�j�|� |�W S tyj } z t|t�� d �W Y d }~S d }~0 0 d S )Nc s2 t | � �\}}}}}}��|�j||||||d�S rw �r9 rp �future�r� rv r� r� r� r� r� r� r� r r r� r s � �z5_UnaryUnaryMultiCallable.future.<locals>.continuationr~ )r/ rq rr r� r` r: r� r� � r rv r1 r2 r3 r4 r5 r� r= r r� r r� ` s � �z _UnaryUnaryMultiCallable.future)NNNNN)NNNNN)NNNNN)NNNNN)r% r& r' r r* rc r( �UnaryUnaryClientInterceptorr r r re r �CallCredentialsrd � Compressionr| r rn rz r� r� r r r r ro � s~ � �� � � 2 � � ��ro c @ sn e Zd ZU eed< eed< ejed< eeejd�dd�Zd e e e e e e ej e e e ej d�d d �ZdS ) �_UnaryStreamMultiCallablerp rq rr rs c C s || _ || _|| _d S r rt ru r r r r � s z"_UnaryStreamMultiCallable.__init__N)rv r1 r2 r3 r4 r5 c sp t �j|||||�� � �fdd�}z�j�|� |�W S tyj } z t|t�� d �W Y d }~S d }~0 0 d S )Nc s0 t | � �\}}}}}}��|�||||||d�S rw �r9 rp r� r� r r r� � s ��z8_UnaryStreamMultiCallable.__call__.<locals>.continuationr~ )r/ rq rr �intercept_unary_streamr` r: r� r� r� r r� r r| � s � �z"_UnaryStreamMultiCallable.__call__)NNNNN)r% r& r' r r* rc r( � UnaryStreamClientInterceptorr r r re r r� rd r� r| r r r r r� � s( � ��r� c @ s e Zd ZU eed< eed< ejed< eeejd�dd�Zde e e e e e ej e e e ej ed�d d �Zde e e e e e ej e e e ej eeejf d�d d �Zde e e e e e ej e e e ej eeejf d�d d�Zde e e e e e ej e e e ej ed�dd�ZdS )�_StreamUnaryMultiCallablerp rq rr rs c C s || _ || _|| _d S r rt ru r r r r � s z"_StreamUnaryMultiCallable.__init__N)�request_iteratorr1 r2 r3 r4 r5 r c C s | j ||||||d�\}}|S rw ry ) r r� r1 r2 r3 r4 r5 rj r{ r r r r| � s � z"_StreamUnaryMultiCallable.__call__c s> t �j|||||�� � �fdd�}�j�|� |�}|�� |fS )Nc s� t | � �\}}}}}}z,��|�j||||||d�\}} t|| �W S tjyl } z| W Y d } ~ S d } ~ 0 ty� } z t| t� � d �W Y d } ~ S d } ~ 0 0 d S r} r ) r� r� r� r� r� r� r� r� rj rk r� r= r� r r r� � s, � � z:_StreamUnaryMultiCallable._with_call.<locals>.continuation)r/ rq rr �intercept_stream_unaryrV ) r r� r1 r2 r3 r4 r5 r� rk r r� r rz � s � �z$_StreamUnaryMultiCallable._with_callc C s | j ||||||d�S rw ry )r r� r1 r2 r3 r4 r5 r r r r� s �z#_StreamUnaryMultiCallable.with_callc sp t �j|||||�� � �fdd�}z�j�|� |�W S tyj } z t|t�� d �W Y d }~S d }~0 0 d S )Nc s2 t | � �\}}}}}}��|�j||||||d�S rw r� �r� r� r� r� r� r� r� r� r� r r r� < s � �z6_StreamUnaryMultiCallable.future.<locals>.continuationr~ )r/ rq rr r� r` r: r� r� � r r� r1 r2 r3 r4 r5 r� r= r r� r r� * s � �z _StreamUnaryMultiCallable.future)NNNNN)NNNNN)NNNNN)NNNNN)r% r& r' r r* rc r( � StreamUnaryClientInterceptorr r r re r r� rd r� r r| r rn rz r� r� r r r r r� � s~ � �� � � 2 � � ��r� c @ sn e Zd ZU eed< eed< ejed< eeejd�dd�Zd e e e e e e ej e e e ej d�d d �ZdS ) �_StreamStreamMultiCallablerp rq rr rs c C s || _ || _|| _d S r rt ru r r r r [ s z#_StreamStreamMultiCallable.__init__N)r� r1 r2 r3 r4 r5 c sp t �j|||||�� � �fdd�}z�j�|� |�W S tyj } z t|t�� d �W Y d }~S d }~0 0 d S )Nc s0 t | � �\}}}}}}��|�||||||d�S rw r� r� r� r r r� w s ��z9_StreamStreamMultiCallable.__call__.<locals>.continuationr~ )r/ rq rr �intercept_stream_streamr` r: r� r� r� r r� r r| e s � �z#_StreamStreamMultiCallable.__call__)NNNNN)r% r& r' r r* rc r( � StreamStreamClientInterceptorr r r re r r� rd r� r| r r r r r� V s( � ��r� c @ s6 e Zd ZU ejed< eejejej ej f ed< ejeejejej ej f d�dd�Z d e e e d�dd �Ze d �d d �Zd ee e e e e e ejd�dd�Zd!ee e e e e e ejd�dd�Zd"ee e e e e e ejd�dd�Zd#ee e e e e e ejd�dd�Zdd� Z dd� Z dd � Z d d � Z d S )$�_Channel�_channelrr )�channelr# c C s || _ || _d S r )r� rr )r r� r# r r r r � s z_Channel.__init__F)rm �try_to_connectc C s | j j||d� d S )N)r� )r� � subscribe)r rm r� r r r r� � s z_Channel.subscribe)rm c C s | j �|� d S r )r� � unsubscriberl r r r r� � s z_Channel.unsubscribeN)r0 �request_serializer�response_deserializer�_registered_methodr c s: � ���fdd�}t �jtj�r.t||�j�S ||�S d S )Nc s �j �| ��� �S r )r� � unary_unary��m�r� r� r� r r r r � s �z&_Channel.unary_unary.<locals>.<lambda>)� isinstancerr r( r� ro �r r0 r� r� r� r r r� r r� � s z_Channel.unary_unaryc s: � ���fdd�}t �jtj�r.t||�j�S ||�S d S )Nc s �j �| ��� �S r )r� � unary_streamr� r� r r r � s �z'_Channel.unary_stream.<locals>.<lambda>)r� rr r( r� r� r� r r� r r� � s z_Channel.unary_streamc s: � ���fdd�}t �jtj�r.t||�j�S ||�S d S )Nc s �j �| ��� �S r )r� � stream_unaryr� r� r r r � s �z'_Channel.stream_unary.<locals>.<lambda>)r� rr r( r� r� r� r r� r r� � s z_Channel.stream_unaryc s: � ���fdd�}t �jtj�r.t||�j�S ||�S d S )Nc s �j �| ��� �S r )r� � stream_streamr� r� r r r � s �z(_Channel.stream_stream.<locals>.<lambda>)r� rr r( r� r� r� r r� r r� � s z_Channel.stream_streamc C s | j �� d S r �r� �closerC r r r �_close s z_Channel._closec C s | S r r rC r r r � __enter__ s z_Channel.__enter__c C s | � � dS rJ )r� )r �exc_type�exc_val�exc_tbr r r �__exit__ s z_Channel.__exit__c C s | j �� d S r r� rC r r r r� s z_Channel.close)F)NNF)NNF)NNF)NNF) r% r& r' r( �Channelr* r r� r� r� r� r r r rd r� r� rc r r �UnaryUnaryMultiCallabler� �UnaryStreamMultiCallabler� �StreamUnaryMultiCallabler� �StreamStreamMultiCallabler� r� r� r� r� r r r r r� � s� �� ��� �� �� �� �� �� r� )r� r r c G sX t t|��D ]F}t|tj�sHt|tj�sHt|tj�sHt|tj�sHtd��t | |�} q | S )Nz�interceptor must be grpc.UnaryUnaryClientInterceptor or grpc.UnaryStreamClientInterceptor or grpc.StreamUnaryClientInterceptor or grpc.StreamStreamClientInterceptor or ) �reversed�listr� r( r� r� r� r� � TypeErrorr� )r� r r# r r r �intercept_channel s � � � �� r� )3�__doc__� collectionsr� ra �typingr r r r r r r( �_typingr r r r r �objectr r) r. � namedtuple�ClientCallDetailsr/ rc re r� rd r� r9 r� �Futurern r: rg r� ro r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r �<module> sj � � ��� 3 �F6 ; ; ���� �
cải xoăn