EVOLUTION-MANAGER
Edit File: kpatch.cpython-36.pyc
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