File: //lib/python3/dist-packages/botocore/__pycache__/discovery.cpython-310.pyc
o
.&�a+ � @ s� d dl Z d dlZd dlZd dlmZ d dlmZmZmZ d dl m
Z
d dl m Z e�
e�ZG dd� de�ZG dd � d e�ZG d
d � d e�Zd d
� ZG dd� de�ZG dd� de�ZG dd� de�ZdS )� N)�
xform_name)�
BotoCoreError�HTTPClientError�ConnectionError)�OperationNotFoundError)�CachedPropertyc @ s e Zd ZdS )�EndpointDiscoveryExceptionN)�__name__�
__module__� __qualname__� r r �4/usr/lib/python3/dist-packages/botocore/discovery.pyr s r c @ � e Zd ZdZdZdS )�EndpointDiscoveryRequiredzD Endpoint Discovery is disabled but is required for this operation. zAEndpoint Discovery is not enabled but this operation requires it.N�r r
r �__doc__�fmtr r r r
r � r c @ r )� EndpointDiscoveryRefreshFailedz? Endpoint Discovery failed to the refresh the known endpoints. z<Endpoint Discovery failed to refresh the required endpoints.Nr r r r r
r " r r c K s" | j }|r
|�d�rt� �d S d S )N�required)�endpoint_discovery�getr )�model�kwargsr r r r
�,block_endpoint_discovery_required_operations' s �r c @ sN e Zd Zdd� Zedd� �Zedd� �Zdd� Zd d
� Zd d � Z ddd�Z
d
S )�EndpointDiscoveryModelc C �
|| _ d S �N)�_service_model)�self�
service_modelr r r
�__init__. �
z EndpointDiscoveryModel.__init__c C s | j j}t|j�S r )r � endpoint_discovery_operationr �name)r �discovery_operationr r r
�discovery_operation_name1 s
z/EndpointDiscoveryModel.discovery_operation_namec C s& | j j}g }|jrt|jj�� �}|S r )r r# � input_shape�list�members�keys)r r% r* r r r
�discovery_operation_keys6 s
z/EndpointDiscoveryModel.discovery_operation_keysc C s2 z| j �|�}|j�dd�W S ty Y dS w )Nr F)r �operation_modelr r r )r �operation_namer, r r r
�discovery_required_for> s �z-EndpointDiscoveryModel.discovery_required_forc sB | j � |�d�s|�dd � |�dd � t� fdd�|�� D ��S )N� Identifiers� Operationc 3 s$ � | ]
\}}|� v r||fV qd S r r ��.0�k�v��
input_keysr r
� <genexpr>K s �" zDEndpointDiscoveryModel.discovery_operation_kwargs.<locals>.<genexpr>)r+ r �pop�dict�items�r r r r5 r
�discovery_operation_kwargsE s
z1EndpointDiscoveryModel.discovery_operation_kwargsc C s | � |j|�S r )� _gather_idsr' �r � operation�paramsr r r
�gather_identifiersM � z)EndpointDiscoveryModel.gather_identifiersNc C sb |d u ri }|j �� D ]#\}}|j�d�r || ||< q |jdkr.||v r.| �||| |� q |S )N�endpointdiscoveryid� structure)r) r: �metadatar � type_namer= )r �shaper@ �ids� member_name� member_shaper r r
r= P s �z"EndpointDiscoveryModel._gather_idsr ) r r
r r! r r& r+ r. r< rA r= r r r r
r - s
r c @ sn e Zd Zddd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� ZdS ) �EndpointDiscoveryManagerNTc C sP |d u ri }|| _ i | _|d u rtj}|| _|| _t�|�| _t|j j
�| _ d S r ) �_cache�_failed_attempts�time�_time�_always_discover�weakref�proxy�_clientr �metar �_model)r �client�cache� current_time�always_discoverr r r
r! ^ s z!EndpointDiscoveryManager.__init__c C s8 |d }| � � }|D ]}|�d�}||d |d< q
|S )N� Endpoints�CachePeriodInMinutes�< �
Expiration)rO r )r �response� endpointsrX �endpoint�
cache_timer r r
�_parse_endpointsm s
z)EndpointDiscoveryManager._parse_endpointsc C s t |t�r
tt|�� ��S |S r )�
isinstancer9 �tuple�sortedr: )r �valuer r r
� _cache_itemu s
z$EndpointDiscoveryManager._cache_itemc s0 � j jdi |��}t� fdd�t|�� �D ��S )Nc 3 s � | ]
\}}� � |�V qd S r )rg r1 �r r r
r7 } s � z=EndpointDiscoveryManager._create_cache_key.<locals>.<genexpr>r )rU r< rd re r: r; r rh r
�_create_cache_key{ s z*EndpointDiscoveryManager._create_cache_keyc C s | j �||�S r )rU rA r> r r r
rA rB z+EndpointDiscoveryManager.gather_identifiersc K s* | j di |��}|| jv r| j|= d S d S )Nr )ri rL )r r � cache_keyr r r
�delete_endpoints� s
�z)EndpointDiscoveryManager.delete_endpointsc K s@ | j jdi |��}| j j}t| j|�}t�d|� |di |��S )Nz%Discovering endpoints with kwargs: %sr )rU r< r& �getattrrS �logger�debug)r r r- r% r r r
�_describe_endpoints� s
z,EndpointDiscoveryManager._describe_endpointsc s. || j vrd S | �� � � fdd�| j | D �S )Nc s g | ]
}� |d k r|�qS )r] r )r2 �e��nowr r
�
<listcomp>� s zCEndpointDiscoveryManager._get_current_endpoints.<locals>.<listcomp>)rL rO )r �keyr rq r
�_get_current_endpoints� s
z/EndpointDiscoveryManager._get_current_endpointsc K st | j di |��}z | jdi |��}| �|�}|| j|< | j�|d � |W S ttfy9 | �� d | j|< Y d S w )Nr\ r ) ri ro rb rL rM r8 r r rO )r r rj r^ r_ r r r
�_refresh_current_endpoints� s
�z3EndpointDiscoveryManager._refresh_current_endpointsc C s0 || j v r| �� }|| j | k rdS | j |= dS )NTF)rM rO )r rj rr r r r
�_recently_failed� s
z)EndpointDiscoveryManager._recently_failedc C s |d d S )Nr �Addressr )r r_ r r r
�_select_endpoint� s z)EndpointDiscoveryManager._select_endpointc K s� |d }| j �|�}| js|st�d| � d S | jdi |��}| �|�}|r,| �|�S | �|�}|sB| j di |��}|rB| �|�S t�d|� | j
� |d �}|rV| �|�S |rl|ri| j di |��}|ri| �|�S t � �d S )Nr0 zAOptional discovery disabled. Skipping discovery for Operation: %sz%Endpoint Discovery has failed for: %sr )
rU r. rP rm rn ri ru ry rw rv rL r r )r r r? �discovery_requiredrj r_ �recently_failed�
stale_entriesr r r
�describe_endpoint� s:
��
z*EndpointDiscoveryManager.describe_endpoint)NNT)r r
r r! rb rg ri rA rk ro ru rv rw ry r} r r r r
rK ] s
rK c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �EndpointDiscoveryHandlerc C r r )�_manager)r �managerr r r
r! � r" z!EndpointDiscoveryHandler.__init__c C s: |� d| | j� |�d| | j� |� d| | j� d S )Nzbefore-parameter-build.%szrequest-created.%szneeds-retry.%s)�registerrA �register_first�discover_endpoint�handle_retries)r �events�
service_idr r r
r� � s
�
�z!EndpointDiscoveryHandler.registerc K s0 |j }|d u r d S | j�||�}d|i|d< d S )N� identifiers� discovery)r r rA )r r@ r �contextr r rH r r r
rA � s
z+EndpointDiscoveryHandler.gather_identifiersc K sn |j �di ��d�}|d u rd S | jj||d�}|d u r#t�d� d S |�d�s,d| }t�d|� ||_d S )Nr� r� �r0 r/ z&Failed to discover and inject endpoint�httpzhttps://z!Injecting discovered endpoint: %s)r� r r r} rm rn �
startswith�url)r �requestr- r rH r` r r r
r� � s �
z*EndpointDiscoveryHandler.discover_endpointc
K s� |d u rd S |\}}|� di �� d�}|� di �� d�}|dkr&|dkr&d S |� di �}|� di �� d �} | d u r;d S | jj|j| d
� d S ) N�ResponseMetadata�HTTPStatusCode�Error�Codei� �InvalidEndpointExceptionr� r� r� r� r )r r rk r$ )
r � request_dictr^ r? r �_�status�
error_coder� rH r r r
r� � s �z'EndpointDiscoveryHandler.handle_retriesN)r r
r r! r� rA r� r� r r r r
r~ � s r~ )rN �loggingrQ �botocorer �botocore.exceptionsr r r �botocore.modelr �botocore.utilsr � getLoggerr rm r r r r �objectr rK r~ r r r r
�<module> s
0|