File: //lib/python3.9/site-packages/dnf-plugins/__pycache__/kpatch.cpython-39.pyc
a
?�%h�2 � @ s� d Z ddlmZ ddlmZ ddlZddlZddlZddlm Z m
Z
ddl Z ddl Z ddl
Z ddlZ ddlZ ddlZdZdZdZd Zd
d � Zd d
� Zdd� ZG dd� de jj�ZG dd� de j�Z dS )z�
The DNF plugin helps customers to install kpatch-patch packages
when the kernel is upgraded and filter kernel-core packages that
are supported by the kpatch team.
� )�absolute_import)�unicode_literalsN)�_�logger�kpatchZ
autoupdateZ
autofilter� 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>5 � z(_get_plugin_cfg_file.<locals>.<listcomp>)Zpluginconfpath�osr
�isfile)Z base_conf�files�filer r r �_get_plugin_cfg_file4 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_pkg< 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_pkgC s �r/ c sh e Zd ZdZdZed�Z� fdd�Zedd� �Z dd � Z
d
d � Z d d
� Z dd� Z
dd� Zdd� Z� ZS )� KpatchCmdz* Extend DNF with kpatch specific commands )r z7Toggles automatic installation of kpatch-patch packagesc s t � �|� t| jj�| _d S )N)�super�__init__r �baser+ �cfg_file)�self�cli�� __class__r r r2 S s zKpatchCmd.__init__c C s | j ddd� dS )z'
argparse python class
�actionzJauto-update|manual-update|auto-filter|no-filter|install|status|auto|manual)�metavarN)� add_argument)�parserr r r �
set_argparserX s �zKpatchCmd.set_argparserc C sD | j j}d|_| jjdv r.d|_d|_d|_nd|_d|_d|_dS )zr
configure DemandSheet
Collection of demands that different CLI parts have on other parts
T)� auto-updater* �status�autoFN)r6 �demandsZ root_user�optsr9 Z resolvingZsack_activationZavailable_repos)r5 rA r r r � configured 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 )r3 r# r$ � installedr% �KERNEL_PKG_NAMEr �runr r r! r) �
difference�append�str) r5 �kpps�installed_kernelsr r- rE Zsub_qr. r" r r r �_list_missing_kpp_pkgsv s0 � �
� �
z KpatchCmd._list_missing_kpp_pkgsc C s2 | j j�� �� jtd�}|D ]}t| j |� q d S )NrD )r3 r# r$ rE r% rF r/ )r5 rL r r r r �_install_missing_kpp_pkgs� 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: {})
r4 r �warning� configparserZ ConfigParser�read� Exceptionr& �
exceptions�Errorr r rJ �r5 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� |�d|t|�� z>t| j ddd��}|� |� W d � n1 s�0 Y W n> t
y� } z&t j �
td��t|����W Y d }~n
d }~0 0 d S )NrO �main�wzutf-8)�encodingz Failed to update conf file: {})r4 r rQ rY � has_sectionZ add_sectionr( rJ �open�writerT r& rU rV r r )r5 �option�valuer+ Z
cfg_streamrX r r r �_update_plugin_cfg� s
,z KpatchCmd._update_plugin_cfgc C s� | j j}|dv r<| �� | �td� t�td��|�� �n�|dv rh| �td� t�td��|�� �nZ|dkr�| �t d� t�td��|�� �n.|dkr�| �t d� t�td��|�� �n|d k�r�| �
� }d
}|d u�r |� d ��r |� d t��r |�
d t��r d
}t�td��|�� d}|d u�rZ|� d ��rZ|� d t ��rZ|�
d t ��rZd}t�td��|�� | �� }|�r�t�td��d�|��� n*|dk�r�| �� ntj�td��|���d S )z:
Decision tree, execution based on config
)r> r@ TzKpatch update setting: {})�
manual-updateZmanualFz auto-filterzKpatch filter setting: {}z no-filterr? rc NrZ r> zAvailable patches: {}z, r* zInvalid argument: {})rB r9 rN rb �KPATCH_UPDATE_OPTr �infor r �KPATCH_FILTER_OPTrY r] �
has_option�
getbooleanrM �joinr& rU rV )r5 r9 r+ Z kp_statusrK r r r rG � sL
�
�
�
�
z
KpatchCmd.run)�__name__�
__module__� __qualname__�__doc__�aliasesr Zsummaryr2 � staticmethodr= rC rM rN rY rb rG �
__classcell__r r r7 r r0 L s
r0 c sT e Zd ZdZeZg d�ZddgZ� fdd�Zdd� Z d d
� Z
d d � Z d
d� Z � Z
S )� KpatchPluginz�
The DNF plugin helps customers to install kpatch-patch packages
when the kernel is upgraded and filter kernel-core packages that
are supported by the kpatch team.
)�kernelr zkernel-moduleszkernel-modules-corezkernel-modules-extrarr zkernel-uname-rc s6 t � �||� d| _d| _d| _|d ur2|�t� d S )NF)r1 r2 �
_commiting� _autoupdate� _autofilterZregister_commandr0 )r5 r3 r6 r7 r r r2 � s zKpatchPlugin.__init__c
C s� | � | jj�}zL|�d�o0|�dt�o0|�dt�| _|�d�oT|�dt�oT|�dt�| _ W n< t
y� } z$t � t
d��t|��� W Y d }~n
d }~0 0 d S )NrZ rP )Z read_configr3 r+ r] rg rd rh rt rf ru rT r rQ r r rJ rW r r r �config s
�
�
�
�zKpatchPlugin.configc C s" d| _ | j�| jjj� d| _ d S )NTF)rs r3 �resolver6 rA Z
allow_erasing)r5 r r r �_commit_changes s z KpatchPlugin._commit_changesc
C s | j s
d S td� | jj�� jdd�}| jjjtjd�}|j| jd� |� � }| jjjtjd�}|jdd� |D ]|}|j
D ]p}t |�� d�}t
|�d k r�qz|d
| jv rz|j|d �}d }|D ]} | j} q�q�|d ur�|�|j|d ��} qpqzqp| jj�|�|�� d S )
NzOPlease note, kpatch filter is enabled, only kpatch supported kernels are shown.T)�empty)�flagsrD zkpatch-patch-*)Z
name__glob� � r )Zprovides)�evr)ru �printr3 r# r$ Zfilterm�hawkeyZIGNORE_EXCLUDES�kernel_pkg_namesZ availableZrequiresrJ �split�len�kpatch_requirementr% r} �unionZ add_excludesrH )
r5 Z kernels_keepZ
kernels_queryZ kpatch_queryZ
kpatch_pkgZrequireZrequire_parsedZ kernel_coreZ
kernel_evrZkernel_core_pkgr r r r# s0
zKpatchPlugin.sackc 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
V s �z)KpatchPlugin.resolved.<locals>.<listcomp>)rt rs r3 Z transactionr9 r& ZFORWARD_ACTIONSr" r rF rI �
startswithr/ rx )r5 Z need_kpp_forZtr_itemr r r� r �resolvedC s zKpatchPlugin.resolved)rj rk rl rm r r r� r� r2 rv rx r# r� rp r r r7 r rq � s
*rq ) rm Z
__future__r r rR Zos.pathr r Zdnfpluginscorer r r& Z dnf.callbackZdnf.cliZdnf.exceptionsZdnf.transactionr r rd rf rF r r r/ r6 ZCommandr0 ZPluginrq r r r r �<module> s, #