#!/usr/bin/python -tt
import os
import sys
import re
import string
import optparse
import gzip
import yum
from yum.i18n import to_unicode
import yum.comps
sys.path.insert(0, '/usr/share/yum-cli')
import output
from urlgrabber.progress import TextMeter
def setup_opts():
version = "0.0.1"
vers_txt = "Manage yum groups metadata version %s" % version
usage_txt = "%prog [pkg-wildcard]..."
parser = optparse.OptionParser(usage = usage_txt, version = vers_txt)
parser.add_option("-n", "--name", help="group name")
parser.add_option("--id", help="group id")
parser.add_option("--mandatory", action="store_true",
help="make the package names be in the mandatory section")
parser.add_option("--optional", action="store_true",
help="make the package names be in the optional section")
parser.add_option("--dependencies", action="store_true",
help="add the dependencies for this package")
parser.add_option("--user-visible", dest="user_visible",
action="store_true", default=None,
help="make this a user visible group (default)")
parser.add_option("--not-user-visible", dest="user_visible",
action="store_false", default=None,
help="make this a non-user visible group")
parser.add_option("--description", help="description for the group")
parser.add_option("--display-order", help="sort order override")
parser.add_option("--load", action="append", default=[],
help="load groups metadata from file")
parser.add_option("--save", action="append", default=[],
help="save groups metadata to file (don't print)")
parser.add_option("--merge",
help="load and save groups metadata to file (don't print)")
parser.add_option("--print", dest="print2stdout",
action="store_true", default=None,
help="print the result to stdout")
parser.add_option("--remove", action="store_true", default=False,
help="remove the listed package instead of adding them")
parser.add_option("--translated-name", action="append",
dest="i18nname",
help="name for the group, translated")
parser.add_option("--translated-description", action="append",
dest="i18ndescription",
help="description for the group, translated")
# Generic options
parser.add_option("--quiet", action="store_true",
help="quiet (no output to stderr)", default=False)
parser.add_option("--verbose", action="store_false",
help="verbose output", dest="quiet")
parser.add_option("--enablerepo", action="append", dest="enablerepos",
help="specify repoids to query, can be specified multiple times (default is all enabled)")
parser.add_option("--disablerepo", action="append", dest="disablerepos",
help="specify repoids to disable, can be specified multiple times")
# tmprepo etc.
parser.add_option("--noplugins", action="store_false", default=True,
dest="plugins",
help="disable yum plugin support")
parser.add_option("-C", "--cache", action="store_true",
help="run from cache only")
parser.add_option("--tempcache", action="store_true",
help="use private cache (default when used as non-root)")
parser.add_option("-c", "--config",
dest="conffile", help="config file location")
return parser
def trans_data(yb, inp):
data = inp.split(':', 2)
if len(data) != 2:
yb.logger.error("Error: Incorrect translated data, should be: 'lang:text'")
sys.exit(50)
lang, text = data
alnum = string.ascii_letters + string.digits
lang = re.sub('[^-' + alnum + '_.@]', '', lang)
if not lang:
yb.logger.error("Error: Incorrect/empty language for translated data")
sys.exit(50)
return lang, to_unicode(text)
__req2pkgs = {}
def req2pkgs(yb, req):
global __req2pkgs
req = str(req)
if req in __req2pkgs:
return __req2pkgs[req]
providers = []
try:
# XXX rhbz#246519, for some reason returnPackagesByDep() fails
# to find some root level directories while
# searchPackageProvides() does... use that for now
matches = yb.searchPackageProvides([req])
providers = matches.keys()
# provider.extend(yum.YumBase.returnPackagesByDep(self, depstring))
except yum.Errors.YumBaseError, err:
yb.logger.error("No package provides %s" % req)
return []
__req2pkgs[req] = providers
return providers
def txt2id(txt):
groupid = txt.lower()
alnum = string.ascii_lowercase + string.digits
groupid = re.sub('[^-' + alnum + '_.:]', '', groupid)
return groupid
def main():
parser = setup_opts()
(opts, args) = parser.parse_args()
comps = yum.comps.Comps()
# Borrowing large sections from repoquery/pkg-tree etc.
initnoise = (not opts.quiet) * 2
yb = yum.YumBase()
if opts.conffile is not None:
yb.preconf.fn = opts.conffile
yb.preconf.debuglevel = initnoise
yb.preconf.init_plugins = opts.plugins
yb.conf
# Show what is going on, if --quiet is not set.
if not opts.quiet and sys.stdout.isatty():
yb.repos.setProgressBar(TextMeter(fo=sys.stdout))
yb.repos.callback = output.CacheProgressCallback()
yumout = output.YumOutput()
freport = ( yumout.failureReport, (), {} )
yb.repos.setFailureCallback( freport )
if os.geteuid() != 0 or opts.tempcache:
cachedir = yum.misc.getCacheDir()
if cachedir is None:
yb.logger.error("Error: Could not make cachedir, exiting")
sys.exit(50)
yb.repos.setCacheDir(cachedir)
yb.conf.cache = 0 # yum set cache=1, if uid != 0
if opts.cache:
yb.conf.cache = True
if not opts.quiet:
yb.logger.info('Running from cache, results might be incomplete.')
if opts.disablerepos:
for repo_match in opts.disablerepos:
for repo in yb.repos.findRepos(repo_match):
repo.disable()
if opts.enablerepos:
for repo_match in opts.enablerepos:
for repo in yb.repos.findRepos(repo_match):
repo.enable()
try:
yb.doRepoSetup()
except yum.Errors.RepoError, e:
yb.logger.error("Could not setup repo: %s" % (e))
sys.exit(50)
archlist = None
try:
yb.doSackSetup(archlist=archlist)
except yum.Errors.RepoError, e:
yb.logger.error(e)
sys.exit(1)
yum.misc.setup_locale()
if opts.merge:
opts.load.insert(0, opts.merge)
opts.save.append(opts.merge)
loaded_files = False
for fname in opts.load:
try:
if not os.path.exists(fname):
yb.logger.error("File not found: %s" % fname)
continue
if fname.endswith('.gz'):
fname = gzip.open(fname)
comps.add(srcfile=fname)
loaded_files = True
except IOError, e:
yb.logger.error(e)
sys.exit(50)
if not loaded_files and opts.remove:
yb.logger.error("Can't remove package(s) when we haven't loaded any")
sys.exit(50)
group = None
if opts.id:
group = comps.return_group(opts.id)
if group is None and opts.name:
group = comps.return_group(opts.name)
if group is None and opts.remove:
yb.logger.error("Can't remove package(s) from non-existent group")
sys.exit(50)
if group is None:
group = yum.comps.Group()
if opts.id:
groupid = txt2id(opts.id)
if not groupid:
yb.logger.error("No valid id for group")
sys.exit(50)
group.groupid = groupid
group.name = groupid
elif opts.name:
group.groupid = txt2id(opts.name)
if not group.groupid:
yb.logger.error("No valid id for group")
sys.exit(50)
else:
yb.logger.error("No name or id for group")
sys.exit(50)
comps.add_group(group)
if opts.name:
if ',' in opts.name:
yb.logger.error("Group name has a comma in it")
if '*' in opts.name or '?' in opts.name:
yb.logger.error("Group name has a wildcard in it, ? or *")
group.name = opts.name
if opts.description:
group.description = opts.description
if opts.display_order:
group.display_order = int(opts.display_order)
if opts.user_visible is not None:
group.user_visible = opts.user_visible
for tn in opts.i18nname or []:
lang, text = trans_data(yb, tn)
if ',' in text:
yb.logger.error("Translated group name (%s) has a comma in it"%lang)
if '*' in text or '?' in text:
yb.logger.error("Translated group name (%s) has a wildcard in it"
", ? or *" % lang)
group.translated_name[lang] = text
for td in opts.i18ndescription or []:
lang, text = trans_data(yb, td)
group.translated_description[lang] = text
try:
if args:
pkgs = yb.pkgSack.returnNewestByName(patterns=args)
else:
pkgs = []
except yum.packageSack.PackageSackError, e:
yb.logger.error(e)
sys.exit(50)
pkgnames = set([pkg.name for pkg in pkgs])
if opts.dependencies:
for pkg in pkgs:
for rptup in pkg.returnPrco('requires'):
if rptup[0].startswith('rpmlib'):
continue
rname = yum.misc.prco_tuple_to_string(rptup)
pkgnames.update([pkg.name for pkg in req2pkgs(yb, rname)])
for pkgname in pkgnames:
if opts.remove:
group.mandatory_packages.pop(pkgname, 0)
group.optional_packages.pop(pkgname, 0)
group.default_packages.pop(pkgname, 0)
elif opts.mandatory:
group.mandatory_packages[pkgname] = 1
elif opts.optional:
group.optional_packages[pkgname] = 1
else:
group.default_packages[pkgname] = 1
for fname in opts.save:
try:
fo = open(fname, "wb")
fo.write(comps.xml())
del fo
except IOError, e:
yb.logger.error(e)
sys.exit(50)
if (opts.print2stdout or (opts.print2stdout is None and not opts.save)):
# Why the to_unicode()? Why is it converting at all?
# Why doesn't setup_locale() fix this? ... all good questions
print to_unicode(comps.xml())
if __name__ == "__main__":
main()
Mini Shell