| Dir : /usr/libexec/kcare/python/kcarectl/__pycache__/ |
| Server: Linux server2.noticiasdecolima.com 4.18.0-553.30.1.el8_10.x86_64 #1 SMP Tue Nov 26 02:30:26 EST 2024 x86_64 IP: 107.161.180.42 |
| Dir : //usr/libexec/kcare/python/kcarectl/__pycache__/utils.cpython-36.pyc |
3
�;j[# � @ sd d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
m
Z
ddl m Z ej
d�ZdZed�Zed �Zed�Zd<d
d�Zd=dd�Zd>dd�Zd?dd�Zdd� Zd@dd�Zdedfdd �Zd d � Zd d � Z d!d"� Z d#d$� Z d%d&� Z d'd(� Z ed)d*� �Z!d+d,� Z"dAd-d.�Z#dBd/d0�Z$d1d2� Z%dCd4d5�Z&d6d7� Z'd8d9� Z(de)fdfd:d;�Z*dS )D� )�print_functionN)�datetime� )� constantsz^(\d+[.]\d+[-]\d+)� � � F�w� c
C s� t jj| �}|r(t jj|� r(t j|� yt j| �j}W n tk
rP |}Y nX tj ||t jj
| �d dd��<}t j |j � |� |j
|� |j� t j|j � � |j}W d Q R X t j|t j�} zt j| � W d t j| � X t j|| � d S )N�.F)�mode�dir�prefix�delete)�os�path�dirname�exists�makedirs�stat�st_mode� Exception�tempfile�NamedTemporaryFile�basename�fchmod�fileno�write�flush�fsync�name�open�O_RDONLY�close�rename)
�fname�content�
ensure_dirr Z create_modeZdnamer �fZ tmp_fnameZ 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* �nstr> s
r4 �latin1c C s t | �tkr| j|�} | S )N)r- �utyper1 )r2 r3 r) r) r* �bstrH s
r7 c C s t | �tkr| j|�} | S )N)r- r/ r0 )r2 r3 r) r) r* �ustrO s
r8 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* �innerZ s zcached.<locals>.inner)� functools�wrapsr@ �clear�orig)rA rB r) )r@ rA r* �cachedV s
rG 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)
�ranger �time�sleep�min�random�uniformr �RETRY_MAX_DELAY�setattr)r= r> Zldelay�_�exZfinal_ex)�backoff� check_retry�count�delayrA �stater) r* rB v s
z'retry.<locals>.decorator.<locals>.innerr) )rA rB )rS rT rU rV rW )rA r* � decoratoru s zretry.<locals>.decorator)r � RETRY_DELAY� RETRY_COUNT�
RETRY_BACKOFF)rT rU rV rS rX r) )rS rT rU rV rW r* �retryj s r\ 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�appendr �st_mtime�sort�isfile�islink�remove�shutil�rmtree)
Z directory� exclude_path�keep_n�patternr2 r; �item� full_pathrQ �entryr) r) r* �clean_directory� s
rp c C sN t tjjtjd�dd� t tjjtjd�dd� tjjtj�rJtjtj� d S )N�modulesr )rk �patches) rp r r ra r � PATCH_CACHEr �CACHE_KEY_DUMP_PATH�unlinkr) r) r) r* �clear_all_cache� s rv 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! rh � copyfileobjr r r )�response�dstZ
parent_dirr( r) r) r* � save_to_file� s
r{ c C s t j| �}|r|jd�p| S )Nr )�
VERSION_RE�match�group)�versionr} r) r) r* �strip_version_timestamp� s
r� c C s0 | j d�\}}}|s$| j d�\}}}tj|d�S )N�T� z%Y-%m-%d)� partitionr �strptime)Zstr_rawZstr_date�seprQ r) r) r* �parse_response_date� s r� 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 r� ra r` �get_patch_server)�partsr� 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 rt 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� r? r) r) r* � read_file� s r� c C s t | d|�}|S )N�rb)r� )r% r� r? r) r) r* �
read_file_bin� s r� c C sB i }| j � }x0|D ](}|r|jd�\}}}|r|j� ||< qW |S )N�:)�
splitlinesr� r� )r2 r? Z
data_lines�line�key� delimiter�valuer) r) r* � data_as_dict� s
r� �cvec C s0 t � }x$| D ] }|j|�}|r |j|� q W |S )a
Extract unique CVEs from a list of patches.
Args:
patches: List of patch dictionaries
cve_field: Field name to extract CVE from (default 'cve' for userspace, 'kpatch-cve' for kernel)
Returns:
Set of unique CVE strings
)�set�get�add)rr � cve_fieldZ unique_cves�patchr� r) r) r* �extract_unique_cves� s
r� c C s t ttj� ��S )N)�str�intrJ 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 j� ���� �fdd��}|S )Nc sn y
�| |�S �k
rh } zB�rZdd� | D �dd� |j � D � }�dj�jdj|�|�� � S d }~X 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 j ||��qS )z{0}={1})�format)r� �k�vr) r) r* r� & s z{0}({1}) failed: {2}z, )r; r� �__name__ra )r= r> �e�arg_list)�default_return�errorsrA �loggerr) r* rB s
z.catch_errors.<locals>.decorator.<locals>.inner)rC rD )rA rB )r� r� r� )rA r* rX s z catch_errors.<locals>.decoratorr) )r� r� r� rX r) )r� r� r� r* � catch_errors s r� )Fr r
)r, )r5 )r5 )NNN)N)N)r� )+�
__future__r r_ rC r rM �rerh r rJ r r r �compiler| Z
CACHE_ENTRIESr- r. r/ r6 r+ r4 r7 r8 rG r\ rp rv r{ r� r� r� r� r� r� r� r� r� r� r� r� r r� r) r) r) r* �<module> sJ
$