File: //lib/python3/dist-packages/botocore/__pycache__/retryhandler.cpython-310.pyc
o
.&�a�5 � @ s$ d dl Z d dlZd dlZd dlmZ d dlmZmZmZm Z m
Z
e� 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�ConnectionClosedError�GENERAL_CONNECTION_ERRORc C s: | dkr t � � } n
| dkrtd| ��| ||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.
�randr z0The 'base' param must be greater than 0, got: %s� )�random�
ValueError)�base�
growth_factor�attempts�
time_to_sleep� r �7/usr/lib/python3/dist-packages/botocore/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 �!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 dkrt |d |d d�S d S )N� __default__�delay�type� exponentialr
r r )r )r r � delay_configr r r r P s ��r c C s� g }d }g }d| v r5| d � dg �}| d d }|D ]}|| }|�t|�� t|�}|d ur4|�|� q|d urc| � |�d urc| | d } | D ]}|�t| | �� t| | �}|d urb|�|� qHt|�dkrqt|d |d�S t|�}
t|
|t|�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% �key�current_config�retry_exception�operation_policies�
multi_checkerr r r r \ s<
� �
� �r c C s2 d| d v rt | d d �S d| d v rt� 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| v rt | d | d d�}|S d| v r t| d d�}|S d| v r*t| d d�}|S td��)N�service_error_code�http_status_code)� status_code�
error_code)r> � crc32body)�headerzUnknown retry policy)�ServiceErrorCodeChecker�HTTPStatusCodeChecker� CRC32Checkerr )r7 r r r r r: � s � ����r: c C sN | d }d|� di �v rtgS d|v r%g }|d D ] }|�t| � q|S d S )Nr8 r@ r7 r9 )r( r r, �
EXCEPTION_MAP)r r8 �
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 � || _ || _d S �N)�_checker�_action)�selfr r r r r �__init__� �
zRetryHandler.__init__c K s8 | � |||�r| j|d�}t�d|� |S t�d� 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)rJ rK �logger�debug)rL r r7 �caught_exception�kwargs�resultr r r �__call__� s
zRetryHandler.__call__N)�__name__�
__module__� __qualname__�__doc__rM rT 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 s0 |dur
| � ||�S |dur| �||�S td��)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 )rL �attempt_numberr7 rQ r r r rT � s �zBaseChecker.__call__c C � d S rI r �rL r\ r7 r r r rZ � � zBaseChecker._check_responsec C r] rI r �rL r\ rQ r r r r[ � r_ z#BaseChecker._check_caught_exceptionN)rU rV rW rX rT rZ r[ r r r r rY � s
rY 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 rI )rJ �
_max_attempts�_retryable_exceptions)rL r r& r' r r r rM � s
z MaxAttemptsDecorator.__init__c C sX | � |||�}|r*|| jkr(|d ur d|d v r d|d d d< t�d|� dS |S dS )N�ResponseMetadatar
T�MaxAttemptsReachedz0Reached the maximum number of retry attempts: %sF)�
_should_retryra rO rP )rL r\ r7 rQ � should_retryr r r rT � s �
�z MaxAttemptsDecorator.__call__c
C sh | j r-|| jk r-z| �|||�W S | j y, } ztjd|dd� W Y d }~dS d }~ww | �|||�S )Nz,retry needed, retryable exception caught: %sT)�exc_info)rb ra rJ rO rP )rL r\ r7 rQ �er r r re s
���z"MaxAttemptsDecorator._should_retryrI )rU rV rW rX rM rT re r r r r r. � s
r. c @ � e Zd Zdd� Zdd� ZdS )rC c C �
|| _ d S rI )� _status_code)rL r> r r r rM �
z HTTPStatusCodeChecker.__init__c C s& |d j | jkrt�d| j� dS dS )Nr z5retry needed: retryable HTTP status code received: %sTF)r> rk rO rP r^ r r r rZ s �z%HTTPStatusCodeChecker._check_responseN�rU rV rW rM rZ r r r r rC � rC c @ ri )rB c C rH rI )rk � _error_code)rL r> r? r r r rM ' rN z ServiceErrorCodeChecker.__init__c C sJ |d j | jkr#|d �di ��d�}|| jkr#t�d| j| j� dS dS )Nr r
�Error�Codez>retry needed: matching HTTP status and error code seen: %s, %sTF)r> rk r( ro rO rP )rL r\ r7 �actual_error_coder r r rZ + s
�z'ServiceErrorCodeChecker._check_responseNrm r r r r rB & � rB c @ ri )r/ c C rj rI �� _checkers)rL r1 r r r rM 7 rl zMultiChecker.__init__c C s( | j D ]}||||�}|r| S qdS )NFrt )rL r\ r7 rQ r �checker_responser r r rT : s
��zMultiChecker.__call__N)rU rV rW rM rT r r r r r/ 6 rn r/ c @ ri )rD c C rj rI )� _header_name)rL rA r r r rM D s
zCRC32Checker.__init__c C sv |d }|j �| j�}|d u rt�d| j� d S t|d j�d@ }|t|�ks9t�dt|�|� 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 )�
checksum_type�expected_checksum�actual_checksum) �headersr( rw rO rP r �content�intr )rL r\ r7 �
http_response� expected_crc� actual_crc32r r r rZ H s" � ���z CRC32Checker._check_responseNrm r r r r rD C rs rD c @ s e Zd ZdZdd� ZdS )r; z`Raise any caught exceptions.
This class will raise any non None ``caught_exception``.
c C s |�rI r r` r r r r[ _ s z'ExceptionRaiser._check_caught_exceptionN)rU rV rW rX r[ r r r r r; Y s r; rI ) r r �logging�binasciir �botocore.exceptionsr r r r r � getLoggerrU rO rE r r r r r r* r: r+ �objectr rY r. rC rB r/ rD r; r r r r �<module> s6
��
,.