관리-도구
편집 파일: 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, #