File: //lib/python3.9/site-packages/dnf-plugins/__pycache__/kpatch.cpython-39.pyc
a
��-d~ � @ s� d dl mZ d dl mZ d dlZd dlZd dlZd dlmZm Z d dl
Z d dl Z d dl
Z d dlZ dZdZdZdd � Zd
d � Zd d
� ZG dd� de jj�ZG dd� de j�ZdS )� )�absolute_import)�unicode_literalsN)�_�logger�kpatchZ
autoupdatez kernel-corec C s2 dd� | j D �}|D ]}tj�|�r| S qd S )Nc S s g | ]}d |t f �qS )z
%s/%s.conf)�KPATCH_PLUGIN_NAME)�.0�path� r
�6/usr/lib/python3.9/site-packages/dnf-plugins/kpatch.py�
<listcomp>+ � z(_get_plugin_cfg_file.<locals>.<listcomp>)Zpluginconfpath�osr �isfile)Z base_conf�files�filer
r
r �_get_plugin_cfg_file* s
r c C s: t �d| j��d�}|�dd�}| j�dd�}d�||�S )Nz
(.*)\.el.*� �.r zkpatch-patch-{}-{})�re�match�release�group�replace�version�format)�
kernel_pkgZkernel_releaseZkpp_kernel_releaseZkpp_kernel_versionr
r
r �_kpp_name_from_kernel_pkg2 s r c C sV t |�}| j�� j||jd�}tj�| j�}|j|� � d� | j
j || j j
d� d S )N��name�arch)�pkg)ZselectZoptional)r �sack�query�filterr �dnf�selectorZSelector�set�latestZgoal�install�conf�strict)Zdnf_baser � kpp_pkg_name�kpp_pkgs_queryZkpp_sltrr
r
r �_install_kpp_pkg9 s �r. c sd e Zd ZdZed�Z� fdd�Zedd� �Zdd� Z d d
� Z
d d � Z d
d� Z dd� Z
dd� Z� ZS )� KpatchCmd)r z7Toggles automatic installation of kpatch-patch packagesc s" t t| ��|� t| jj�| _d S )N)�superr/ �__init__r �baser* �cfg_file)�self�cli�� __class__r
r r1 H s zKpatchCmd.__init__c C s | j ddd� d S )N�actionzauto|manual|install|status)�metavar)� add_argument)�parserr
r
r �
set_argparserM s zKpatchCmd.set_argparserc C sD | j j}d|_| jjdv r.d|_d|_d|_nd|_d|_d|_d S )NT)�autor) �statusF)r5 �demandsZ root_user�optsr8 Z resolvingZsack_activationZavailable_repos)r4 r? r
r
r � configureR s zKpatchCmd.configurec C s� g }| j j�� �� jtd�}|D ]�}t|�}| j j�� �� j|d��� }|r�| j j�� j||d j|d j d�}| j j�� j||j
d�� � � |�}n | j j�� j||j
d�� � }|D ]}|�
t|�� q�q |S )N�r r )r r r r )r2 r" r# � installedr$ �KERNEL_PKG_NAMEr �runr r r r( �
difference�append�str) r4 �kpps�installed_kernelsr r, rC Zsub_qr- r! r
r
r �_list_missing_kpp_pkgs` s $$ z KpatchCmd._list_missing_kpp_pkgsc C s2 | j j�� �� jtd�}|D ]}t| j |� q d S )NrB )r2 r" r# rC r$ rD r. )r4 rJ r r
r
r �_install_missing_kpp_pkgst s z#KpatchCmd._install_missing_kpp_pkgsc
C sv | j d u rt�d� d S zt�� }|�| j � |W S typ } z&tj� t
d�� t |����W Y d }~n
d }~0 0 d S )N� Couldn't find configuration file�Parsing file failed: {})
r3 r �warning� configparserZ ConfigParser�read� Exceptionr% �
exceptions�Errorr r rH �r4 r; �er
r
r �
_read_conf{ s
zKpatchCmd._read_confc
C s� | j d u rt�d� d S | �� }|d u r,d S |�d�s@|�d� |�dtt|�� z:t | j d��}|�
|� W d � n1 s�0 Y W n> t y� } z&t j
�td��t|����W Y d }~n
d }~0 0 d S )NrM �main�wz Failed to update conf file: {})r3 r rO rW � has_sectionZ add_sectionr' �KPATCH_UPDATE_OPTrH �open�writerR r% rS rT r r )r4 �valuer* Z
cfg_streamrV r
r
r �_update_plugin_cfg� s
,z KpatchCmd._update_plugin_cfgc C s� | j j}|dkr$| �� | �d� n�|dkr8| �d� n�|dkr�| �� }d}|d urz|�d�rz|�dt�rz|�dt�rzd}t �
t d�� |�� | �
� }|r�t �
t d�� d �|��� n(|d
kr�| �� ntj�t d �� |���d S ) Nr= TZmanualFr> rX zKpatch update setting: {}zAvailable patches: {}z, r) zInvalid argument: {})r@ r8 rL r_ rW rZ �
has_optionr[ �
getbooleanr �infor r rK �joinr% rS rT )r4 r8 r* Z kp_statusrI r
r
r rE � s,
�
�
z
KpatchCmd.run)�__name__�
__module__� __qualname__�aliasesr Zsummaryr1 � staticmethodr< rA rK rL rW r_ rE �
__classcell__r
r
r6 r r/ B s
r/ c s8 e Zd ZeZ� fdd�Zdd� Zdd� Zdd� Z� Z S ) � KpatchPluginc s4 t t| ��||� d| _d| _|d ur0|�t� d S )NF)r0 rj r1 �
_commiting� _autoupdateZregister_commandr/ )r4 r2 r5 r6 r
r r1 � s
zKpatchPlugin.__init__c
C sx | � | jj�}z(|�d�o0|�dt�o0|�dt�| _W n< tyr } z$t �
t d�� t
|��� W Y d }~n
d }~0 0 d S )NrX rN )Z read_configr2 r* rZ r` r[ ra rl rR r rO r r rH rU r
r
r �config� s
�
�zKpatchPlugin.configc C s" d| _ | j�| jjj� d| _ d S )NTF)rk r2 �resolver5 r? Z
allow_erasing)r4 r
r
r �_commit_changes� s z KpatchPlugin._commit_changesc s� | j r | jrd S g }g � | jjD ]H}|jtjjv r |jjt krL|�
|j� q |jj� d�r � �
|jj� q � fdd�|D �}|D ]}t | j|� q�|r�| �
� d S )Nz
kpatch-patch-c s g | ]}t |�� vr|�qS r
)r )r r! �Zexplicit_kpp_installr
r r � s �z)KpatchPlugin.resolved.<locals>.<listcomp>)rl rk r2 Z transactionr8 r% ZFORWARD_ACTIONSr! r rD rG �
startswithr. ro )r4 Z need_kpp_forZtr_itemr r
rp r �resolved� s zKpatchPlugin.resolved)
rd re rf r r r1 rm ro rr ri r
r
r6 r rj � s
rj )Z
__future__r r rP Zos.pathr r Zdnfpluginscorer r Z dnf.callbackr% Zdnf.cliZdnf.exceptionsZdnf.transactionr r[ rD r r r. r5 ZCommandr/ ZPluginrj r
r
r
r �<module> s" u