File: //opt/alt/python37/lib64/python3.7/site-packages/grpc/__pycache__/_interceptor.cpython-37.pyc
B
��fja � @ 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 �D/opt/alt/python37/lib64/python3.7/site-packages/grpc/_interceptor.py�__init__! s z_ServicePipeline.__init__)�thunk�index�returnc s � ��fdd�S )Nc s �� �� | �S )N)�
_intercept_at)�context)r r r r r �<lambda>% � z0_ServicePipeline._continuation.<locals>.<lambda>r )r 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 Zintercept_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 ZHandlerCallDetailsZRpcMethodHandlerr r r r r r r s
r )r r c C s | r t | �S d S )N)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� y
| j }W n tk
r$ |j }Y nX y
| j}W n tk
rJ |j}Y nX y
| j}W n tk
rp |j}Y nX y
| j}W n tk
r� |j}Y nX y
| j}W n tk
r� |j}Y nX y
| j}W n tk
r� |j}Y nX ||||||fS )N)r) �AttributeErrorr* r+ r, r- r. )r/ r0 r) r* r+ r, r- r. r r r �_unwrap_client_call_detailsN s2
r2 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 )N)�superr3 r r4 r5 )r r6 r7 )� __class__r r r � s z_FailureOutcome.__init__)r c C s d S )Nr )r r r r �initial_metadata� s z _FailureOutcome.initial_metadatac C s d S )Nr )r r r r �trailing_metadata� s z!_FailureOutcome.trailing_metadatac C s t jjS )N)r# �
StatusCodeZINTERNAL)r r r r �code� s z_FailureOutcome.codec C s dS )Nz+Exception raised while intercepting the RPCr )r r r r �details� s z_FailureOutcome.detailsc C s dS )NFr )r r r r �cancel� s z_FailureOutcome.cancelc C s dS )NFr )r r r r � cancelled� s z_FailureOutcome.cancelledc C s dS )NFr )r r r r � is_active� s z_FailureOutcome.is_activec C s d S )Nr )r r r r �time_remaining� s z _FailureOutcome.time_remainingc C s dS )NFr )r r r r �running� s z_FailureOutcome.runningc C s dS )NTr )r r r r �done� s z_FailureOutcome.doneN)�ignored_timeoutc C s
| j �d S )N)r4 )r rE r r r �result� s z_FailureOutcome.result)rE r c C s | j S )N)r4 )r rE r r r r6 � s z_FailureOutcome.exceptionc C s | j S )N)r5 )r rE r r r r7 � s z_FailureOutcome.tracebackc C s dS )NFr )r Zunused_callbackr r r � add_callback� s z _FailureOutcome.add_callback)�fnr c C s || � d S )Nr )r rH r r r �add_done_callback� s z!_FailureOutcome.add_done_callbackc C s | S )Nr )r r r r �__iter__� s z_FailureOutcome.__iter__c C s
| j �d S )N)r4 )r r r r �__next__� s z_FailureOutcome.__next__c C s | � � S )N)rK )r r r r �next� s z_FailureOutcome.next)N)N)N)#r r! r" � Exceptionr% �types�
TracebackTyper r r r: r; r# r< r= �strr> �boolr? r@ rA �floatrB rC rD rF r6 r7 rG r
rI rJ rK rL �
__classcell__r r )r9 r r3 � s.
r3 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 )N)rU rV )r rW rX r r r r � s z_UnaryOutcome.__init__)r c C s
| j �� S )N)rV r: )r r r r r: � s z _UnaryOutcome.initial_metadatac C s
| j �� S )N)rV r; )r r r r r; � s z _UnaryOutcome.trailing_metadatac C s
| j �� S )N)rV r= )r r r r r= � s z_UnaryOutcome.codec C s
| j �� S )N)rV r> )r r r r r> � s z_UnaryOutcome.detailsc C s
| j �� S )N)rV rA )r r r r rA � s z_UnaryOutcome.is_activec C s
| j �� S )N)rV rB )r r r r rB � s z _UnaryOutcome.time_remainingc C s
| j �� S )N)rV r? )r r r r r? � s z_UnaryOutcome.cancelc C s | j �|�S )N)rV rG )r �callbackr r r rG � s z_UnaryOutcome.add_callbackc C s dS )NFr )r r r r r@ � s z_UnaryOutcome.cancelledc C s dS )NFr )r r r r rC � s z_UnaryOutcome.runningc C s dS )NTr )r r r r rD � s z_UnaryOutcome.doneN)rE c C s | j S )N)rU )r rE r r r rF � s z_UnaryOutcome.resultc C s d S )Nr )r rE r r r r6 � s z_UnaryOutcome.exceptionc C s d S )Nr )r rE r r r r7 � s z_UnaryOutcome.traceback)rH r c C s || � d S )Nr )r rH r r r rI � s z _UnaryOutcome.add_done_callback)N)N)N) r r! r" r r% r# �Callr r r r: r; r< r= rP r> rQ rA rR rB r? rG r@ rC rD rF r6 r7 r
rI r r r r rT � s$
rT 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 r) r c C s || _ || _|| _d S )N)r\ r] r^ )r r r) r r r r r s z!_UnaryUnaryMultiCallable.__init__N)�requestr* r+ r, r- r. r c C s | j ||||||d�\}}|S )N)r* r+ r, r- r. )�
_with_call) r r_ r* r+ r, r- r. rW � ignored_callr r r �__call__ s z!_UnaryUnaryMultiCallable.__call__c s> t �j|||||�� � �fdd�}�j�|� |�}|�� |fS )Nc
s� t | � �\}}}}}}y*��|�j||||||d�\}} t|| �S tjk
rd }
z|
S d }
~
X Y n0 tk
r� } zt| t� � d �S d } ~ X Y nX d S )N)r* r+ r, r- r. � )
r2 r\ � with_callrT r# �RpcErrorrM r3 �sys�exc_info) � new_detailsr_ �
new_method� new_timeout� new_metadata�new_credentials�new_wait_for_ready�new_compressionrW rX � rpc_errorr6 )�client_call_detailsr r r � continuation1 s
z9_UnaryUnaryMultiCallable._with_call.<locals>.continuation)r( r] r^ �intercept_unary_unaryrF ) r r_ r* r+ r, r- r. rq rX r )rp r r r` s
z#_UnaryUnaryMultiCallable._with_callc C s | j ||||||d�S )N)r* r+ r, r- r. )r` )r r_ r* r+ r, r- r. r r r rd N s z"_UnaryUnaryMultiCallable.with_callc
sh t �j|||||�� � �fdd�}y�j�|� |�S tk
rb } zt|t�� d �S d }~X Y nX d S )Nc s2 t | � �\}}}}}}��|�j||||||d�S )N)r* r+ r, r- r. )r2 r\ �future)rh r_ ri rj rk rl rm rn )rp r r r rq r s
z5_UnaryUnaryMultiCallable.future.<locals>.continuationrc )r( r] r^ rr rM r3 rf rg ) r r_ r* r+ r, r- r. rq r6 r )rp r r rs ` s
z _UnaryUnaryMultiCallable.future)NNNNN)NNNNN)NNNNN)NNNNN)r r! r" r r% rP r# �UnaryUnaryClientInterceptorr r r rR r �CallCredentialsrQ � Compressionrb r rZ r` rd rs r r r r r[ � s<
& &* &
&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 ) �_UnaryStreamMultiCallabler\ r] r^ )r r) r c C s || _ || _|| _d S )N)r\ r] r^ )r r r) r r r r r � s z"_UnaryStreamMultiCallable.__init__N)r_ r* r+ r, r- r. c
sh t �j|||||�� � �fdd�}y�j�|� |�S tk
rb } zt|t�� d �S d }~X Y nX d S )Nc s0 t | � �\}}}}}}��|�||||||d�S )N)r* r+ r, r- r. )r2 r\ )rh r_ ri rj rk rl rm rn )rp r r r rq � s z8_UnaryStreamMultiCallable.__call__.<locals>.continuationrc )r( r] r^ Zintercept_unary_streamrM r3 rf rg ) r r_ r* r+ r, r- r. rq r6 r )rp r r rb � s
z"_UnaryStreamMultiCallable.__call__)NNNNN)r r! r" r r% rP r# � UnaryStreamClientInterceptorr r r rR r ru rQ rv rb r r r r rw � s
rw 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 )�_StreamUnaryMultiCallabler\ r] r^ )r r) r c C s || _ || _|| _d S )N)r\ r] r^ )r r r) r r r r r � s z"_StreamUnaryMultiCallable.__init__N)�request_iteratorr* r+ r, r- r. r c C s | j ||||||d�\}}|S )N)r* r+ r, r- r. )r` ) r rz r* r+ r, r- r. rW ra r r r rb � s z"_StreamUnaryMultiCallable.__call__c s> t �j|||||�� � �fdd�}�j�|� |�}|�� |fS )Nc
s� t | � �\}}}}}}y*��|�j||||||d�\}} t|| �S tjk
rd }
z|
S d }
~
X Y n0 tk
r� } zt| t� � d �S d } ~ X Y nX d S )N)r* r+ r, r- r. rc )
r2 r\ rd rT r# re rM r3 rf rg ) rh rz ri rj rk rl rm rn rW rX ro r6 )rp r r r rq � s
z:_StreamUnaryMultiCallable._with_call.<locals>.continuation)r( r] r^ �intercept_stream_unaryrF ) r rz r* r+ r, r- r. rq rX r )rp r r r` � s
z$_StreamUnaryMultiCallable._with_callc C s | j ||||||d�S )N)r* r+ r, r- r. )r` )r rz r* r+ r, r- r. r r r rd s z#_StreamUnaryMultiCallable.with_callc
sh t �j|||||�� � �fdd�}y�j�|� |�S tk
rb } zt|t�� d �S d }~X Y nX d S )Nc s2 t | � �\}}}}}}��|�j||||||d�S )N)r* r+ r, r- r. )r2 r\ rs )rh rz ri rj rk rl rm rn )rp r r r rq < s
z6_StreamUnaryMultiCallable.future.<locals>.continuationrc )r( r] r^ r{ rM r3 rf rg ) r rz r* r+ r, r- r. rq r6 r )rp r r rs * s
z _StreamUnaryMultiCallable.future)NNNNN)NNNNN)NNNNN)NNNNN)r r! r" r r% rP r# � StreamUnaryClientInterceptorr r r rR r ru rQ rv r rb r rZ r` rd rs r r r r ry � s<
& &* &
&ry 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 ) �_StreamStreamMultiCallabler\ r] r^ )r r) r c C s || _ || _|| _d S )N)r\ r] r^ )r r r) r r r r r [ s z#_StreamStreamMultiCallable.__init__N)rz r* r+ r, r- r. c
sh t �j|||||�� � �fdd�}y�j�|� |�S tk
rb } zt|t�� d �S d }~X Y nX d S )Nc s0 t | � �\}}}}}}��|�||||||d�S )N)r* r+ r, r- r. )r2 r\ )rh rz ri rj rk rl rm rn )rp r r r rq w s z9_StreamStreamMultiCallable.__call__.<locals>.continuationrc )r( r] r^ Zintercept_stream_streamrM r3 rf rg ) r rz r* r+ r, r- r. rq r6 r )rp r r rb e s
z#_StreamStreamMultiCallable.__call__)NNNNN)r r! r" r r% rP r# � StreamStreamClientInterceptorr r r rR r ru rQ rv rb r r r r r} V s
r} c @ s 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 ejd�dd�Zd!ee
e e
e ejd�dd�Zd"ee
e e
e ejd�dd�Zd#ee
e e
e ejd�dd�Zdd� Z dd� Z dd � Z d d � Z d
S )$�_Channel�_channelr^ )�channelr c C s || _ || _d S )N)r� r^ )r r� r r r r r � s
z_Channel.__init__F)rY �try_to_connectc C s | j j||d� d S )N)r� )r� � subscribe)r rY r� r r r r� � s z_Channel.subscribe)rY c C s | j �|� d S )N)r� � unsubscribe)r rY r r r r� � s z_Channel.unsubscribeN)r) �request_serializer�response_deserializerr c s8 � ��fdd�}t �jtj�r,t||�j�S ||�S d S )Nc s �j �| � ��S )N)r� � unary_unary)�m)r� r� r r r r � s z&_Channel.unary_unary.<locals>.<lambda>)�
isinstancer^ r# rt r[ )r r) r� r� r r )r� r� r r r� � s z_Channel.unary_unaryc s8 � ��fdd�}t �jtj�r,t||�j�S ||�S d S )Nc s �j �| � ��S )N)r� � unary_stream)r� )r� r� r r r r � s z'_Channel.unary_stream.<locals>.<lambda>)r� r^ r# rx rw )r r) r� r� r r )r� r� r r r� � s z_Channel.unary_streamc s8 � ��fdd�}t �jtj�r,t||�j�S ||�S d S )Nc s �j �| � ��S )N)r� � stream_unary)r� )r� r� r r r r � s z'_Channel.stream_unary.<locals>.<lambda>)r� r^ r# r| ry )r r) r� r� r r )r� r� r r r� � s z_Channel.stream_unaryc s8 � ��fdd�}t �jtj�r,t||�j�S ||�S d S )Nc s �j �| � ��S )N)r� �
stream_stream)r� )r� r� r r r r � s z(_Channel.stream_stream.<locals>.<lambda>)r� r^ r# r~ r} )r r) r� r� r r )r� r� r r r� � s z_Channel.stream_streamc C s | j �� d S )N)r� �close)r r r r �_close� s z_Channel._closec C s | S )Nr )r r r r � __enter__� s z_Channel.__enter__c C s | � � dS )NF)r� )r �exc_type�exc_val�exc_tbr r r �__exit__� s z_Channel.__exit__c C s | j �� d S )N)r� r� )r r r r r� � s z_Channel.close)F)NN)NN)NN)NN) r r! r" r# �Channelr% r rt rx r~ r| r r r rQ r� r� rP r
r �UnaryUnaryMultiCallabler� �UnaryStreamMultiCallabler� �StreamUnaryMultiCallabler� �StreamStreamMultiCallabler� r� r� r� r� r r r r r � s<
r )r� r r c G s\ xVt t|��D ]F}t|tj�sJt|tj�sJt|tj�sJt|tj�sJtd��t | |�} qW | S )Nz�interceptor must be grpc.UnaryUnaryClientInterceptor or grpc.UnaryStreamClientInterceptor or grpc.StreamUnaryClientInterceptor or grpc.StreamStreamClientInterceptor or )
�reversed�listr� r# rt rx r| r~ � TypeErrorr )r� r r r r r �intercept_channel� s
r� )3�__doc__� collectionsrf rN �typingr r r r r r r# �_typingr r
r r r
�objectr r$ r' �
namedtupleZClientCallDetailsr( rP rR ru rQ rv r2 re ZFuturerZ r3 rT r� r[ r� rw r� ry r� r} r� r rt rx r~ r| r� r r r r �<module> sR
/
E6 ; ;e