관리-도구
편집 파일: utils.cpython-39.pyc
a d4hF � @ s8 d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlmZ ddlm Z ddlm Z e�d�ZdZed�Zed �Zed�Zd4d d �Zd5dd�Zd6dd�Zd7dd�Zdd� Z d8dd�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&d9d,d-�Z'd:d.d/�Z(d0d1� Z)d2d3� Z*dS );� N)�datetime� )� constants)�configz^(\d+[.]\d+[-]\d+)� � � F�wc C s� | d }t j�|�}|r.t j�|�s.t �|� t||��0}|�|� |�� t �|� � � W d � n1 sn0 Y t �|t j �}zt �|� W t � |� n t � |� 0 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 | �tu r| S t | �tu r&| �|�S | �|�S d S �N)�type�ntype�btype�decode�encode��data�encodingr r r �nstr8 s r) �latin1c C s t | �tu r| �|�} | S r )r! �utyper% r& r r r �bstrB s r, c C s t | �tu r| �|�} | S r )r! r# r$ r& r r r �ustrI s r- c s4 i � t ���� �fdd��}� |_� j|_�|_|S )Nc sL | t t|�� ��f}z � | W S ty0 Y n0 �| i |�� }� |< |S r )�tuple�sorted�items�KeyError)�args�kwargsZ cache_key�result��cache�fnr r �innerT s zcached.<locals>.inner)� functools�wrapsr6 �clearZorig�r7 r8 r r5 r �cachedP s r= c sF �d u rt j��d u r t j�� d u r*t j� i �� ����fdd�}|S )Nc s ����� �fdd�}|S )Nc s� �}t ��D ]l}z�| i |��W S tyT } z�|��s@� W Y d }~n d }~0 0 t�|� t|t�d� � tj�}q z�| i |��W S ty� } zt |d�� � W Y d }~n d }~0 0 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< �rJ rK rL rM rN )r7 r � decoratoro s zretry.<locals>.decorator)r � RETRY_DELAY� RETRY_COUNT� RETRY_BACKOFF)rK rL rM rJ rP r rO r �retryd s rT c C s� t j�| �sd S g }t �| �}|d ur2t�||�}|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_pathrH �entryr r r �clean_directory� s ri c C sN t tj�tjd�dd� t tj�tjd�dd� tj�tj�rJt�tj� d S )N�modulesr )rd �patches) ri r r rY r � PATCH_CACHEr �CACHE_KEY_DUMP_PATH�unlinkr r r r �clear_all_cache� s ro c C sr t j�|�}t j�|�s"t �|� t|d��2}t�| |� |�� t � |� � � W d � n1 sd0 Y d S )N�wb) r r r r r r ra � copyfileobjr r r )�response�dstZ parent_dirr r r r � save_to_file� s rt c C s t �| �}|r|�d�p| S )Nr )� VERSION_RE�match�group)�versionrv r r r �strip_version_timestamp� s ry c C s0 | � d�\}}}|s$| � d�\}}}t�|d�S )N�T� z%Y-%m-%d)� partitionr �strptime)Zstr_rawZstr_date�seprH r r r �parse_response_date� s r c G s" d� dd� td tjf| �D ��S )N�/c s s | ]}|� d �V qdS )r� N)�strip)�.0�itr r r � <genexpr>� r z'get_patch_server_url.<locals>.<genexpr>)rY rX r � PATCH_SERVER)�partsr r r �get_patch_server_url� s r� c C sH t j�| �sd S t| �� }|�� �� W d � S 1 s:0 Y 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 rm r r r r � get_cache_key� s r� c C sF t j�| �s|S t| |��}|�� W d � S 1 s80 Y 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�intrA r r r r � timestamp_str� s r� )Fr )r )r* )r* )NNN)N)N)+r rW r9 �rera rD rA r r r r �Any�Optional�Union�Callable�TypeVarrz �compileru Z CACHE_ENTRIESr! r" r# r+ r r) r, r- r= rT ri ro rt ry r r� r� r� r� r� r� r� r� r r r r �<module> sD $