File: //usr/lib/python3.6/site-packages/botocore/__pycache__/retryhandler.cpython-36.pyc
3
�T_�5 � @ s$ d dl Z d dlZd dlZd dlmZ d dlmZmZmZm Z m
Z
ej e �Z
de e
eegiZdd� Zdd� Zd%d d
�Zd&d d �Zd'd
d�Zdd� Zdd� Zdd� ZG dd� de�ZG dd� de�ZG dd� de�ZG dd � d e�ZG d d � d e�Z G d d � d e�Z G d!d"� d"e�Z G d#d$� d$e�Z dS )(� N)�crc32)�
ChecksumError�EndpointConnectionError�ReadTimeoutError�ConnectionError�ConnectionClosedErrorZGENERAL_CONNECTION_ERRORc C s: | dkrt j � } n| dkr&td| ��| ||d }|S )a1 Calculate time to sleep based on exponential function.
The format is::
base * growth_factor ^ (attempts - 1)
If ``base`` is set to 'rand' then a random number between
0 and 1 will be used as the base.
Base must be greater than 0, otherwise a ValueError will be
raised.
Zrandr z0The 'base' param must be greater than 0, got: %s� )�random�
ValueError)�base�
growth_factor�attemptsZ
time_to_sleep� r �"/usr/lib/python3.6/retryhandler.py�delay_exponential' s
r c C s t jt| |d�S )z�Create an exponential delay function based on the attempts.
This is used so that you only have to pass it the attempts
parameter to calculate the delay.
)r r )� functools�partialr )r r r r r �!create_exponential_delay_function= s r c C s$ t | |d�}t| |d�}t||d�S )N)�operation_name)�checker�action)� create_checker_from_retry_config� create_retry_action_from_config� RetryHandler)�configr r r r r r �create_retry_handlerH s
r c C s0 | d d }|d dkr,t |d |d d�S d S )N� __default__Zdelay�typeZ exponentialr r )r r )r )r r Z delay_configr r r r P s
r c C s g }d }g }d| krn| d j dg �}| d d }x<|D ]4}|| }|jt|�� t|�}|d k r6|j|� q6W |d k r�| j |�d k r�| | d } x<| D ]4}|jt| | �� t| | �}|d k r�|j|� q�W t|�dkr�t|d |d�S t|�}
t|
|t|�d�S d S )Nr �policies� max_attemptsr r )r )r �retryable_exceptions) �get�append�_create_single_checker� _extract_retryable_exception�extend�len�MaxAttemptsDecorator� MultiChecker�tuple) r r �checkersr r r �keyZcurrent_configZretry_exceptionZoperation_policiesZ
multi_checkerr r r r \ s4
r c C s2 d| d kr t | d d �S d| d kr.t� S d S )N�response� applies_when�
socket_errors)� _create_single_response_checker�ExceptionRaiser)r r r r r# { s
r# c C s^ d| kr t | d | d d�}n<d| kr6t| d d�}n$d| krNt| d d�}n tdt ��|S )NZservice_error_codeZhttp_status_code)� status_code�
error_code)r1 � crc32body)�headerzUnknown retry policy: %s)�ServiceErrorCodeChecker�HTTPStatusCodeChecker� CRC32Checkerr
r )r, r r r r r/ � s r/ c C sR | d }d|j di �kr tgS d|krNg }x |d D ]}|jt| � q4W |S d S )Nr- r3 r, r. )r! r r% �
EXCEPTION_MAP)r r- �
exceptions�namer r r r$ � s r$ c @ s e Zd ZdZdd� Zdd� ZdS )r a� Retry handler.
The retry handler takes two params, ``checker`` object
and an ``action`` object.
The ``checker`` object must be a callable object and based on a response
and an attempt number, determines whether or not sufficient criteria for
a retry has been met. If this is the case then the ``action`` object
(which also is a callable) determines what needs to happen in the event
of a retry.
c C s || _ || _d S )N)�_checker�_action)�selfr r r r r �__init__� s zRetryHandler.__init__c K s8 | j |||�r*| j|d�}tjd|� |S tjd� dS )z�Handler for a retry.
Intended to be hooked up to an event handler (hence the **kwargs),
this will process retries appropriately.
)r
zRetry needed, action of: %szNo retry needed.N)r; r< �logger�debug)r= r
r, �caught_exception�kwargs�resultr r r �__call__� s
zRetryHandler.__call__N)�__name__�
__module__� __qualname__�__doc__r> rD r r r r r � s r c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � BaseCheckerz�Base class for retry checkers.
Each class is responsible for checking a single criteria that determines
whether or not a retry should not happen.
c C s4 |dk r| j ||�S |dk r(| j||�S td��dS )a� Determine if retry criteria matches.
Note that either ``response`` is not None and ``caught_exception`` is
None or ``response`` is None and ``caught_exception`` is not None.
:type attempt_number: int
:param attempt_number: The total number of times we've attempted
to send the request.
:param response: The HTTP response (if one was received).
:type caught_exception: Exception
:param caught_exception: Any exception that was caught while trying to
send the HTTP response.
:return: True, if the retry criteria matches (and therefore a retry
should occur. False if the criteria does not match.
Nz,Both response and caught_exception are None.)�_check_response�_check_caught_exceptionr
)r= �attempt_numberr, rA r r r rD � s zBaseChecker.__call__c C s d S )Nr )r= rL r, r r r rJ � s zBaseChecker._check_responsec C s d S )Nr )r= rL rA r r r rK � s z#BaseChecker._check_caught_exceptionN)rE rF rG rH rD rJ rK r r r r rI � s rI c @ s* e Zd ZdZd dd�Zdd� Zdd� ZdS )
r' a� Allow retries up to a maximum number of attempts.
This will pass through calls to the decorated retry checker, provided
that the number of attempts does not exceed max_attempts. It will
also catch any retryable_exceptions passed in. Once max_attempts has
been exceeded, then False will be returned or the retryable_exceptions
that was previously being caught will be raised.
Nc C s || _ || _|| _d S )N)r; �
_max_attempts�_retryable_exceptions)r= r r r r r r r> � s z MaxAttemptsDecorator.__init__c C s^ | j |||�}|rV|| jkrP|d k r@d|d kr@d|d d d< tjd|� dS |S ndS d S )NZResponseMetadatar TZMaxAttemptsReachedz0Reached the maximum number of retry attempts: %sF)�
_should_retryrM r? r@ )r= rL r, rA Z should_retryr r r rD � s
z MaxAttemptsDecorator.__call__c C sf | j rT|| jk rTy| j|||�S | j k
rP } ztjd|dd� dS d }~X qbX n| j|||�S d S )Nz,retry needed, retryable exception caught: %sT)�exc_info)rN rM r; r? r@ )r= rL r, rA �er r r rO s
z"MaxAttemptsDecorator._should_retry)N)rE rF rG rH r> rD rO r r r r r' � s
r' c @ s e Zd Zdd� Zdd� ZdS )r6 c C s
|| _ d S )N)� _status_code)r= r1 r r r r> s z HTTPStatusCodeChecker.__init__c C s* |d j | jkr"tjd| j� dS dS d S )Nr z5retry needed: retryable HTTP status code received: %sTF)r1 rR r? r@ )r= rL r, r r r rJ s z%HTTPStatusCodeChecker._check_responseN)rE rF rG r> rJ r r r r r6 s r6 c @ s e Zd Zdd� Zdd� ZdS )r5 c C s || _ || _d S )N)rR � _error_code)r= r1 r2 r r r r> ' s z ServiceErrorCodeChecker.__init__c C sJ |d j | jkrF|d jdi �jd�}|| jkrFtjd| j| j� dS dS )Nr r �ErrorZCodez>retry needed: matching HTTP status and error code seen: %s, %sTF)r1 rR r! rS r? r@ )r= rL r, Zactual_error_coder r r rJ + s
z'ServiceErrorCodeChecker._check_responseN)rE rF rG r> rJ r r r r r5 & s r5 c @ s e Zd Zdd� Zdd� ZdS )r( c C s
|| _ d S )N)� _checkers)r= r* r r r r> 7 s zMultiChecker.__init__c C s( x"| j D ]}||||�}|r|S qW dS )NF)rU )r= rL r, rA r Zchecker_responser r r rD : s zMultiChecker.__call__N)rE rF rG r> rD r r r r r( 6 s r( c @ s e Zd Zdd� Zdd� ZdS )r7 c C s
|| _ d S )N)� _header_name)r= r4 r r r r> D s zCRC32Checker.__init__c C st |d }|j j| j�}|d kr.tjd| j� nBt|d j�d@ }|t|�ksptjdt|�|� tdt|�|d��d S )Nr z?crc32 check skipped, the %s header is not in the http response.l �� z>retry needed: crc32 check failed, expected != actual: %s != %sr )Z
checksum_typeZexpected_checksumZactual_checksum) Zheadersr! rV r? r@ r Zcontent�intr )r= rL r, Z
http_responseZ expected_crcZ actual_crc32r r r rJ H s
z CRC32Checker._check_responseN)rE rF rG r> rJ r r r r r7 C s r7 c @ s e Zd ZdZdd� ZdS )r0 z`Raise any caught exceptions.
This class will raise any non None ``caught_exception``.
c C s |�d S )Nr )r= rL rA r r r rK _ s z'ExceptionRaiser._check_caught_exceptionN)rE rF rG rH rK r r r r r0 Y s r0 )N)N)N) r r ZloggingZbinasciir Zbotocore.exceptionsr r r r r Z getLoggerrE r? r8 r r r r r r# r/ r$ �objectr rI r' r6 r5 r( r7 r0 r r r r �<module> s0
,.