Mini Shell
3
�$`P � @ s� d dl Z d dlZd dlZd dlmZmZ d dlmZmZm Z m
Z
m Z d dl m
Z
mZmZ d dlmZ G dd� dejj�ZG dd � d e�ZdS )
� N)�
DownloadError� RepoError)� STATUS_OK� STATUS_DRPM�STATUS_ALREADY_EXISTS�
STATUS_MIRROR�
STATUS_FAILED)�_�log1�log2)�DebugInfoDownloadc s4 e Zd Z� fdd�Zdd� Zdd� Zdd� Z� ZS ) � DNFProgressc s t t| �j� || _d S )N)�superr
�__init__�observer)�selfr )� __class__� �$/usr/lib64/python3.6/dnfdebuginfo.pyr $ s zDNFProgress.__init__c C sj |t ttgkr"| jjt|�d� nD|tkr<td|p4d � n*|tkrVt d|pNd � nt
j j d| � d S )N�d zMirror failed: %sz DNF did not provide more detailszDownloading failed: %sz Unknown DNF download status: %s
)
r r r r �update�strr r
r r �sys�stderr�write)r �payloadZstatus�msgr r r �end) s zDNFProgress.endc C s, t d� | jjt|�td||j �� d S )NzUpdated a packager )r r r r �intZ
download_size)r r �doner r r �progress9 s zDNFProgress.progressc C s t d� d S )Nz Started downloading of a package)r )r Z total_filesZ
total_sizer r r �start= s zDNFProgress.start)�__name__�
__module__� __qualname__r r r r! �
__classcell__r r )r r r
" s r
c sF e Zd Zd� fdd� Zdd� Zd d
� Zd d � Zd
d� Zdd� Z� Z S )�DNFDebugInfoDownload�*debug*FTNc sB t t| �j|||||� d | _tj� | _|d k r>|| jjjd< d S )N�
releasever) r r& r r �dnfZBase�baseZconfZ
substitutions)r �cacheZtmp� repo_patternZ keep_rpmsZnoninteractiver( )r r r r C s
z DNFDebugInfoDownload.__init__c C sP y| j j� W n< tjjk
rJ } z ttd�jt|��� W Y d d }~X nX d S )Nz/Error reading repository configuration: '{0!s}') r* Zread_all_reposr) �
exceptions�Error�printr �formatr )r �exr r r �prepareU s z DNFDebugInfoDownload.preparec C s t |�| _d S )N)r
r )r Zupdaterr r r �initialize_progress[ s z(DNFDebugInfoDownload.initialize_progressc C s� x | j jj� D ] }|j� qW x&| j jj| j�D ]}d|_|j� q0W y| j j� W n8 t k
r� } z t
t d�j t
|��� W Y d d }~X nX d S )NTz&Error setting up repositories: '{0!s}')r* Zrepos�all�disableZ get_matchingr, Zskip_if_unavailable�enableZ fill_sackr r/ r r0 r )r Zrepor1 r r r �initialize_repositories^ s z,DNFDebugInfoDownload.initialize_repositoriesc
s0 | j jj� }|j� }i }g }d}d}� fdd�� x�|D ]�}g } |j|d�}
|
sdtd|� |j|� q6| j|
d � |
d jrĈ ||
d |
d �} x2| D ]*} | | kr�| j| � tdj| |
d �� q�W xZ| D ]R} | |j � kr�|| j|� n&|g|| < |t
| j �7 }|t
| j �7 }td|| � q�W q6W ||||fS )Nr c s\ g }|j rX| j|j |jd�}x:|D ]2}|j|jkr"||kr"|j|� |� | ||�7 }q"W |S )z�
Recursive function to find all required packages of required packages of ...
origin - should stop infinite recursion (A => B => ... => X => A)
)Zprovides�arch)�requires�filterr8 �name�append)�query�package�originZrequired_pkg_listZpkg_reqs�p)�required_packagesr r rA t s
z6DNFDebugInfoDownload.triage.<locals>.required_packages)�fileznot found package for %sz"found required package {0} for {1}zfound packages for %s: %s)
r* Zsackr= Z availabler: r r< r9 r0 �keys�floatZ downloadsizeZ installsize)
r �filesZ dnf_queryZ
dnf_availableZpackage_files_dictZ not_foundZtodownload_sizeZinstalled_sizeZdebuginfo_pathZdi_package_listZpackagesZ package_reqs�pkgr )rA r �triagel s8
zDNFDebugInfoDownload.triagec C sL y| j j|g| j� W n( tk
r> } z d t|�fS d }~X nX |j� d fS )N)r* Zdownload_packagesr r r ZlocalPkg)r rF r1 r r r �download_package� s
z%DNFDebugInfoDownload.download_package)r' FTN)
r" r# r$ r r2 r3 r7 rG rH r% r r )r r r& A s 4r& )r r) Zdnf.rpmZdnf.exceptionsr r Z dnf.callbackr r r r r Z reportclientr r
r Zreportclient.debuginfor �callbackZDownloadProgressr
r& r r r r �<module> s