Mini Shell
3
�$`�! � @ s� d dl Z d dlZd dlmZmZ d dlmZ d dlmZ d dl m
Z
d dl m Z m
Z
mZ G dd� de�Zd d
� ZG d d � d e
�ZdS )
� N)�_�YumBase)�DownloadBaseCallback)� YumBaseError)�DebugInfoDownload)�log1�log2�RETURN_FAILUREc @ s e Zd ZdZdd� Zdd� ZdS )�YumDownloadCallbackzR
This class serves as a download progress handler for yum's progress bar.
c C s t j| � || _dS )zx
Sets up instance variables
Arguments:
total_pkgs - number of packages to download
N)r �__init__�observer)�selfr � r �$/usr/lib64/python3.6/yumdebuginfo.pyr # s
z YumDownloadCallback.__init__c C s | j j|t|d �� dS )a
A method used to update the progress
Arguments:
name - filename
frac - progress fracment (0 -> 1)
fread - formated string containing BytesRead
ftime - formated string containing remaining or elapsed time
�d N)r �update�int)r
�nameZfracZfreadZftimer r r �updateProgress/ s z"YumDownloadCallback.updateProgressN)�__name__�
__module__� __qualname__�__doc__r r r r r r r
s r
c C s$ t td�jt| j�| j�� ddiS )z0
A callback function for mirror errors.
zOProblem '{0!s}' occured while downloading from mirror: '{1!s}'. Trying next oneZfailr )�printr �format�strZ exceptionZmirror)Z callBackObjr r r �downloadErrorCallback= s
r c sF e Zd Zd� fdd� Zdd� Zd d
� Zd d � Zd
d� Zdd� Z� Z S )�YumDebugInfoDownload�*debug*FTNc s: t t| �j|||||� t� | _|d k r6|| jjjd< d S )N�
releasever)�superr r r �baseZconfZ
substitutions)r
�cacheZtmp� repo_patternZ keep_rpmsZnoninteractiver )� __class__r r r K s z YumDebugInfoDownload.__init__c C s, t |�| _| jjj| j� | jjjt� d S )N)r
Zprogressr! �reposZsetProgressBarZsetMirrorFailureCallbackr )r
Zupdaterr r r �initialize_progressS s
z(YumDebugInfoDownload.initialize_progressc C s� | j � y| jj� W nH tk
r^ } z,| j� ttd�jt|��� t d� W Y d d }~X nX | j� | jj
� s�ttd�� t S xn| jj j
� D ]^}y |j� | jj j|j� W q� tk
r� } z ttd�j|jt|��� W Y d d }~X q�X q�W d S )Nz7Error initializing yum (YumBase.doConfigSetup): '{0!s}'� z#Error: can't make cachedir, exitingz'Can't disable repository '{0!s}': {1!s})Z mute_stdoutr! Z
doConfigSetupr Z
unmute_stdoutr r r r �exitZ setCacheDirr r% Z listEnabled�close� disableRepo�id)r
�exZrepor r r �prepareX s"
z YumDebugInfoDownload.preparec C s� x�| j jj| jd�D ]�}y�| j jj|j�}| j jjt|j�d� td|� t |dd� yt |dd� W n> t
t fk
r� } z t t|�� t t
d�� W Y d d }~X nX W q tk
r� } z0t t
d �j|jt|��� | j jj|j� W Y d d }~X qX qW y| j jjd
d d � W n: tk
�rJ } z t t
d
�jt|��� W Y d d }~X nX y| j jjdd d � W n: tk
�r� } z t t
d�jt|��� W Y d d }~X nX d S )N)�pattern)Zthisrepozenabled repo %sZskip_if_unavailableTZ_asyncFzACan't disable async download, the output might contain artifacts!z Can't setup {0}: {1}, disablingZmetadatar' )ZmdtypeZ cacheonlyz"Error retrieving metadata: '{0!s}'Z filelistsz#Error retrieving filelists: '{0!s}')r! r% Z findReposr# Z
enableRepor+ ZdoSetupr r �setattr� NameError�AttributeErrorr r r r r* Z populateSack)r
�rZridr, r r r �initialize_repositoriesv s,
"&(z,YumDebugInfoDownload.initialize_repositoriesc C s� g }i }d}d}x�|D ]�}t d|� | jjj|�}|r�|d |j� krZ||d j|� n2|g||d < |t|d j�7 }|t|d j�7 }t d||d � qt d|� |j|� qW ||||fS )Nr zyum whatprovides %szfound pkg for %s: %sznot found pkg for %s) r r! ZpkgSackZ searchFiles�keys�append�float�sizeZ
installedsize)r
�filesZ not_foundZpackage_files_dictZtodownload_sizeZinstalled_sizeZdebuginfo_path�pkgr r r �triage� s"
zYumDebugInfoDownload.triagec C s� |j d�}tjj|�}tjj| j|�}|j d�}d }|jd�r�|dd � }td|� ytj ||� W q� t
k
r� } zt d�j |t
|��}W Y d d }~X q�X n||_| jj|gd�}||fS )NZ relativepathZ
remote_urlzfile:///� zcopying from local repo: %szCannot copy file '{0}': {1})Zpkglist)Z returnSimple�os�path�basename�joinZtmpdir�
startswithr �shutil�copy�OSErrorr r r Z localpathr! Z downloadPkgs)r
r9 ZremoteZlocalZ remote_path�errZpkg_pathr, r r r �download_package� s
(z%YumDebugInfoDownload.download_package)r FTN)
r r r r r& r- r3 r: rE �
__classcell__r r )r$ r r I s /r )r< rA Zyumr r Z
yum.callbacksr Z
yum.Errorsr Zreportclient.debuginfor Z reportclientr r r r
r r r r r r �<module> s