File: //usr/libexec/kcare/python/kcarectl/__pycache__/utils.cpython-38.pyc
U
n�h � @ sR d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl m Z ddl
m Z e�d�ZdZed�Zed �Zed�Zd8d d
�Zd9dd�Zd:dd�Z d;dd�Z dd� Z d<dd�Z dedfdd�Z d d � Z!d d � Z"d d!� Z#d"d#� Z$d$d%� Z%d&d'� Z&e d(d)� �Z'd*d+� Z(d=d,d-�Z)d>d.d/�Z*d0d1� Z+d2d3� Z,d4d5� Z-de.fdfd6d7�Z/dS )?� )�print_functionN)�datetime� )� constantsz^(\d+[.]\d+[-]\d+)� � � F�wc C s� | d }t j�|�}|r.t j�|�s.t �|� t||��&}|�|� |�� t �|� � � W 5 Q R X t �|t j
�}zt �|� W 5 t � |� X t � || � d S )Nz.tmp)
�os�path�dirname�exists�makedirs�open�write�flush�fsync�fileno�O_RDONLY�close�rename)�fname�content�
ensure_dir�modeZ tmp_fnameZdname�fZ folder_fd� r �+/usr/libexec/kcare/python/kcarectl/utils.py� atomic_write! s
r �utf-8c C s4 t | �tkr| S t | �tkr&| �|�S | �|�S d S �N)�type�ntype�btype�decode�encode��data�encodingr r r �nstr8 s
r) �latin1c C s t | �tkr| �|�} | S r )r! �utyper% r& r r r �bstrB s
r, c C s t | �tkr| �|�} | S r )r! r# r$ r& r r r �ustrI s
r- c s4 i � t ���� �fdd��}� |_� j|_�|_|S )Nc sJ | t t|�� ��f}z
� | W S tk
r2 Y nX �| |� }� |< |S r )�tuple�sorted�items�KeyError)�args�kwargsZ cache_key�result��cache�fnr r �innerT s
zcached.<locals>.inner)� functools�wrapsr6 �clear�orig�r7 r8 r r5 r �cachedP s r> c sF �d krt j��d kr t j�� d kr*t j� i �� ����fdd�}|S )Nc s ����� �fdd�}|S )Nc s� �}t ��D ]f}z�| |�W S tk
rN } z�|��s>� W 5 d }~X Y nX t�|� t|t�d� � tj�}q z �| |�W S tk
r� } zt |d�� � W 5 d }~X Y nX d S )Nr �attempts)
�range� Exception�time�sleep�min�random�uniformr �RETRY_MAX_DELAY�setattr)r2 r3 Zldelay�_�exZfinal_ex)�backoff� check_retry�count�delayr7 �stater r r8 p s
z'retry.<locals>.decorator.<locals>.innerr r= �rK rL rM rN rO �r7 r � decoratoro s zretry.<locals>.decorator)r � RETRY_DELAY� RETRY_COUNT�
RETRY_BACKOFF)rL rM rN rK rR r rP r �retryd s rV c
C s� t j�| �sd S g }t �| �}|d k r2t�||�}|D ]<}t j�| |�}||krf|�t �|�j |f� |j
dd� q6||d � D ]6\}} t j� | �s�t j� | �r�t �
| � q�t�| � q�d S )NT)�reverse)r
r r
�listdir�fnmatch�filter�join�append�stat�st_mtime�sort�isfile�islink�remove�shutil�rmtree)
Z directory� exclude_path�keep_n�patternr' r0 �item� full_pathrI �entryr r r �clean_directory� s
rk c C sN t tj�tjd�dd� t tj�tjd�dd� tj�tj�rJt�tj� d S )N�modulesr )rf �patches) rk r
r r[ r � PATCH_CACHEr
�CACHE_KEY_DUMP_PATH�unlinkr r r r �clear_all_cache� s rq c C s^ t j�|�}t j�|�s"t �|� t|d��(}t�| |� |�� t � |�
� � W 5 Q R X d S )N�wb) r
r r r
r r rc � copyfileobjr r r )�response�dstZ
parent_dirr r r r � save_to_file� s
rv c C s t �| �}|r|�d�p| S )Nr )�
VERSION_RE�match�group)�versionrx r r r �strip_version_timestamp� s
r{ c C s0 | � d�\}}}|s$| � d�\}}}t�|d�S )N�T� z%Y-%m-%d)� partitionr �strptime)Zstr_rawZstr_date�seprI r r r �parse_response_date� s r� c G s0 ddl m} d�dd� td |�� f| �D ��S )Nr )� ipv6_support�/c s s | ]}|� d �V qdS )r� N)�strip)�.0�itr r r � <genexpr>� s z'get_patch_server_url.<locals>.<genexpr>)r r� r[ rZ �get_patch_server)�partsr� r r r �get_patch_server_url� s r� c
C s< t j�| �sd S t| ��}|�� �� W 5 Q R � S Q R X d S r )r
r r
r �readr� )�filenamer r r r � try_to_read� s
r� c C s
t tj�S r )r� r ro r r r r �
get_cache_key� s r� c
C s: t j�| �s|S t| |��}|�� W 5 Q R � S Q R X d S r )r
r r
r r� )r r �defaultr r r r �
_read_file� s r� c C s t | d|�}|S )N�r�r� �r r� r4 r r r � read_file� s r� c C s t | d|�}|S )N�rbr� r� r r r �
read_file_bin� s r� c C s> i }| � � }|D ](}|r|�d�\}}}|r|�� ||< q|S )N�:)�
splitlinesr~ r� )r' r4 Z
data_lines�line�key� delimiter�valuer r r � data_as_dict� s r� c C s t tt�� ��S r )�str�intrB r r r r �
timestamp_str� s r� c G s t | � dS )za workaround to fix T201N)�print)�valuesr r r �
print_wrapper� s r� c s � ��fdd�}|S )Nc s t �� ���� �fdd��}|S )Nc
sz z �| |�W S �k
rt } zJ�r\dd� | D �dd� |� � D � }�d��jd�|�|�� � W Y �S d }~X Y nX d S )Nc S s g | ] }t |��qS r )r� )r� �ar r r �
<listcomp> s zBcatch_errors.<locals>.decorator.<locals>.inner.<locals>.<listcomp>c S s g | ]\}}d � ||��qS )z{0}={1})�format)r� �k�vr r r r� s z{0}({1}) failed: {2}z, )r0 r� �__name__r[ )r2 r3 �e�arg_list)�default_return�errorsr7 �loggerr r r8 s z.catch_errors.<locals>.decorator.<locals>.inner)r9 r: r= �r� r� r� rQ r rR s
z catch_errors.<locals>.decoratorr )r� r� r� rR r r� r � catch_errors s r� )Fr )r )r* )r* )NNN)N)N)0�
__future__r rY r9 r
rE �rerc rB r r r �Any�Callable�Optional�TupleZTypeVar�Unionr| �compilerw Z
CACHE_ENTRIESr! r" r# r+ r r) r, r- r> rV rk rq rv r{ r� r� r� r� r� r� r� r� r� r� rA r� r r r r �<module> sH
$