111 lines
3.5 KiB
Python
111 lines
3.5 KiB
Python
#
|
|
# dnfhelper.py
|
|
#
|
|
# Copyright (C) 2010-2015 Red Hat, Inc.
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation; either version 2 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
# Red Hat Author(s): Martin Gracik <mgracik@redhat.com>
|
|
# Brian C. Lane <bcl@redhat.com>
|
|
#
|
|
|
|
import logging
|
|
logger = logging.getLogger("pylorax.dnfhelper")
|
|
import dnf
|
|
import dnf.transaction
|
|
import collections
|
|
import time
|
|
import pylorax.output as output
|
|
|
|
__all__ = ['LoraxDownloadCallback', 'LoraxRpmCallback']
|
|
|
|
def _paced(fn):
|
|
"""Execute `fn` no more often then every 2 seconds."""
|
|
def paced_fn(self, *args):
|
|
now = time.time()
|
|
if now - self.last_time < 2:
|
|
return
|
|
self.last_time = now
|
|
return fn(self, *args)
|
|
return paced_fn
|
|
|
|
|
|
class LoraxDownloadCallback(dnf.callback.DownloadProgress):
|
|
def __init__(self):
|
|
self.downloads = collections.defaultdict(int)
|
|
self.last_time = time.time()
|
|
self.total_files = 0
|
|
self.total_size = 0
|
|
|
|
self.pkgno = 0
|
|
self.total = 0
|
|
|
|
self.output = output.LoraxOutput()
|
|
|
|
@_paced
|
|
def _update(self):
|
|
msg = "Downloading %(pkgno)s / %(total_files)s RPMs, " \
|
|
"%(downloaded)s / %(total_size)s (%(percent)d%%) done.\n"
|
|
downloaded = sum(self.downloads.values())
|
|
vals = {
|
|
'downloaded' : downloaded,
|
|
'percent' : int(100 * downloaded/self.total_size),
|
|
'pkgno' : self.pkgno,
|
|
'total_files' : self.total_files,
|
|
'total_size' : self.total_size
|
|
}
|
|
self.output.write(msg % vals)
|
|
|
|
def end(self, payload, status, msg):
|
|
nevra = str(payload)
|
|
if status is dnf.callback.STATUS_OK:
|
|
self.downloads[nevra] = payload.download_size
|
|
self.pkgno += 1
|
|
self._update()
|
|
return
|
|
logger.critical("Failed to download '%s': %d - %s", nevra, status, msg)
|
|
|
|
def progress(self, payload, done):
|
|
nevra = str(payload)
|
|
self.downloads[nevra] = done
|
|
self._update()
|
|
|
|
# dnf 2.5.0 adds a new argument, accept it if it is passed
|
|
# pylint: disable=arguments-differ
|
|
def start(self, total_files, total_size, total_drpms=0):
|
|
self.total_files = total_files
|
|
self.total_size = total_size
|
|
|
|
|
|
class LoraxRpmCallback(dnf.callback.TransactionProgress):
|
|
def __init__(self):
|
|
super(LoraxRpmCallback, self).__init__()
|
|
self._last_ts = None
|
|
|
|
def progress(self, package, action, ti_done, ti_total, ts_done, ts_total):
|
|
if action == dnf.transaction.PKG_INSTALL:
|
|
# do not report same package twice
|
|
if self._last_ts == ts_done:
|
|
return
|
|
self._last_ts = ts_done
|
|
|
|
msg = '(%d/%d) %s' % (ts_done, ts_total, package)
|
|
logger.info(msg)
|
|
elif action == dnf.transaction.TRANS_POST:
|
|
msg = "Performing post-installation setup tasks"
|
|
logger.info(msg)
|
|
|
|
def error(self, message):
|
|
logger.warning(message)
|