118 lines
3.9 KiB
Python
118 lines
3.9 KiB
Python
#
|
|
# yumhelper.py
|
|
#
|
|
# Copyright (C) 2010 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>
|
|
#
|
|
|
|
import logging
|
|
logger = logging.getLogger("pylorax.yumhelper")
|
|
import sys, os, re
|
|
import yum, yum.callbacks, yum.rpmtrans
|
|
import output
|
|
|
|
__all__ = ['LoraxDownloadCallback', 'LoraxTransactionCallback',
|
|
'LoraxRpmCallback']
|
|
|
|
class LoraxDownloadCallback(yum.callbacks.DownloadBaseCallback):
|
|
|
|
def __init__(self):
|
|
yum.callbacks.DownloadBaseCallback.__init__(self)
|
|
self.output = output.LoraxOutput()
|
|
|
|
pattern = "\((?P<pkgno>\d+)/(?P<total>\d+)\):\s+(?P<pkgname>.*)"
|
|
self.pattern = re.compile(pattern)
|
|
|
|
def updateProgress(self, name, frac, fread, ftime):
|
|
"""
|
|
Update the progress bar
|
|
@param name: filename
|
|
@param frac: progress fraction (0 -> 1)
|
|
@param fread: formated string containing BytesRead
|
|
@param ftime: formated string containing remaining or elapsed time
|
|
"""
|
|
|
|
match = self.pattern.match(name)
|
|
|
|
pkgno = 0
|
|
total = 0
|
|
pkgname = "error"
|
|
if match:
|
|
pkgno = int(match.group("pkgno"))
|
|
total = int(match.group("total"))
|
|
pkgname = match.group("pkgname")
|
|
|
|
info = "({0:3d}/{1:3d}) [{2:3.0f}%] downloading "
|
|
info = info.format(pkgno, total, frac * 100)
|
|
|
|
infolen, pkglen = len(info), len(pkgname)
|
|
if (infolen + pkglen) > self.output.width:
|
|
pkgname = "{0}...".format(pkgname[:self.output.width-infolen-3])
|
|
|
|
msg = "{0}<b>{1}</b>\r".format(info, pkgname)
|
|
self.output.write(msg)
|
|
if frac == 1:
|
|
self.output.write("\n")
|
|
|
|
|
|
class LoraxTransactionCallback(object):
|
|
|
|
def __init__(self):
|
|
self.output = output.LoraxOutput()
|
|
|
|
def event(self, state, data=None):
|
|
if state == yum.callbacks.PT_DOWNLOAD:
|
|
self.output.write("downloading packages\n")
|
|
elif state == yum.callbacks.PT_DOWNLOAD_PKGS:
|
|
pass
|
|
elif state == yum.callbacks.PT_GPGCHECK:
|
|
self.output.write("checking package signatures\n")
|
|
elif state == yum.callbacks.PT_TEST_TRANS:
|
|
self.output.write("running test transaction\n")
|
|
elif state == yum.callbacks.PT_TRANSACTION:
|
|
self.output.write("running transaction\n")
|
|
|
|
|
|
class LoraxRpmCallback(yum.rpmtrans.RPMBaseCallback):
|
|
|
|
def __init__(self):
|
|
yum.rpmtrans.RPMBaseCallback.__init__(self)
|
|
self.output = output.LoraxOutput()
|
|
|
|
def event(self, package, action, te_current, te_total,
|
|
ts_current, ts_total):
|
|
|
|
info = "({0:3d}/{1:3d}) [{2:3.0f}%] {3} "
|
|
info = info.format(ts_current, ts_total,
|
|
float(te_current) / float(te_total) * 100,
|
|
self.action[action].lower())
|
|
|
|
pkg = "{0}".format(package)
|
|
|
|
infolen, pkglen = len(info), len(pkg)
|
|
if (infolen + pkglen) > self.output.width:
|
|
pkg = "{0}...".format(pkg[:self.output.width-infolen-3])
|
|
|
|
msg = "{0}<b>{1}</b>\r".format(info, pkg)
|
|
self.output.write(msg)
|
|
if te_current == te_total:
|
|
self.output.write("\n")
|
|
|
|
def filelog(self, package, action):
|
|
if self.fileaction.get(action) == "Installed":
|
|
logger.debug("{0} installed successfully".format(package))
|