File: //usr/lib/python3.6/site-packages/botocore/__pycache__/discovery.cpython-36.pyc
3
�T_+ � @ s� d dl Z d dlZd dlZd dlmZ d dlmZmZmZ d dl m
Z
d dl m Z ej
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 � /usr/lib/python3.6/discovery.pyr s r c @ s 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 s r c @ s e Zd ZdZdZdS )� EndpointDiscoveryRefreshFailedz? Endpoint Discovery failed to the refresh the known endpoints. z<Endpoint Discovery failed to refresh the required endpoints.N)r r
r r r r r r r
r " s r c K s | j }|r|jd�rt� �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 s
|| _ d S )N)�_service_model)�self�
service_modelr r r
�__init__. s z EndpointDiscoveryModel.__init__c C s | j j}t|j�S )N)r � endpoint_discovery_operationr �name)r �discovery_operationr r r
�discovery_operation_name1 s z/EndpointDiscoveryModel.discovery_operation_namec C s& | j j}g }|jr"t|jjj� �}|S )N)r r � input_shape�list�members�keys)r r r$ r r r
�discovery_operation_keys6 s
z/EndpointDiscoveryModel.discovery_operation_keysc C s4 y| j j|�}|jjdd�S tk
r. dS X d S )Nr F)r �operation_modelr r r )r �operation_namer& r r r
�discovery_required_for> s
z-EndpointDiscoveryModel.discovery_required_forc sB | j � |jd�s(|jdd � |jdd � t� fdd�|j� D ��S )N� Identifiers� Operationc 3 s" | ]\}}|� kr||fV qd S )Nr )�.0�k�v)�
input_keysr r
� <genexpr>K s zDEndpointDiscoveryModel.discovery_operation_kwargs.<locals>.<genexpr>)r% r �pop�dict�items)r r r )r. r
�discovery_operation_kwargsE s
z1EndpointDiscoveryModel.discovery_operation_kwargsc C s | j |j|�S )N)� _gather_idsr! )r � operation�paramsr r r
�gather_identifiersM s z)EndpointDiscoveryModel.gather_identifiersNc C sf |d kr i }xT|j j� D ]F\}}|jjd�r:|| ||< q|jdkr||kr| j||| |� qW |S )NZendpointdiscoveryidZ structure)r# r2 Zmetadatar Z type_namer4 )r �shaper6 �ids� member_nameZ member_shaper r r
r4 P s z"EndpointDiscoveryModel._gather_ids)N) r r
r r r r r% r( r3 r7 r4 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 kr i }|| _ i | _|d kr&tj}|| _|| _tj|�| _t|j j
�| _ d S )N) �_cache�_failed_attempts�time�_time�_always_discover�weakref�proxy�_clientr �metar �_model)r Zclient�cache� current_timeZalways_discoverr r r
r ^ s z!EndpointDiscoveryManager.__init__c C s< |d }| j � }x&|D ] }|jd�}||d |d< qW |S )NZ EndpointsZCachePeriodInMinutes�< �
Expiration)r? r )r �response� endpointsrG �endpointZ
cache_timer r r
�_parse_endpointsm s
z)EndpointDiscoveryManager._parse_endpointsc C s" t |t�rtt|j� ��S |S d S )N)�
isinstancer1 �tuple�sortedr2 )r �valuer r r
� _cache_itemu s
z$EndpointDiscoveryManager._cache_itemc s, � j jf |�}t� fdd�t|j� �D ��S )Nc 3 s | ]\}}� j |�V qd S )N)rR )r+ r, r- )r r r
r/ } s z=EndpointDiscoveryManager._create_cache_key.<locals>.<genexpr>)rE r3 rO rP r2 )r r r )r r
�_create_cache_key{ s z*EndpointDiscoveryManager._create_cache_keyc C s | j j||�S )N)rE r7 )r r5 r6 r r r
r7 s z+EndpointDiscoveryManager.gather_identifiersc K s" | j f |�}|| jkr | j|= d S )N)rS r< )r r � cache_keyr r r
�delete_endpoints� s
z)EndpointDiscoveryManager.delete_endpointsc K s8 | j jf |�}| j j}t| j|�}tjd|� |f |�S )Nz%Discovering endpoints with kwargs: %s)rE r3 r �getattrrC �logger�debug)r r r' r r r r
�_describe_endpoints� s
z,EndpointDiscoveryManager._describe_endpointsc s. || j krd S | j� � � fdd�| j | D �S )Nc s g | ]}� |d k r|�qS )rI r )r+ �e)�nowr r
�
<listcomp>� s zCEndpointDiscoveryManager._get_current_endpoints.<locals>.<listcomp>)r< r? )r �keyr )r[ r
�_get_current_endpoints� s
z/EndpointDiscoveryManager._get_current_endpointsc K sn | j f |�}y2| jf |�}| j|�}|| j|< | jj|d � |S ttfk
rh | j� d | j|< d S X d S )NrH ) rS rY rM r< r= r0 r r r? )r r rT rJ rK r r r
�_refresh_current_endpoints� s
z3EndpointDiscoveryManager._refresh_current_endpointsc C s0 || j kr,| j� }|| j | k r$dS | j |= dS )NTF)r= r? )r rT r[ r r r
�_recently_failed� s
z)EndpointDiscoveryManager._recently_failedc C s |d d S )Nr ZAddressr )r rK r r r
�_select_endpoint� s z)EndpointDiscoveryManager._select_endpointc K s� |d }| j j|�}| j r4| r4tjd| � d S | jf |�}| j|�}|rX| j|�S | j|�}|s�| j f |�}|r�| j|�S tjd|� | j
j |d �}|r�| j|�S |r�|r�| j f |�}|r�| j|�S t � �d S )Nr* zAOptional discovery disabled. Skipping discovery for Operation: %sz%Endpoint Discovery has failed for: %s)
rE r( r@ rW rX rS r^ ra r` r_ r< r r )r r r5 Zdiscovery_requiredrT rK Zrecently_failedZ
stale_entriesr r r
�describe_endpoint� s6
z*EndpointDiscoveryManager.describe_endpoint)NNT)r r
r r rM rR rS r7 rU rY r^ r_ r` ra rb r r r r
r; ] s
r; c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �EndpointDiscoveryHandlerc C s
|| _ d S )N)�_manager)r Zmanagerr r r
r � s z!EndpointDiscoveryHandler.__init__c C s: |j d| | j� |jd| | j� |j d| | j� d S )Nzbefore-parameter-build.%szrequest-created.%szneeds-retry.%s)�registerr7 Zregister_first�discover_endpoint�handle_retries)r ZeventsZ
service_idr r r
re � s
z!EndpointDiscoveryHandler.registerc K s0 |j }|d krd S | jj||�}d|i|d< d S )N� identifiers� discovery)r rd r7 )r r6 r �contextr r r9 r r r
r7 � s
z+EndpointDiscoveryHandler.gather_identifiersc K sn |j jdi �jd�}|d kr d S | jj||d�}|d krFtjd� d S |jd�sXd| }tjd|� ||_d S )Nri rh )r* r) z&Failed to discover and inject endpointZhttpzhttps://z!Injecting discovered endpoint: %s)rj r rd rb rW rX �
startswithZurl)r Zrequestr' r r9 rL r r r
rf � s
z*EndpointDiscoveryHandler.discover_endpointc
K s� |d kr d S |\}}|j di �j d�}|j di �j d�}|dkrL|dkrLd S |j di �}|j di �j d �} | d krvd S | jj|j| d
� d S ) NZResponseMetadataZHTTPStatusCode�ErrorZCodei� ZInvalidEndpointExceptionrj ri rh )r* r) r )r rd rU r )
r Z request_dictrJ r5 r �_ZstatusZ
error_coderj r9 r r r
rg � s z'EndpointDiscoveryHandler.handle_retriesN)r r
r r re r7 rf rg r r r r
rc � s
rc )r> ZloggingrA Zbotocorer Zbotocore.exceptionsr r r Zbotocore.modelr Zbotocore.utilsr Z getLoggerr rW r r r r �objectr r; rc r r r r
�<module>
s
0|