Sindbad~EG File Manager
3
�&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 �}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>+ s 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 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_pkg2 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_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 )� KpatchCmdr z7Toggles automatic installation of kpatch-patch packagesc s" t t| �j|� t| jj�| _d S )N)�superr, �__init__r �baser' �cfg_file)�self�cli)� __class__r
r r. 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|_| jjdkr.d|_d|_d|_nd|_d|_d|_d S )NT�autor& �statusF)r9 r& r: )r2 �demandsZ root_user�optsr4 Z resolvingZsack_activationZavailable_repos)r1 r; r
r
r � configureR 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_pkgs` 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 rE r r
r
r �_install_missing_kpp_pkgst 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 rC )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� |jdtt|�� y&t | j d��}|j
|� W d Q R X W n< t k
r� } z t j
jtd�jt|����W Y d d }~X nX d S )Nz Couldn't find configuration file�main�wz Failed to update conf file: {})r0 r rH rO � has_sectionZ add_sectionr$ �KPATCH_UPDATE_OPTrC �open�writerK r# rL rM r r )r1 �valuer' Z
cfg_streamrN r
r
r �_update_plugin_cfg� s
z KpatchCmd._update_plugin_cfgc C s� | j j}|dkr$| j� | jd� n�|dkr8| jd� n�|dkr�| j� }d}|d k rz|jd�rz|jdt�rz|jdt�rzd}t j
t d�j |�� | j
� }|r�t j
t d�j d j|��� n(|d
kr�| j� ntjjt d �j |���d S ) Nr9 TZmanualFr: rP zKpatch update setting: {}zAvailable patches: {}z, r& zInvalid argument: {})r< r4 rG rW rO rR �
has_optionrS �
getbooleanr �infor r rF �joinr# rL rM )r1 r4 r' Z kp_statusrD r
r
r r@ � s(
z
KpatchCmd.run)r )�__name__�
__module__� __qualname__�aliasesr Zsummaryr. � staticmethodr8 r= rF rG rO rW r@ �
__classcell__r
r
)r3 r r, B s r, c s8 e Zd ZeZ� fdd�Zdd� Zdd� Zdd� Z� Z S ) � KpatchPluginc s4 t t| �j||� d| _d| _|d k r0|jt� d S )NF)r- rb r. �
_commiting� _autoupdateZregister_commandr, )r1 r/ r2 )r3 r
r r. � s
zKpatchPlugin.__init__c C sv | j | jj�}y(|jd�o0|jdt�o0|jdt�| _W n: tk
rp } z t j
t d�j t
|��� W Y d d }~X nX d S )NrP zParsing file failed: {})Z read_configr/ r' rR rX rS rY rd rK r rH r r rC )r1 r7 rN r
r
r �config� s
zKpatchPlugin.configc C s" d| _ | jj| jjj� d| _ d S )NTF)rc r/ Zresolver2 r; Z
allow_erasing)r1 r
r
r �_commit_changes� s z KpatchPlugin._commit_changesc 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 � s z)KpatchPlugin.resolved.<locals>.<listcomp>)rd rc r/ Z transactionr4 r# ZFORWARD_ACTIONSr r r? rB �
startswithr+ rf )r1 Z need_kpp_forZtr_itemr r
)rg r �resolved� s
zKpatchPlugin.resolved)
r\ r] r^ r r r. re rf ri ra r
r
)r3 r rb � s
rb )Z
__future__r r rI Zos.pathr
r Zdnfpluginscorer r Z dnf.callbackr# Zdnf.cliZdnf.exceptionsZdnf.transactionr rS r? r r r+ r2 ZCommandr, ZPluginrb r
r
r
r �<module> s" u
Sindbad File Manager Version 1.0, Coded By Sindbad EG ~ The Terrorists