Tryag File Manager
Home
-
Turbo Force
Current Path :
/
proc
/
self
/
root
/
usr
/
share
/
yum-cli
/
Upload File :
New :
File
Dir
//proc/self/root/usr/share/yum-cli/output.py
#!/usr/bin/python -t """This handles actual output from the cli""" # 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 Library General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program; if not, write to the Free Software # Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. # Copyright 2005 Duke University import sys import time import logging import types import gettext import rpm import re # For YumTerm from urlgrabber.progress import TextMeter import urlgrabber.progress from urlgrabber.grabber import URLGrabError from yum.misc import prco_tuple_to_string from yum.i18n import to_str, to_utf8, to_unicode import yum.misc from rpmUtils.miscutils import checkSignals from yum.constants import * from yum import logginglevels, _ from yum.rpmtrans import RPMBaseCallback from yum.packageSack import packagesNewestByNameArch from yum.i18n import utf8_width, utf8_width_fill, utf8_text_fill def _term_width(): """ Simple terminal width, limit to 20 chars. and make 0 == 80. """ if not hasattr(urlgrabber.progress, 'terminal_width_cached'): return 80 ret = urlgrabber.progress.terminal_width_cached() if ret == 0: return 80 if ret < 20: return 20 return ret class YumTextMeter(TextMeter): """ Text progress bar output. """ def update(self, amount_read, now=None): checkSignals() TextMeter.update(self, amount_read, now) class YumTerm: """some terminal "UI" helpers based on curses""" # From initial search for "terminfo and python" got: # http://aspn.activestate.com/ASPN/Cookbook/Python/Recipe/475116 # ...it's probably not copyrightable, but if so ASPN says: # # Except where otherwise noted, recipes in the Python Cookbook are # published under the Python license. __enabled = True if hasattr(urlgrabber.progress, 'terminal_width_cached'): columns = property(lambda self: _term_width()) __cap_names = { 'underline' : 'smul', 'reverse' : 'rev', 'normal' : 'sgr0', } __colors = { 'black' : 0, 'blue' : 1, 'green' : 2, 'cyan' : 3, 'red' : 4, 'magenta' : 5, 'yellow' : 6, 'white' : 7 } __ansi_colors = { 'black' : 0, 'red' : 1, 'green' : 2, 'yellow' : 3, 'blue' : 4, 'magenta' : 5, 'cyan' : 6, 'white' : 7 } __ansi_forced_MODE = { 'bold' : '\x1b[1m', 'blink' : '\x1b[5m', 'dim' : '', 'reverse' : '\x1b[7m', 'underline' : '\x1b[4m', 'normal' : '\x1b(B\x1b[m' } __ansi_forced_FG_COLOR = { 'black' : '\x1b[30m', 'red' : '\x1b[31m', 'green' : '\x1b[32m', 'yellow' : '\x1b[33m', 'blue' : '\x1b[34m', 'magenta' : '\x1b[35m', 'cyan' : '\x1b[36m', 'white' : '\x1b[37m' } __ansi_forced_BG_COLOR = { 'black' : '\x1b[40m', 'red' : '\x1b[41m', 'green' : '\x1b[42m', 'yellow' : '\x1b[43m', 'blue' : '\x1b[44m', 'magenta' : '\x1b[45m', 'cyan' : '\x1b[46m', 'white' : '\x1b[47m' } def __forced_init(self): self.MODE = self.__ansi_forced_MODE self.FG_COLOR = self.__ansi_forced_FG_COLOR self.BG_COLOR = self.__ansi_forced_BG_COLOR def reinit(self, term_stream=None, color='auto'): self.__enabled = True if not hasattr(urlgrabber.progress, 'terminal_width_cached'): self.columns = 80 self.lines = 24 if color == 'always': self.__forced_init() return # Output modes: self.MODE = { 'bold' : '', 'blink' : '', 'dim' : '', 'reverse' : '', 'underline' : '', 'normal' : '' } # Colours self.FG_COLOR = { 'black' : '', 'blue' : '', 'green' : '', 'cyan' : '', 'red' : '', 'magenta' : '', 'yellow' : '', 'white' : '' } self.BG_COLOR = { 'black' : '', 'blue' : '', 'green' : '', 'cyan' : '', 'red' : '', 'magenta' : '', 'yellow' : '', 'white' : '' } if color == 'never': self.__enabled = False return assert color == 'auto' # Curses isn't available on all platforms try: import curses except: self.__enabled = False return # If the stream isn't a tty, then assume it has no capabilities. if not term_stream: term_stream = sys.stdout if not term_stream.isatty(): self.__enabled = False return # Check the terminal type. If we fail, then assume that the # terminal has no capabilities. try: curses.setupterm(fd=term_stream.fileno()) except: self.__enabled = False return self._ctigetstr = curses.tigetstr if not hasattr(urlgrabber.progress, 'terminal_width_cached'): self.columns = curses.tigetnum('cols') self.lines = curses.tigetnum('lines') # Look up string capabilities. for cap_name in self.MODE: mode = cap_name if cap_name in self.__cap_names: cap_name = self.__cap_names[cap_name] self.MODE[mode] = self._tigetstr(cap_name) or '' # Colors set_fg = self._tigetstr('setf') if set_fg: for (color, val) in self.__colors.items(): self.FG_COLOR[color] = curses.tparm(set_fg, val) or '' set_fg_ansi = self._tigetstr('setaf') if set_fg_ansi: for (color, val) in self.__ansi_colors.items(): self.FG_COLOR[color] = curses.tparm(set_fg_ansi, val) or '' set_bg = self._tigetstr('setb') if set_bg: for (color, val) in self.__colors.items(): self.BG_COLOR[color] = curses.tparm(set_bg, val) or '' set_bg_ansi = self._tigetstr('setab') if set_bg_ansi: for (color, val) in self.__ansi_colors.items(): self.BG_COLOR[color] = curses.tparm(set_bg_ansi, val) or '' def __init__(self, term_stream=None, color='auto'): self.reinit(term_stream, color) def _tigetstr(self, cap_name): # String capabilities can include "delays" of the form "$<2>". # For any modern terminal, we should be able to just ignore # these, so strip them out. cap = self._ctigetstr(cap_name) or '' return re.sub(r'\$<\d+>[/*]?', '', cap) def sub(self, haystack, beg, end, needles, escape=None, ignore_case=False): if not self.__enabled: return haystack if not escape: escape = re.escape render = lambda match: beg + match.group() + end for needle in needles: pat = escape(needle) if ignore_case: pat = re.template(pat, re.I) haystack = re.sub(pat, render, haystack) return haystack def sub_norm(self, haystack, beg, needles, **kwds): return self.sub(haystack, beg, self.MODE['normal'], needles, **kwds) def sub_mode(self, haystack, mode, needles, **kwds): return self.sub_norm(haystack, self.MODE[mode], needles, **kwds) def sub_bold(self, haystack, needles, **kwds): return self.sub_mode(haystack, 'bold', needles, **kwds) def sub_fg(self, haystack, color, needles, **kwds): return self.sub_norm(haystack, self.FG_COLOR[color], needles, **kwds) def sub_bg(self, haystack, color, needles, **kwds): return self.sub_norm(haystack, self.BG_COLOR[color], needles, **kwds) class YumOutput: """ Main output class for the yum command line. """ def __init__(self): self.logger = logging.getLogger("yum.cli") self.verbose_logger = logging.getLogger("yum.verbose.cli") if hasattr(rpm, "expandMacro"): self.i18ndomains = rpm.expandMacro("%_i18ndomains").split(":") else: self.i18ndomains = ["redhat-dist"] self.term = YumTerm() self._last_interrupt = None def printtime(self): months = [_('Jan'), _('Feb'), _('Mar'), _('Apr'), _('May'), _('Jun'), _('Jul'), _('Aug'), _('Sep'), _('Oct'), _('Nov'), _('Dec')] now = time.localtime(time.time()) ret = months[int(time.strftime('%m', now)) - 1] + \ time.strftime(' %d %T ', now) return ret def failureReport(self, errobj): """failure output for failovers from urlgrabber""" self.logger.error('%s: %s', errobj.url, errobj.exception) self.logger.error(_('Trying other mirror.')) raise errobj.exception def simpleProgressBar(self, current, total, name=None): progressbar(current, total, name) def _highlight(self, highlight): hibeg = '' hiend = '' if not highlight: pass elif not isinstance(highlight, basestring) or highlight == 'bold': hibeg = self.term.MODE['bold'] elif highlight == 'normal': pass # Minor opt. else: # Turn a string into a specific output: colour, bold, etc. for high in highlight.replace(',', ' ').split(): if False: pass elif high == 'normal': hibeg = '' elif high in self.term.MODE: hibeg += self.term.MODE[high] elif high in self.term.FG_COLOR: hibeg += self.term.FG_COLOR[high] elif (high.startswith('fg:') and high[3:] in self.term.FG_COLOR): hibeg += self.term.FG_COLOR[high[3:]] elif (high.startswith('bg:') and high[3:] in self.term.BG_COLOR): hibeg += self.term.BG_COLOR[high[3:]] if hibeg: hiend = self.term.MODE['normal'] return (hibeg, hiend) def _sub_highlight(self, haystack, highlight, needles, **kwds): hibeg, hiend = self._highlight(highlight) return self.term.sub(haystack, hibeg, hiend, needles, **kwds) @staticmethod def _calc_columns_spaces_helps(current, data_tups, left): """ Spaces left on the current field will help how many pkgs? """ ret = 0 for tup in data_tups: if left < (tup[0] - current): break ret += tup[1] return ret def calcColumns(self, data, columns=None, remainder_column=0, total_width=None, indent=''): """ Dynamically calculate the width of the fields in the data, data is of the format [column-number][field_length] = rows. """ if total_width is None: total_width = self.term.columns cols = len(data) # Convert the data to ascending list of tuples, (field_length, pkgs) pdata = data data = [None] * cols # Don't modify the passed in data for d in range(0, cols): data[d] = sorted(pdata[d].items()) if columns is None: columns = [1] * cols total_width -= (sum(columns) + (cols - 1) + utf8_width(indent)) while total_width > 0: # Find which field all the spaces left will help best helps = 0 val = 0 for d in xrange(0, cols): thelps = self._calc_columns_spaces_helps(columns[d], data[d], total_width) if not thelps: continue if thelps < helps: continue helps = thelps val = d # If we found a column to expand, move up to the next level with # that column and start again with any remaining space. if helps: diff = data[val].pop(0)[0] - columns[val] columns[val] += diff total_width -= diff continue # Split the remaining spaces among each column equally, except the # last one. And put the rest into the remainder column cols -= 1 norm = total_width / cols for d in xrange(0, cols): columns[d] += norm columns[remainder_column] += total_width - (cols * norm) total_width = 0 return columns @staticmethod def _fmt_column_align_width(width): if width < 0: return (u"-", -width) return (u"", width) def _col_data(self, col_data): assert len(col_data) == 2 or len(col_data) == 3 if len(col_data) == 2: (val, width) = col_data hibeg = hiend = '' if len(col_data) == 3: (val, width, highlight) = col_data (hibeg, hiend) = self._highlight(highlight) return (val, width, hibeg, hiend) def fmtColumns(self, columns, msg=u'', end=u''): """ Return a string for columns of data, which can overflow.""" total_width = len(msg) data = [] for col_data in columns[:-1]: (val, width, hibeg, hiend) = self._col_data(col_data) if not width: # Don't count this column, invisible text msg += u"%s" data.append(val) continue (align, width) = self._fmt_column_align_width(width) if utf8_width(val) <= width: msg += u"%s " val = utf8_width_fill(val, width, left=(align == u'-'), prefix=hibeg, suffix=hiend) data.append(val) else: msg += u"%s%s%s\n" + " " * (total_width + width + 1) data.extend([hibeg, val, hiend]) total_width += width total_width += 1 (val, width, hibeg, hiend) = self._col_data(columns[-1]) (align, width) = self._fmt_column_align_width(width) val = utf8_width_fill(val, width, left=(align == u'-'), prefix=hibeg, suffix=hiend) msg += u"%%s%s" % end data.append(val) return msg % tuple(data) def simpleList(self, pkg, ui_overflow=False, indent='', highlight=False, columns=None): """ Simple to use function to print a pkg as a line. """ if columns is None: columns = (-40, -22, -16) # Old default ver = pkg.printVer() na = '%s%s.%s' % (indent, pkg.name, pkg.arch) hi_cols = [highlight, 'normal', 'normal'] columns = zip((na, ver, pkg.repoid), columns, hi_cols) print self.fmtColumns(columns) def simpleEnvraList(self, pkg, ui_overflow=False, indent='', highlight=False, columns=None): """ Simple to use function to print a pkg as a line, with the pkg itself in envra format so it can be pased to list/install/etc. """ if columns is None: columns = (-63, -16) # Old default envra = '%s%s' % (indent, str(pkg)) hi_cols = [highlight, 'normal', 'normal'] columns = zip((envra, pkg.repoid), columns, hi_cols) print self.fmtColumns(columns) def fmtKeyValFill(self, key, val): """ Return a key value pair in the common two column output format. """ val = to_str(val) keylen = utf8_width(key) cols = self.term.columns nxt = ' ' * (keylen - 2) + ': ' ret = utf8_text_fill(val, width=cols, initial_indent=key, subsequent_indent=nxt) if ret.count("\n") > 1 and keylen > (cols / 3): # If it's big, redo it again with a smaller subsequent off ret = utf8_text_fill(val, width=cols, initial_indent=key, subsequent_indent=' ...: ') return ret def fmtSection(self, name, fill='='): name = to_str(name) cols = self.term.columns - 2 name_len = utf8_width(name) if name_len >= (cols - 4): beg = end = fill * 2 else: beg = fill * ((cols - name_len) / 2) end = fill * (cols - name_len - len(beg)) return "%s %s %s" % (beg, name, end) def _enc(self, s): """Get the translated version from specspo and ensure that it's actually encoded in UTF-8.""" s = to_utf8(s) if len(s) > 0: for d in self.i18ndomains: try: t = gettext.dgettext(d, s) except AttributeError: # Horrible hack for python bug t = s if t != s: s = t break return to_unicode(s) def infoOutput(self, pkg, highlight=False): (hibeg, hiend) = self._highlight(highlight) print _("Name : %s%s%s") % (hibeg, to_unicode(pkg.name), hiend) print _("Arch : %s") % to_unicode(pkg.arch) if pkg.epoch != "0": print _("Epoch : %s") % to_unicode(pkg.epoch) print _("Version : %s") % to_unicode(pkg.version) print _("Release : %s") % to_unicode(pkg.release) print _("Size : %s") % self.format_number(float(pkg.size)) print _("Repo : %s") % to_unicode(pkg.repoid) if self.verbose_logger.isEnabledFor(logginglevels.DEBUG_3): print _("Committer : %s") % to_unicode(pkg.committer) print _("Committime : %s") % time.ctime(pkg.committime) print _("Buildtime : %s") % time.ctime(pkg.buildtime) if hasattr(pkg, 'installtime'): print _("Installtime: %s") % time.ctime(pkg.installtime) print self.fmtKeyValFill(_("Summary : "), self._enc(pkg.summary)) if pkg.url: print _("URL : %s") % to_unicode(pkg.url) print _("License : %s") % to_unicode(pkg.license) print self.fmtKeyValFill(_("Description: "), self._enc(pkg.description)) print "" def updatesObsoletesList(self, uotup, changetype, columns=None): """takes an updates or obsoletes tuple of pkgobjects and returns a simple printed string of the output and a string explaining the relationship between the tuple members""" (changePkg, instPkg) = uotup if columns is not None: # New style, output all info. for both old/new with old indented chi = self.conf.color_update_remote if changePkg.repo.id != 'installed' and changePkg.verifyLocalPkg(): chi = self.conf.color_update_local self.simpleList(changePkg, columns=columns, highlight=chi) self.simpleList(instPkg, columns=columns, indent=' ' * 4, highlight=self.conf.color_update_installed) return # Old style c_compact = changePkg.compactPrint() i_compact = '%s.%s' % (instPkg.name, instPkg.arch) c_repo = changePkg.repoid print '%-35.35s [%.12s] %.10s %-20.20s' % (c_compact, c_repo, changetype, i_compact) def listPkgs(self, lst, description, outputType, highlight_na={}, columns=None, highlight_modes={}): """outputs based on whatever outputType is. Current options: 'list' - simple pkg list 'info' - similar to rpm -qi output ...also highlight_na can be passed, and we'll highlight pkgs with (names, arch) in that set.""" if outputType in ['list', 'info']: thingslisted = 0 if len(lst) > 0: thingslisted = 1 print '%s' % description for pkg in sorted(lst): key = (pkg.name, pkg.arch) highlight = False if False: pass elif key not in highlight_na: highlight = highlight_modes.get('not in', 'normal') elif pkg.verEQ(highlight_na[key]): highlight = highlight_modes.get('=', 'normal') elif pkg.verLT(highlight_na[key]): highlight = highlight_modes.get('>', 'bold') else: highlight = highlight_modes.get('<', 'normal') if outputType == 'list': self.simpleList(pkg, ui_overflow=True, highlight=highlight, columns=columns) elif outputType == 'info': self.infoOutput(pkg, highlight=highlight) else: pass if thingslisted == 0: return 1, ['No Packages to list'] return 0, [] def userconfirm(self): """gets a yes or no from the user, defaults to No""" yui = (to_unicode(_('y')), to_unicode(_('yes'))) nui = (to_unicode(_('n')), to_unicode(_('no'))) aui = (yui[0], yui[1], nui[0], nui[1]) while True: try: choice = raw_input(_('Is this ok [y/N]: ')) except UnicodeEncodeError: raise except UnicodeDecodeError: raise except: choice = '' choice = to_unicode(choice) choice = choice.lower() if len(choice) == 0 or choice in aui: break # If the enlish one letter names don't mix, allow them too if u'y' not in aui and u'y' == choice: choice = yui[0] break if u'n' not in aui and u'n' == choice: break if len(choice) == 0 or choice not in yui: return False else: return True def _cli_confirm_gpg_key_import(self, keydict): # FIXME what should we be printing here? return self.userconfirm() def _group_names2aipkgs(self, pkg_names): """ Convert pkg_names to installed pkgs or available pkgs, return value is a dict on pkg.name returning (apkg, ipkg). """ ipkgs = self.rpmdb.searchNames(pkg_names) apkgs = self.pkgSack.searchNames(pkg_names) apkgs = packagesNewestByNameArch(apkgs) # This is somewhat similar to doPackageLists() pkgs = {} for pkg in ipkgs: pkgs[(pkg.name, pkg.arch)] = (None, pkg) for pkg in apkgs: key = (pkg.name, pkg.arch) if key not in pkgs: pkgs[(pkg.name, pkg.arch)] = (pkg, None) elif pkg.verGT(pkgs[key][1]): pkgs[(pkg.name, pkg.arch)] = (pkg, pkgs[key][1]) # Convert (pkg.name, pkg.arch) to pkg.name dict ret = {} for (apkg, ipkg) in pkgs.itervalues(): pkg = apkg or ipkg ret.setdefault(pkg.name, []).append((apkg, ipkg)) return ret def _calcDataPkgColumns(self, data, pkg_names, pkg_names2pkgs, indent=' '): for item in pkg_names: if item not in pkg_names2pkgs: continue for (apkg, ipkg) in pkg_names2pkgs[item]: pkg = ipkg or apkg envra = utf8_width(str(pkg)) + utf8_width(indent) rid = len(pkg.repoid) for (d, v) in (('envra', envra), ('rid', rid)): data[d].setdefault(v, 0) data[d][v] += 1 def _displayPkgsFromNames(self, pkg_names, verbose, pkg_names2pkgs, indent=' ', columns=None): if not verbose: for item in sorted(pkg_names): print '%s%s' % (indent, item) else: for item in sorted(pkg_names): if item not in pkg_names2pkgs: print '%s%s' % (indent, item) continue for (apkg, ipkg) in sorted(pkg_names2pkgs[item], key=lambda x: x[1] or x[0]): if ipkg and apkg: highlight = self.conf.color_list_installed_older elif apkg: highlight = self.conf.color_list_available_install else: highlight = False self.simpleEnvraList(ipkg or apkg, ui_overflow=True, indent=indent, highlight=highlight, columns=columns) def displayPkgsInGroups(self, group): print _('\nGroup: %s') % group.ui_name verb = self.verbose_logger.isEnabledFor(logginglevels.DEBUG_3) if verb: print _(' Group-Id: %s') % to_unicode(group.groupid) pkg_names2pkgs = None if verb: pkg_names2pkgs = self._group_names2aipkgs(group.packages) if group.ui_description: print _(' Description: %s') % to_unicode(group.ui_description) sections = ((_(' Mandatory Packages:'), group.mandatory_packages), (_(' Default Packages:'), group.default_packages), (_(' Optional Packages:'), group.optional_packages), (_(' Conditional Packages:'), group.conditional_packages)) columns = None if verb: data = {'envra' : {}, 'rid' : {}} for (section_name, pkg_names) in sections: self._calcDataPkgColumns(data, pkg_names, pkg_names2pkgs) data = [data['envra'], data['rid']] columns = self.calcColumns(data) columns = (-columns[0], -columns[1]) for (section_name, pkg_names) in sections: if len(pkg_names) > 0: print section_name self._displayPkgsFromNames(pkg_names, verb, pkg_names2pkgs, columns=columns) def depListOutput(self, results): """take a list of findDeps results and 'pretty print' the output""" for pkg in results: print _("package: %s") % pkg.compactPrint() if len(results[pkg]) == 0: print _(" No dependencies for this package") continue for req in results[pkg]: reqlist = results[pkg][req] print _(" dependency: %s") % prco_tuple_to_string(req) if not reqlist: print _(" Unsatisfied dependency") continue for po in reqlist: print " provider: %s" % po.compactPrint() def format_number(self, number, SI=0, space=' '): """Turn numbers into human-readable metric-like numbers""" symbols = ['', # (none) 'k', # kilo 'M', # mega 'G', # giga 'T', # tera 'P', # peta 'E', # exa 'Z', # zetta 'Y'] # yotta if SI: step = 1000.0 else: step = 1024.0 thresh = 999 depth = 0 max_depth = len(symbols) - 1 # we want numbers between 0 and thresh, but don't exceed the length # of our list. In that event, the formatting will be screwed up, # but it'll still show the right number. while number > thresh and depth < max_depth: depth = depth + 1 number = number / step if type(number) == type(1) or type(number) == type(1L): format = '%i%s%s' elif number < 9.95: # must use 9.95 for proper sizing. For example, 9.99 will be # rounded to 10.0 with the .1f format string (which is too long) format = '%.1f%s%s' else: format = '%.0f%s%s' return(format % (float(number or 0), space, symbols[depth])) @staticmethod def format_time(seconds, use_hours=0): return urlgrabber.progress.format_time(seconds, use_hours) def matchcallback(self, po, values, matchfor=None, verbose=None, highlight=None): """ Output search/provides type callback matches. po is the pkg object, values are the things in the po that we've matched. If matchfor is passed, all the strings in that list will be highlighted within the output. verbose overrides logginglevel, if passed. """ if self.conf.showdupesfromrepos: msg = '%s : ' % po else: msg = '%s.%s : ' % (po.name, po.arch) msg = self.fmtKeyValFill(msg, self._enc(po.summary)) if matchfor: if highlight is None: highlight = self.conf.color_search_match msg = self._sub_highlight(msg, highlight, matchfor,ignore_case=True) print msg if verbose is None: verbose = self.verbose_logger.isEnabledFor(logginglevels.DEBUG_3) if not verbose: return print _("Repo : %s") % po.repoid print _('Matched from:') for item in yum.misc.unique(values): item = to_utf8(item) if to_utf8(po.name) == item or to_utf8(po.summary) == item: continue # Skip double name/summary printing can_overflow = True if False: pass elif to_utf8(po.description) == item: key = _("Description : ") item = self._enc(item) elif to_utf8(po.url) == item: key = _("URL : %s") can_overflow = False elif to_utf8(po.license) == item: key = _("License : %s") can_overflow = False elif item.startswith("/"): key = _("Filename : %s") item = self._enc(item) can_overflow = False else: key = _("Other : ") if matchfor: item = self._sub_highlight(item, highlight, matchfor, ignore_case=True) if can_overflow: print self.fmtKeyValFill(key, to_unicode(item)) else: print key % item print '\n\n' def matchcallback_verbose(self, po, values, matchfor=None): return self.matchcallback(po, values, matchfor, verbose=True) def reportDownloadSize(self, packages): """Report the total download size for a set of packages""" totsize = 0 locsize = 0 error = False for pkg in packages: # Just to be on the safe side, if for some reason getting # the package size fails, log the error and don't report download # size try: size = int(pkg.size) totsize += size try: if pkg.verifyLocalPkg(): locsize += size except: pass except: error = True self.logger.error(_('There was an error calculating total download size')) break if (not error): if locsize: self.verbose_logger.log(logginglevels.INFO_1, _("Total size: %s"), self.format_number(totsize)) if locsize != totsize: self.verbose_logger.log(logginglevels.INFO_1, _("Total download size: %s"), self.format_number(totsize - locsize)) def listTransaction(self): """returns a string rep of the transaction in an easy-to-read way.""" self.tsInfo.makelists(True, True) out = u"" pkglist_lines = [] data = {'n' : {}, 'v' : {}, 'r' : {}} a_wid = 0 # Arch can't get "that big" ... so always use the max. def _add_line(lines, data, a_wid, po, obsoletes=[]): (n,a,e,v,r) = po.pkgtup evr = po.printVer() repoid = po.repoid pkgsize = float(po.size) size = self.format_number(pkgsize) if a is None: # gpgkeys are weird a = 'noarch' # none, partial, full? if po.repo.id == 'installed': hi = self.conf.color_update_installed elif po.verifyLocalPkg(): hi = self.conf.color_update_local else: hi = self.conf.color_update_remote lines.append((n, a, evr, repoid, size, obsoletes, hi)) # Create a dict of field_length => number of packages, for # each field. for (d, v) in (("n",len(n)), ("v",len(evr)), ("r",len(repoid))): data[d].setdefault(v, 0) data[d][v] += 1 if a_wid < len(a): # max() is only in 2.5.z a_wid = len(a) return a_wid for (action, pkglist) in [(_('Installing'), self.tsInfo.installed), (_('Updating'), self.tsInfo.updated), (_('Removing'), self.tsInfo.removed), (_('Reinstalling'), self.tsInfo.reinstalled), (_('Downgrading'), self.tsInfo.downgraded), (_('Installing for dependencies'), self.tsInfo.depinstalled), (_('Updating for dependencies'), self.tsInfo.depupdated), (_('Removing for dependencies'), self.tsInfo.depremoved)]: lines = [] for txmbr in pkglist: a_wid = _add_line(lines, data, a_wid, txmbr.po, txmbr.obsoletes) pkglist_lines.append((action, lines)) for (action, pkglist) in [(_('Skipped (dependency problems)'), self.skipped_packages),]: lines = [] for po in pkglist: a_wid = _add_line(lines, data, a_wid, po) pkglist_lines.append((action, lines)) if data['n']: data = [data['n'], {}, data['v'], data['r'], {}] columns = [1, a_wid, 1, 1, 5] columns = self.calcColumns(data, indent=" ", columns=columns, remainder_column=2) (n_wid, a_wid, v_wid, r_wid, s_wid) = columns assert s_wid == 5 out = u""" %s %s %s """ % ('=' * self.term.columns, self.fmtColumns(((_('Package'), -n_wid), (_('Arch'), -a_wid), (_('Version'), -v_wid), (_('Repository'), -r_wid), (_('Size'), s_wid)), u" "), '=' * self.term.columns) for (action, lines) in pkglist_lines: if lines: totalmsg = u"%s:\n" % action for (n, a, evr, repoid, size, obsoletes, hi) in lines: columns = ((n, -n_wid, hi), (a, -a_wid), (evr, -v_wid), (repoid, -r_wid), (size, s_wid)) msg = self.fmtColumns(columns, u" ", u"\n") hibeg, hiend = self._highlight(self.conf.color_update_installed) for obspo in obsoletes: appended = _(' replacing %s%s%s.%s %s\n\n') appended %= (hibeg, obspo.name, hiend, obspo.arch, obspo.printVer()) msg = msg+appended totalmsg = totalmsg + msg if lines: out = out + totalmsg summary = _(""" Transaction Summary %s """) % ('=' * self.term.columns,) out = out + summary num_in = len(self.tsInfo.installed + self.tsInfo.depinstalled) num_up = len(self.tsInfo.updated + self.tsInfo.depupdated) summary = _("""\ Install %5.5s Package(s) Upgrade %5.5s Package(s) """) % (num_in, num_up,) if num_in or num_up: # Always do this? out = out + summary num_rm = len(self.tsInfo.removed + self.tsInfo.depremoved) num_re = len(self.tsInfo.reinstalled) num_dg = len(self.tsInfo.downgraded) summary = _("""\ Remove %5.5s Package(s) Reinstall %5.5s Package(s) Downgrade %5.5s Package(s) """) % (num_rm, num_re, num_dg) if num_rm or num_re or num_dg: out = out + summary return out def postTransactionOutput(self): out = '' self.tsInfo.makelists() # Works a bit like calcColumns, but we never overflow a column we just # have a dynamic number of columns. def _fits_in_cols(msgs, num): """ Work out how many columns we can use to display stuff, in the post trans output. """ if len(msgs) < num: return [] left = self.term.columns - ((num - 1) + 2) if left <= 0: return [] col_lens = [0] * num col = 0 for msg in msgs: if len(msg) > col_lens[col]: diff = (len(msg) - col_lens[col]) if left <= diff: return [] left -= diff col_lens[col] = len(msg) col += 1 col %= len(col_lens) for col in range(len(col_lens)): col_lens[col] += left / num col_lens[col] *= -1 return col_lens for (action, pkglist) in [(_('Removed'), self.tsInfo.removed), (_('Dependency Removed'), self.tsInfo.depremoved), (_('Installed'), self.tsInfo.installed), (_('Dependency Installed'), self.tsInfo.depinstalled), (_('Updated'), self.tsInfo.updated), (_('Dependency Updated'), self.tsInfo.depupdated), (_('Skipped (dependency problems)'), self.skipped_packages), (_('Replaced'), self.tsInfo.obsoleted), (_('Failed'), self.tsInfo.failed)]: msgs = [] if len(pkglist) > 0: out += '\n%s:\n' % action for txmbr in pkglist: (n,a,e,v,r) = txmbr.pkgtup msg = "%s.%s %s:%s-%s" % (n,a,e,v,r) msgs.append(msg) for num in (8, 7, 6, 5, 4, 3, 2): cols = _fits_in_cols(msgs, num) if cols: break if not cols: cols = [-(self.term.columns - 2)] while msgs: current_msgs = msgs[:len(cols)] out += ' ' out += self.fmtColumns(zip(current_msgs, cols), end=u'\n') msgs = msgs[len(cols):] return out def setupProgressCallbacks(self): """sets up the progress callbacks and various output bars based on debug level""" # if we're below 2 on the debug level we don't need to be outputting # progress bars - this is hacky - I'm open to other options # One of these is a download if self.conf.debuglevel < 2 or not sys.stdout.isatty(): self.repos.setProgressBar(None) self.repos.callback = None else: self.repos.setProgressBar(YumTextMeter(fo=sys.stdout)) self.repos.callback = CacheProgressCallback() # setup our failure report for failover freport = (self.failureReport,(),{}) self.repos.setFailureCallback(freport) # setup callback for CTRL-C's self.repos.setInterruptCallback(self.interrupt_callback) # setup our depsolve progress callback dscb = DepSolveProgressCallBack() self.dsCallback = dscb def setupProgessCallbacks(self): # api purposes only to protect the typo self.setupProgressCallbacks() def setupKeyImportCallbacks(self): self.repos.confirm_func = self._cli_confirm_gpg_key_import self.repos.gpg_import_func = self.getKeyForRepo def interrupt_callback(self, cbobj): '''Handle CTRL-C's during downloads If a CTRL-C occurs a URLGrabError will be raised to push the download onto the next mirror. If two CTRL-C's occur in quick succession then yum will exit. @param cbobj: urlgrabber callback obj ''' delta_exit_chk = 2.0 # Delta between C-c's so we treat as exit delta_exit_str = _("two") # Human readable version of above now = time.time() if not self._last_interrupt: hibeg = self.term.MODE['bold'] hiend = self.term.MODE['normal'] msg = _(""" Current download cancelled, %sinterrupt (ctrl-c) again%s within %s%s%s seconds to exit. """) % (hibeg, hiend, hibeg, delta_exit_str, hiend) self.verbose_logger.log(logginglevels.INFO_2, msg) elif now - self._last_interrupt < delta_exit_chk: # Two quick CTRL-C's, quit raise KeyboardInterrupt # Go to next mirror self._last_interrupt = now raise URLGrabError(15, _('user interrupt')) def download_callback_total_cb(self, remote_pkgs, remote_size, download_start_timestamp): if len(remote_pkgs) <= 1: return if not hasattr(urlgrabber.progress, 'TerminalLine'): return tl = urlgrabber.progress.TerminalLine(8) self.verbose_logger.log(logginglevels.INFO_2, "-" * tl.rest()) dl_time = time.time() - download_start_timestamp if dl_time <= 0: # This stops divide by zero, among other problems dl_time = 0.01 ui_size = tl.add(' | %5sB' % self.format_number(remote_size)) ui_time = tl.add(' %9s' % self.format_time(dl_time)) ui_end = tl.add(' ' * 5) ui_bs = tl.add(' %5sB/s' % self.format_number(remote_size / dl_time)) msg = "%s%s%s%s%s" % (utf8_width_fill(_("Total"), tl.rest(), tl.rest()), ui_bs, ui_size, ui_time, ui_end) self.verbose_logger.log(logginglevels.INFO_2, msg) class DepSolveProgressCallBack: """provides text output callback functions for Dependency Solver callback""" def __init__(self): """requires yum-cli log and errorlog functions as arguments""" self.verbose_logger = logging.getLogger("yum.verbose.cli") self.loops = 0 def pkgAdded(self, pkgtup, mode): modedict = { 'i': _('installed'), 'u': _('updated'), 'o': _('obsoleted'), 'e': _('erased')} (n, a, e, v, r) = pkgtup modeterm = modedict[mode] self.verbose_logger.log(logginglevels.INFO_2, _('---> Package %s.%s %s:%s-%s set to be %s'), n, a, e, v, r, modeterm) def start(self): self.loops += 1 def tscheck(self): self.verbose_logger.log(logginglevels.INFO_2, _('--> Running transaction check')) def restartLoop(self): self.loops += 1 self.verbose_logger.log(logginglevels.INFO_2, _('--> Restarting Dependency Resolution with new changes.')) self.verbose_logger.debug('---> Loop Number: %d', self.loops) def end(self): self.verbose_logger.log(logginglevels.INFO_2, _('--> Finished Dependency Resolution')) def procReq(self, name, formatted_req): self.verbose_logger.log(logginglevels.INFO_2, _('--> Processing Dependency: %s for package: %s'), formatted_req, name) def unresolved(self, msg): self.verbose_logger.log(logginglevels.INFO_2, _('--> Unresolved Dependency: %s'), msg) def procConflict(self, name, confname): self.verbose_logger.log(logginglevels.INFO_2, _('--> Processing Conflict: %s conflicts %s'), name, confname) def transactionPopulation(self): self.verbose_logger.log(logginglevels.INFO_2, _('--> Populating transaction set ' 'with selected packages. Please wait.')) def downloadHeader(self, name): self.verbose_logger.log(logginglevels.INFO_2, _('---> Downloading header for %s ' 'to pack into transaction set.'), name) class CacheProgressCallback: ''' The class handles text output callbacks during metadata cache updates. ''' def __init__(self): self.logger = logging.getLogger("yum.cli") self.verbose_logger = logging.getLogger("yum.verbose.cli") self.file_logger = logging.getLogger("yum.filelogging.cli") def log(self, level, message): self.verbose_logger.log(level, message) def errorlog(self, level, message): self.logger.log(level, message) def filelog(self, level, message): self.file_logger.log(level, message) def progressbar(self, current, total, name=None): progressbar(current, total, name) class YumCliRPMCallBack(RPMBaseCallback): """ Yum specific callback class for RPM operations. """ width = property(lambda x: _term_width()) def __init__(self): RPMBaseCallback.__init__(self) self.lastmsg = to_unicode("") self.lastpackage = None # name of last package we looked at self.output = logging.getLogger("yum.verbose.cli").isEnabledFor(logginglevels.INFO_2) # for a progress bar self.mark = "#" self.marks = 22 def event(self, package, action, te_current, te_total, ts_current, ts_total): # this is where a progress bar would be called process = self.action[action] if type(package) not in types.StringTypes: pkgname = package.name else: pkgname = package self.lastpackage = package if te_total == 0: percent = 0 else: percent = (te_current*100L)/te_total if self.output and (sys.stdout.isatty() or te_current == te_total): (fmt, wid1, wid2) = self._makefmt(percent, ts_current, ts_total, pkgname=pkgname) msg = fmt % (utf8_width_fill(process, wid1, wid1), utf8_width_fill(pkgname, wid2, wid2)) if msg != self.lastmsg: sys.stdout.write(to_unicode(msg)) sys.stdout.flush() self.lastmsg = msg if te_current == te_total: print " " def scriptout(self, package, msgs): if msgs: sys.stdout.write(to_unicode(msgs)) sys.stdout.flush() def _makefmt(self, percent, ts_current, ts_total, progress = True, pkgname=None): l = len(str(ts_total)) size = "%s.%s" % (l, l) fmt_done = "%" + size + "s/%" + size + "s" done = fmt_done % (ts_current, ts_total) # This should probably use TerminLine, but we don't want to dep. on # that. So we kind do an ok job by hand ... at least it's dynamic now. if pkgname is None: pnl = 22 else: pnl = utf8_width(pkgname) overhead = (2 * l) + 2 # Length of done, above overhead += 19 # Length of begining overhead += 1 # Space between pn and done overhead += 2 # Ends for progress overhead += 1 # Space for end width = self.width if width < overhead: width = overhead # Give up width -= overhead if pnl > width / 2: pnl = width / 2 marks = self.width - (overhead + pnl) width = "%s.%s" % (marks, marks) fmt_bar = "[%-" + width + "s]" # pnl = str(28 + marks + 1) full_pnl = pnl + marks + 1 if progress and percent == 100: # Don't chop pkg name on 100% fmt = "\r %s: %s " + done wid2 = full_pnl elif progress: bar = fmt_bar % (self.mark * int(marks * (percent / 100.0)), ) fmt = "\r %s: %s " + bar + " " + done wid2 = pnl elif percent == 100: fmt = " %s: %s " + done wid2 = full_pnl else: bar = fmt_bar % (self.mark * marks, ) fmt = " %s: %s " + bar + " " + done wid2 = pnl return fmt, 15, wid2 def progressbar(current, total, name=None): """simple progress bar 50 # marks""" mark = '#' if not sys.stdout.isatty(): return if current == 0: percent = 0 else: if total != 0: percent = float(current) / total else: percent = 0 width = _term_width() if name is None and current == total: name = '-' end = ' %d/%d' % (current, total) width -= len(end) + 1 if width < 0: width = 0 if name is None: width -= 2 if width < 0: width = 0 hashbar = mark * int(width * percent) output = '\r[%-*s]%s' % (width, hashbar, end) elif current == total: # Don't chop name on 100% output = '\r%s%s' % (utf8_width_fill(name, width, width), end) else: width -= 4 if width < 0: width = 0 nwid = width / 2 if nwid > utf8_width(name): nwid = utf8_width(name) width -= nwid hashbar = mark * int(width * percent) output = '\r%s: [%-*s]%s' % (utf8_width_fill(name, nwid, nwid), width, hashbar, end) if current <= total: sys.stdout.write(output) if current == total: sys.stdout.write('\n') sys.stdout.flush() if __name__ == "__main__": if len(sys.argv) > 1 and sys.argv[1] == "progress": print "" print " Doing progress, small name" print "" for i in xrange(0, 101): progressbar(i, 100, "abcd") time.sleep(0.1) print "" print " Doing progress, big name" print "" for i in xrange(0, 101): progressbar(i, 100, "_%s_" % ("123456789 " * 5)) time.sleep(0.1) print "" print " Doing progress, no name" print "" for i in xrange(0, 101): progressbar(i, 100) time.sleep(0.1) if len(sys.argv) > 1 and sys.argv[1] in ("progress", "rpm-progress"): cb = YumCliRPMCallBack() cb.output = True cb.action["foo"] = "abcd" cb.action["bar"] = "_12345678_.end" print "" print " Doing CB, small proc / small pkg" print "" for i in xrange(0, 101): cb.event("spkg", "foo", i, 100, i, 100) time.sleep(0.1) print "" print " Doing CB, big proc / big pkg" print "" for i in xrange(0, 101): cb.event("lpkg" + "-=" * 15 + ".end", "bar", i, 100, i, 100) time.sleep(0.1) if len(sys.argv) > 1 and sys.argv[1] in ("progress", "i18n-progress", "rpm-progress", 'i18n-rpm-progress'): yum.misc.setup_locale() if len(sys.argv) > 1 and sys.argv[1] in ("progress", "i18n-progress"): print "" print " Doing progress, i18n: small name" print "" for i in xrange(0, 101): progressbar(i, 100, to_unicode('\xe6\xad\xa3\xe5\x9c\xa8\xe5\xae\x89\xe8\xa3\x85')) time.sleep(0.1) print "" print "" print " Doing progress, i18n: big name" print "" for i in xrange(0, 101): progressbar(i, 100, to_unicode('\xe6\xad\xa3\xe5\x9c\xa8\xe5\xae\x89\xe8\xa3\x85' * 5 + ".end")) time.sleep(0.1) print "" if len(sys.argv) > 1 and sys.argv[1] in ("progress", "i18n-progress", "rpm-progress", 'i18n-rpm-progress'): cb = YumCliRPMCallBack() cb.output = True cb.action["foo"] = to_unicode('\xe6\xad\xa3\xe5\x9c\xa8\xe5\xae\x89\xe8\xa3\x85') cb.action["bar"] = cb.action["foo"] * 5 + ".end" print "" print " Doing CB, i18n: small proc / small pkg" print "" for i in xrange(0, 101): cb.event("spkg", "foo", i, 100, i, 100) time.sleep(0.1) print "" print " Doing CB, i18n: big proc / big pkg" print "" for i in xrange(0, 101): cb.event("lpkg" + "-=" * 15 + ".end", "bar", i, 100, i, 100) time.sleep(0.1) print ""