3
�gt`L+ � @ sN d dl mZmZ d dlZd dlmZ G dd� dej�ZG dd� dejj �Z
dS )� )�_�loggerN)�Packagec s, e Zd ZdZdZ� fdd�Zdd� Z� ZS )�DebuginfoInstallz5DNF plugin supplying the 'debuginfo-install' command.zdebuginfo-installc s4 t t| �j||� || _|| _|dk r0|jt� dS )z Initialize the plugin instance.N)�superr �__init__�base�cliZregister_command�DebuginfoInstallCommand)�selfr r )� __class__� �'/usr/lib/python3.6/debuginfo-install.pyr s
zDebuginfoInstall.__init__c C sf | j | jj�}|jd�o.|jdd�o.|jdd�}|rbtjj| j�j � j
dd�}t |�rb| jj j
� d S )N�main�
autoupdatez *-debuginfo)Z
name__glob)Z read_configr �confZ has_sectionZ
has_optionZ
getboolean�dnf�sackZ _rpmdb_sack�query�filterm�len�repos�enable_debug_repos)r Zcpr Zdbginfor
r
r �config( s
zDebuginfoInstall.config)�__name__�
__module__� __qualname__�__doc__�namer r �
__classcell__r
r
)r r r 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 )r
z! DebuginfoInstall plugin for DNF �debuginfo-installzinstall debuginfo packagesc s4 t t| �j|� t� | _t� | _t� | _t� | _d S )N)r r
r �set�available_debuginfo_missing� available_debugsource_missing�installed_debuginfo_missing� installed_debugsource_missing)r r )r r
r r : s
z DebuginfoInstallCommand.__init__c C s | j ddd� d S )N�package�+)�nargs)� add_argument)�parserr
r
r �
set_argparserB s z%DebuginfoInstallCommand.set_argparserc C s0 | j j}d|_d|_d|_d|_| jjj� d S )NT) r �demandsZ resolvingZ root_userZsack_activationZavailable_reposr r r )r r, r
r
r � configureF s z!DebuginfoInstallCommand.configurec C s� g }t tj�}t tj�}�x�| jjD �]�}tjj|�j | j
j dd�}|d }|sxt j
td�| j
jjj|�� |j|� q$|j� j� }|j|j� j� � xdt|j� �D ]T}|jtj�r�|d | � |kr�|j|� |jtj�r�|d | � |kr�|j|� q�W �x�|j� D �]�} | d }
|
j �r�i } x"| D ]} | j | j g �j| � �q(W xj| j� D ]^}
|
d } | j | j | ��s�| j | j!| ��s�| j"j#t$| �� | j | j%| ��sP| j&j#t$| �� �qPW �q|
j'jtj��s�|
j'jtj��r�| j(| � �q|d d k �rb| j)|
j |d ��s2| j)|
j!|d ��s2| j*j#dj+|
j'|
j,�� | j)|
j%|d ��s| j-j#dj+|
j'|
j,�� �q| j.|
j | ��s�| j.|
j!| ��s�| j*j#dj+|
j'|
j,�� | j.|
j%| ��s| j-j#dj+|
j'|
j,�� �qW q$W | j*�r�t j
td�d j/t0| j*��� | j-�rt j
td
�d j/t0| j-��� | j"�r8t j
td �d j/t0| j"��� | j&�r\t j
td �d j/t0| j&��� |�r�| j
j1j2�r�tj3j4td
�dj/|�d��d S )NF)Zwith_srcr zNo match for argument: %sr Znevraz{}-{}zICould not find debuginfo package for the following available packages: %sz, zKCould not find debugsource package for the following available packages: %szICould not find debuginfo package for the following installed packages: %szKCould not find debugsource package for the following installed packages: %szUnable to find a match� )Zpkg_spec)5r r ZDEBUGINFO_SUFFIXZDEBUGSOURCE_SUFFIXZoptsr& r ZsubjectZSubjectZget_best_solutionr r r �infor �outputZtermZbold�appendZ availableZ
_name_dict�updateZ installed�list�keys�endswith�pop�valuesZ _from_system�
setdefault�arch�_install_debug_from_system�
debug_nameZsource_debug_namer$ �add�strZdebugsource_namer% r �_install�_install_debugr" �format�evrr# �_install_debug_no_nevra�join�sortedr �strict�
exceptionsZPackagesNotAvailableError)r Z errors_specZdebuginfo_suffix_lenZdebugsource_suffix_lenZpkgspecZsolutionr Z package_dictr �pkgsZ first_pkgZ arch_dict�pkgZpackage_arch_listr
r
r �runN s�
zDebuginfoInstallCommand.runc C s: | j jj� j||j|j|j|jd�}|r6| j|� dS dS )N)r �epoch�version�releaser9 TF) r r r �filterrJ rK rL r9 r> )r r; rH r r
r
r r: � s
z2DebuginfoInstallCommand._install_debug_from_systemc C s� i }|j d k r|j |d< |jd k r,|j|d< |jd k r@|j|d< |jd k rT|j|d< | jjj� jf d|i|��}|r�| j|� dS dS )NZ epoch__globZ
version__globZ
release__globZ
arch__globr TF) rJ rK rL r9 r r r rM r> )r r; Z
base_nevra�kwargsr r
r
r r? � s
z&DebuginfoInstallCommand._install_debugc s8 | j jj� j� fdd�|D �d�}|r4| j|� dS dS )Nc s g | ]}d j � |j|j��qS )z{}-{}.{})r@ rA r9 )�.0�p)r; r
r �
<listcomp>� s zCDebuginfoInstallCommand._install_debug_no_nevra.<locals>.<listcomp>)Z nevra_strictTF)r r r r r> )r r; rG r r
)r; r rB � s
z/DebuginfoInstallCommand._install_debug_no_nevrac C s: t jj| jj�}|j|d� | jjj|| jjj d� d S )N)rH )ZselectZoptional)
r �selectorZSelectorr r r! ZgoalZinstallr rE )r rG rR r
r
r r> � s z DebuginfoInstallCommand._install)r )r r r r �aliasesr Zsummaryr � staticmethodr+ r- rI r: r? rB r> r r
r
)r r r
4 s |
r
) Zdnfpluginscorer r r Z dnf.packager ZPluginr r ZCommandr
r
r
r
r �<module> s