File: //usr/libexec/oracle-cloud-agent/plugins/osms/grpc/_plugin_wrapping.pyc
a
K��f � @ s� d dl Z d dlZd dlZd dlmZmZmZ d dlZd dlmZ d dl m
Z
d dl m Z e�
e�ZG dd� de �dd �ej�ZG d
d � d e�ZG d d
� d
ej�ZG dd� de�Zejee ejd�dd�ZdS )� N)�Callable�Optional�Type)�_common)�cygrpc)� MetadataTypec @ s e Zd ZdS )�_AuthMetadataContextN)�__name__�
__module__� __qualname__� r r �y/sparta/input/_build_configuration/image_build+validate/lib/bmcenv/lib64/python3.9/site-packages/grpc/_plugin_wrapping.pyr s
r �AuthMetadataContext)� service_url� method_namec @ s e Zd Zdd� ZdS )�_CallbackStatec C s t �� | _d| _d | _d S )NF)� threading�Lock�lock�called� exception)�selfr r r
�__init__* s
z_CallbackState.__init__N)r r
r r r r r r
r ) s r c @ sF e Zd ZU eed< eed< eed�dd�Zeee e
d�dd�Z d S )
�_AuthMetadataPluginCallback�_state� _callback)�state�callbackc C s || _ || _d S �N)r r )r r r r r r
r 4 s z$_AuthMetadataPluginCallback.__init__)�metadata�errorc C s� | j j�J | j jd u r2| j jr(td��qDd| j _ntd�| j j���W d � n1 sX0 Y |d u r~| �|tjj d � n | �d tjj
t � t
|��� d S )Nz2AuthMetadataPluginCallback invoked more than once!Tz1AuthMetadataPluginCallback raised exception "{}"!)r r r r � RuntimeError�formatr r �
StatusCode�ok�internalr �encode�str)r r r r r r
�__call__8 s"
�
��"�z$_AuthMetadataPluginCallback.__call__N) r r
r r �__annotations__r r r r r �
BaseExceptionr( r r r r
r 0 s
�r c @ s: e Zd ZU ejed< ejd�dd�Zeeed�dd�Z dS ) �_Plugin�_metadata_plugin)�metadata_pluginc C s: || _ d | _zdd l}|�� | _W n ty4 Y n0 d S )Nr )r, � _stored_ctx� contextvars� copy_context� ImportError)r r- r/ r r r
r T s z_Plugin.__init__)r r r c C s� t t�|�t�|��}t� }z| �|t||�� W n� ty� } z�t�d| j� |j �6 ||_|j
r~W d � W Y d }~d S W d � n1 s�0 Y |d t j j
t�t|��� W Y d }~n
d }~0 0 d S )Nz1AuthMetadataPluginCallback "%s" raised exception!)r r �decoder r, r � Exception�_LOGGERr r r r r# r% r& r' )r r r r �context�callback_stater r r r
r( c s( �
��8�z_Plugin.__call__N)
r r
r �grpc�AuthMetadataPluginr) r r' r r( r r r r
r+ Q s
r+ )r- �name�returnc C sP |d u r0z
| j }W q4 ty, | jj }Y q40 n|}t�t�t| �t� |���S r )
r �AttributeError� __class__r7 �CallCredentialsr � MetadataPluginCallCredentialsr+ r r& )r- r9 �effective_namer r r
� metadata_plugin_call_credentialsz s
��r@ ) � collections�loggingr �typingr r r r7 r � grpc._cythonr � grpc._typingr � getLoggerr r4 �
namedtupler r �objectr �AuthMetadataPluginCallbackr r+ r8 r' r= r@ r r r r
�<module> s*
��
!*
�