| Current File : //usr/libexec/kcare/python/kcarectl/__pycache__/utils.cpython-36.pyc |
3
��why � @ s, d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlmZ ddlm Z ej
d�Z dZ e
d�Ze
d�Ze
d�Zd3d d �Zd4dd�Zd5dd�Zd6dd�Zdd� Zd7dd�Zde dfdd�Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Z d%d&� Z ed'd(� �Z d)d*� Z d8d+d,�Z d9d-d.�Z!d/d0� Z"d1d2� Z#dS ):� N)�datetime� )� constantsz^(\d+[.]\d+[-]\d+)� � � F�wc C s� | d }t jj|�}|r0t jj|� r0t j|� t||��&}|j|� |j� t j|j � � W d Q R X t j|t j
�}zt j|� W d t j |� X t j || � 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&| j|�S | j|�S d S )N)�type�ntype�btype�decode�encode)�data�encodingr r r �nstr7 s
r& �latin1c C s t | �tkr| j|�} | S )N)r �utyper# )r$ r% r r r �bstrA s
r) c C s t | �tkr| j|�} | S )N)r r! r" )r$ r% r r r �ustrH s
r* c s4 i � t j��� �fdd��}� |_� j|_�|_|S )Nc sH | t t|j� ��f}y� | S tk
r0 Y nX �| |� }� |< |S )N)�tuple�sorted�items�KeyError)�args�kwargsZ cache_key�result)�cache�fnr r �innerS s zcached.<locals>.inner)� functools�wrapsr2 �clearZorig)r3 r4 r )r2 r3 r �cachedO s
r8 c sF �d krt j��d kr t j�� d kr*t j� i �� ����fdd�}|S )Nc s ����� �fdd�}|S )Nc s� �}xlt ��D ]`}y
�| |�S tk
rJ } z�|��s:� W Y d d }~X nX tj|� t|tjd� � tj�}qW y
�| |�S tk
r� } zt |d�� � W Y d d }~X nX d S )Nr �attempts)
�range� Exception�time�sleep�min�random�uniformr �RETRY_MAX_DELAY�setattr)r/ r0 Zldelay�_�exZfinal_ex)�backoff� check_retry�count�delayr3 �stater r r4 o s
z'retry.<locals>.decorator.<locals>.innerr )r3 r4 )rE rF rG rH rI )r3 r � decoratorn s zretry.<locals>.decorator)r � RETRY_DELAY� RETRY_COUNT�
RETRY_BACKOFF)rF rG rH rE rJ r )rE rF rG rH rI r �retryc s rN c
C s� t jj| �sd S g }t j| �}|d k r2tj||�}xD|D ]<}t jj| |�}||krh|jt j|�j |f� |j
dd� q8W xF||d � D ]6\}} t jj | �s�t jj | �r�t j
| � q�tj| � q�W 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$ r- �item� full_pathrC �entryr r r �clean_directory� s
rc c C sN t tjjtjd�dd� t tjjtjd�dd� tjjtj�rJtjtj� d S )N�modulesr )r^ �patches) rc r r
rS r � PATCH_CACHEr �CACHE_KEY_DUMP_PATH�unlinkr r r r �clear_all_cache� s ri c
C s^ t jj|�}t jj|�s"t j|� t|d��(}tj| |� |j� t j |j
� � W d Q R X d S )N�wb) r r
r r r
r r[ � copyfileobjr r r )�response�dstZ
parent_dirr r r r � save_to_file� s
rn c C s t j| �}|r|jd�p| S )Nr )�
VERSION_RE�match�group)�versionrp r r r �strip_version_timestamp� s
rs c C s0 | j d�\}}}|s$| j d�\}}}tj|d�S )N�T� z%Y-%m-%d)� partitionr �strptime)Zstr_rawZstr_date�seprC r r r �parse_response_date� s ry c G s0 ddl m} djdd� td |j� f| �D ��S )Nr )� ipv6_support�/c s s | ]}|j d �V qdS )r{ N)�strip)�.0�itr r r � <genexpr>� s z'get_patch_server_url.<locals>.<genexpr>)r rz rS rR �get_patch_server)�partsrz r r r �get_patch_server_url� s r� c C s0 t jj| �sd S t| ��}|j� j� S Q R X d S )N)r r
r r �readr| )�filenamer r r r � try_to_read� s
r� c C s
t tj�S )N)r� r rg r r r r �
get_cache_key� s r� c C s. t jj| �s|S t| |��
}|j� S Q R X d S )N)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� r1 r r r � read_file� s r� c C s t | d|�}|S )N�rb)r� )r r� r1 r r r �
read_file_bin� s r� c C sB i }| j � }x0|D ](}|r|jd�\}}}|r|j� ||< qW |S )N�:)�
splitlinesrv r| )r$ r1 Z
data_lines�line�key� delimiter�valuer r r � data_as_dict� s
r� c C s t ttj� ��S )N)�str�intr< r r r r �
timestamp_str� s r� )Fr )r )r' )r' )NNN)N)N)$r rQ r5 �rer[ r? r< r r r �compilero Z
CACHE_ENTRIESr r r! r( r r& r) r* r8 rN rc ri rn rs ry r� r� r� r� r� r� r� r� r r r r �<module> s@
$