File: //var/opt/nydus/ops/primordial/__pycache__/utils.cpython-39.pyc
a
�,�h' � @ sN d dl Z d dlmZmZ d dlmZ d dlmZmZm Z m
Z
m Z m Z m
Z
dZe �e�ZG dd� de�Ze
e e
e d�d d
�Zeeef e e eeef d �d d
�Zeeef e e eeef d �dd�Zd ed�eeed�dd�Zd dd�eeed�dd�Zefedddfe e e df eee
e
egef e
e
egef d�dd �Z dS ) � N)�partial�update_wrapper)�sleep)�Any�Dict�Iterable�Optional�Tuple�Type�Callable� c @ s e Zd ZdZdS )�RetriesExhaustedErrorz9A special type which signals the failure of a retry loop.N)�__name__�
__module__� __qualname__�__doc__� r r �A/opt/nydus/tmp/pip-target-wkfpz8uv/lib/python/primordial/utils.pyr
s r
)�hostname�returnc C sF | du r dS | � � �d�r&| dd� } | �d�D ]}|r0| S q0| S )z�Abbreviate hostname for use on a Windows machine.
:param hostname: the hostname
:returns: the first non-empty domain in the hostname, excluding "www."
Nzwww.� �.)�lower�
startswith�split)r �domainr r r � abbreviate_hostname_for_windows s
r )�dict_�keysr c s � fdd�| � � D �S )a Filter a dictionary to contain only a certain set of keys.
:param dict_: The original dictionary to be filtered.
:param keys: A list, or other iterable, containing the desired dictionary keys.
:returns: A dictionary containing only the desired keys.
c s i | ]\}}|� v r||�qS r r ��.0�k�v�r r r �
<dictcomp>+ � zsubdict.<locals>.<dictcomp>��items�r r r r# r �subdict$ s r) c s � fdd�| � � D �S )z�Filter a dictionary to omit a set of keys.
:param dict_: The original dictionary to be filtered.
:param keys: An iterable containing the keys to omit.
:returns: A dictionary with the desired keys omitted.
c s i | ]\}}|� vr||�qS r r r r# r r r$ 5 r% z subdict_omit.<locals>.<dictcomp>r& r( r r# r � subdict_omit. s r* )� curr_attempt� max_attempts)r+ r, r c O s | |k S �Nr )r+ r, �_args�_kwargsr r r �
_should_retry8 s r0 � )r+ �
sleep_secs)r+ r2 r c O s |S r- r )r+ r2 r. r/ r r r � _retry_after= s r3 .)�
on_ex_classesr, r2 � should_retry� retry_afterc s4 �pt t|d���p t t|d��� ��fdd�}|S )a� Decorator that adds retry on error functionality to a function.
Currently the retry strategy is 'linear' on errors. i.e. this function waits a set period
of time before retrying the failed function again.
:param on_ex_classes: A tuple of exceptions to retry on.
By default, its all exceptions that derive from the 'Exception' class.
:param max_attempts: Limit to how many times we'll retry
:param sleep_secs: How long to wait between retries.
:param should_retry: A predicate which when called will return a boolean saying whether
the call should be retried or not. This parameter overrides the max_attempts
parameter and gives more control to dynamically choose on if we need to
continue retrying a call.
:param retry_after: A callable that returns how long to wait between retries.
This parameter overrides the sleep_secs parameter and gives more control
to dynamically choose the wait time.
:returns: This returns a decorator function that actually provides the retry functionality.
)r, )r2 c s. t dd� �D ��� � ���fdd�}t|��S )Nc s s | ]}t |t�r|V qd S r- )�
issubclass� Exception)r Zex_clsr r r � <genexpr>[ r% z.retry_this.<locals>.wrapper.<locals>.<genexpr>c
s� d}z�| i |��W S � y� } z�|d7 }t �dt|��j� �| d|i|��shd��j�}t|�|��| d|i|��}t �d|�jt|�� t|� W Y d }~qd }~0 0 qd S )Nr r1 z)Exception (%s) occured while executing %sr+ z Max attempts exhausted for {}z:Sleeping %s secs before retrying %s, due to exception (%s))�LOGGER�error�strr �formatr
�debugr )Zpargs�kwargsr+ �e�msgZs_secs)�
ex_classes�fr6 r5 r r �new_func] s
�z-retry_this.<locals>.wrapper.<locals>.new_func)�tupler )rC rD �r4 r6 r5 )rB rC r �wrapperZ s zretry_this.<locals>.wrapper)r r0 r3 )r4 r, r2 r5 r6 rG r rF r �
retry_thisA s rH ) �logging� functoolsr r �timer �typingr r r r r r
r Z MAX_RETRIES� getLoggerr r: r8 r
r< r r) r* �int�boolr0 r3 rH r r r r �<module> s4 $
&
&
�� ��