File: //proc/self/root/lib/python3.6/site-packages/dnf-plugins/__pycache__/kpatch.cpython-36.pyc
3
�=vh�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
autofilterz kernel-corec C s2 dd� | j D �}x |D ]}tjj|�r|S qW d S )Nc S s g | ]}d |t f �qS )z
%s/%s.conf)�KPATCH_PLUGIN_NAME)�.0�path� r
� /usr/lib/python3.6/kpatch.py�
<listcomp>5 s 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 jd| j�jd�}|jdd�}| jjdd�}dj||�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 |�}| jj� j||jd�}tjj| j�}|j|j � d� | j
j || j j
d� d S )N)�name�arch)�pkg)ZselectZoptional)r �sack�query�filterr �dnfZselectorZSelector�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 � j|� t| jj�| _d S )N)�super�__init__r �baser' �cfg_file)�self�cli)� __class__r
r r. 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|_| jjdkr.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)r9 r& r: r; )r2 �demandsZ root_user�optsr4 Z resolvingZsack_activationZavailable_repos)r1 r< r
r
r � configured s zKpatchCmd.configurec C s� g }| j jj� j� jtd�}x�|D ]�}t|�}| j jj� j� j|d�j� }|r�| j jj� j||d j|d j d�}| j jj� j||j
d�j � j |�}n | j jj� j||j
d�j � }x|D ]}|j
t|�� q�W q"W |S )N)r r )r r r )r r )r/ r r! � installedr" �KERNEL_PKG_NAMEr �runr r r r% �
difference�append�str) r1 �kpps�installed_kernelsr r) r? Zsub_qr* r r
r
r �_list_missing_kpp_pkgsv s(
z KpatchCmd._list_missing_kpp_pkgsc C s6 | j jj� j� jtd�}x|D ]}t| j |� q W d S )N)r )r/ r r! r? r" r@ r+ )r1 rF r r
r
r �_install_missing_kpp_pkgs� s
z#KpatchCmd._install_missing_kpp_pkgsc C sr | j d krtjd� d S ytj� }|j| j � |S tk
rl } z tjj t
d�j t |����W Y d d }~X nX d S )Nz Couldn't find configuration filezParsing file failed: {})
r0 r �warning� configparserZ ConfigParser�read� Exceptionr# �
exceptions�Errorr r rD )r1 r7 �er
r
r �
_read_conf� s
zKpatchCmd._read_confc C s� | j d krtjd� d S | j� }|d kr,d S |jd�s@|jd� |jd|t|�� y*t| j ddd��}|j |� W d Q R X W n< t
k
r� } z t j j
td�jt|����W Y d d }~X nX d S )Nz Couldn't find configuration file�main�wzutf-8)�encodingz Failed to update conf file: {})r0 r rI rP � has_sectionZ add_sectionr$ rD �open�writerL r# rM rN r r )r1 Zoption�valuer' Z
cfg_streamrO r
r
r �_update_plugin_cfg� s
z KpatchCmd._update_plugin_cfgc C s� | j j}|dkr<| j� | jtd� tjtd�j|�� �n�|dkrh| jtd� tjtd�j|�� �nZ|dkr�| jt d� tjtd �j|�� �n.|d
kr�| jt d� tjtd �j|�� �n|d k�r�| j
� }d}|d k �r |j d
��r |j d
t��r |j
d
t��r d}tjtd�j|�� d
}|d k �rZ|j d
��rZ|j d
t ��rZ|j
d
t ��rZd}tjtd �j|�� | j� }|�r�tjtd�jdj|��� n*|dk�r�| j� ntjjtd�j|���d S )z:
Decision tree, execution based on config
� auto-updater; TzKpatch update setting: {}�
manual-update�manualFz auto-filterzKpatch filter setting: {}z no-filterr: NrQ zAvailable patches: {}z, r& zInvalid argument: {})rY r; )rZ r[ )r= r4 rH rX �KPATCH_UPDATE_OPTr �infor r �KPATCH_FILTER_OPTrP rT �
has_option�
getbooleanrG �joinr# rM rN )r1 r4 r' Z kp_statusrE r
r
r rA � sD
z
KpatchCmd.run)r )�__name__�
__module__� __qualname__�__doc__�aliasesr Zsummaryr. � staticmethodr8 r> rG rH rP rX rA �
__classcell__r
r
)r3 r r, L s r, c sZ e Zd ZdZeZdddddgZddgZ� fdd �Zd
d � Z d d
� Z
dd� Z dd� 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.
Zkernelz kernel-corezkernel-moduleszkernel-modules-corezkernel-modules-extrazkernel-uname-rc s6 t � j||� d| _d| _d| _|d k r2|jt� d S )NF)r- r. �
_commiting� _autoupdate� _autofilterZregister_commandr, )r1 r/ r2 )r3 r
r r. � s zKpatchPlugin.__init__c C s� | j | jj�}yL|jd�o0|jdt�o0|jdt�| _|jd�oT|jdt�oT|jdt�| _ W n: t
k
r� } z t j t
d�jt|��� W Y d d }~X nX d S )NrQ zParsing file failed: {})Z read_configr/ r' rT r_ r\ r` rk r^ rl rL r rI r r rD )r1 r7 rO r
r
r �config s
zKpatchPlugin.configc C s" d| _ | jj| jjj� d| _ d S )NTF)rj r/ Zresolver2 r< Z
allow_erasing)r1 r
r
r �_commit_changes s z KpatchPlugin._commit_changesc
C s | j s
d S td� | jjj� jdd�}| jjjtjd�}|j| jd� |j � }| jjjtjd�}|jdd� x�|D ]�}xz|j
D ]p}t |�j d�}t
|�d k r�q~|d
| jkr~|j|d �}d }x|D ] } | j}P q�W |d k r�|j|j|d ��}P q~W qrW | jjj|j|�� d S )
NzOPlease note, kpatch filter is enabled, only kpatch supported kernels are shown.T)�empty)�flags)r zkpatch-patch-*)Z
name__glob� � r )Zprovides)�evr)rl �printr/ r r! Zfilterm�hawkeyZIGNORE_EXCLUDES�kernel_pkg_namesZ availableZrequiresrD �split�len�kpatch_requirementr" rs �unionZ add_excludesrB )
r1 Z kernels_keepZ
kernels_queryZ kpatch_queryZ
kpatch_pkgZrequireZrequire_parsedZ kernel_coreZ
kernel_evrZkernel_core_pkgr
r
r r s0
zKpatchPlugin.sackc s� | j s| jrd S g }g � xT| jjD ]H}|jtjjkr$|jjt krP|j
|j� q$|jjj d�r$� j
|jj� q$W � fdd�|D �}x|D ]}t | j|� q�W |r�| j
� d S )Nz
kpatch-patch-c s g | ]}t |�� kr|�qS r
)r )r r )�explicit_kpp_installr
r r V s z)KpatchPlugin.resolved.<locals>.<listcomp>)rk rj r/ Z transactionr4 r# ZFORWARD_ACTIONSr r r@ rC �
startswithr+ rn )r1 Z need_kpp_forZtr_itemr r
)r{ r �resolvedC s
zKpatchPlugin.resolved)rb rc rd re r r rv ry r. rm rn r r} rh r
r
)r3 r ri � s
*ri ) re Z
__future__r r rJ Zos.pathr
r Zdnfpluginscorer r r# Z dnf.callbackZdnf.cliZdnf.exceptionsZdnf.transactionru r r\ r^ r@ r r r+ r2 ZCommandr, ZPluginri r
r
r
r �<module> s, #