132 lines
4.3 KiB
Python
Executable File
132 lines
4.3 KiB
Python
Executable File
#!/usr/bin/python -tt
|
|
# 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.
|
|
|
|
import sys
|
|
import yum
|
|
|
|
class Gather(yum.YumBase):
|
|
def __init__(self, opts):
|
|
yum.YumBase.__init__(self)
|
|
self.logger = logging.getLogger("yum.verbose.fist")
|
|
self.opts = opts
|
|
|
|
def getPackageObjects(self)
|
|
|
|
def findDeps(self, po):
|
|
"""Return the dependencies for a given package, as well
|
|
possible solutions for those dependencies.
|
|
|
|
Returns the deps as a dict of:
|
|
dict[reqs] = [list of satisfying pkgs]"""
|
|
|
|
|
|
reqs = po.returnPrco('requires');
|
|
reqs.sort()
|
|
pkgresults = {}
|
|
|
|
for req in reqs:
|
|
(r,f,v) = req
|
|
if r.startswith('rpmlib('):
|
|
continue
|
|
|
|
pkgresults[req] = list(self.whatProvides(r, f, v))
|
|
|
|
return pkgresults
|
|
|
|
def downloadPackages(self, polist):
|
|
"""Cycle through the list of package objects and
|
|
download them from their respective repos."""
|
|
|
|
|
|
for pkg in polist:
|
|
repo = self.repos.getRepo(pkg.repoid)
|
|
remote = pkg.returnSimple('relativepath')
|
|
local = os.path.basename(remote)
|
|
local = os.path.join(opts.destdir, local)
|
|
if (os.path.exists(local) and
|
|
str(os.path.getsize(local)) == pkg.returnSimple('packagesize')):
|
|
|
|
if not opts.quiet:
|
|
self.logger.info("%s already exists and appears to be complete" % local)
|
|
continue
|
|
|
|
# Disable cache otherwise things won't download
|
|
repo.cache = 0
|
|
if not opts.quiet:
|
|
self.logger.info('Downloading %s' % os.path.basename(remote))
|
|
pkg.localpath = local # Hack: to set the localpath to what we want.
|
|
repo.getPackage(pkg)
|
|
|
|
|
|
def create_yumobj(yumconf):
|
|
# Create a yum object to act upon. This may move to a higher
|
|
# level file and get passed to this module.
|
|
myYum = yum.yumBase()
|
|
myYum.doConfigSetup(fn=yumconf)
|
|
myYum.doRepoSetup()
|
|
myYum.doSackSetup()
|
|
return myYum
|
|
|
|
def download_packages(yumobj, pkglist):
|
|
# for now a simple function to download packages.
|
|
# Needed are ways to define where to put the packages,
|
|
# cleaning up multiple returns from searching
|
|
pkgobjs = []
|
|
for pkg in pkglist:
|
|
pkgobjs.extend(yumobj.pkgSack.searchNevra(name=pkg))
|
|
for pkgobj in pkgobjs:
|
|
pkgobj.repo.getPackage(pkgobj)
|
|
|
|
def main():
|
|
# This is used for testing the module
|
|
(opts, args) = get_arguments()
|
|
try:
|
|
compsobj = yum.comps.Comps()
|
|
compsobj.add(opts.comps)
|
|
|
|
print get_packagelist(compsobj)
|
|
|
|
except IOError:
|
|
print >> sys.stderr, "gather.py: No such file:\'%s\'" % opts.comps
|
|
sys.exit(1)
|
|
|
|
if __name__ == '__main__':
|
|
from optparse import OptionParser
|
|
def get_arguments():
|
|
# hack job for now, I'm sure this could be better for our uses
|
|
usage = "usage: %s [options]" % sys.argv[0]
|
|
parser = OptionParser(usage=usage)
|
|
parser.add_option("--destdir", default=".", dest="destdir",
|
|
help='destination directory (defaults to current directory)')
|
|
parser.add_option("--comps", default="comps.xml", dest="comps",
|
|
help='comps file to use')
|
|
parser.add_option("--yumconf", default="yum.conf", dest="yumconf",
|
|
help='yum config file to use')
|
|
|
|
|
|
(opts, args) = parser.parse_args()
|
|
#if len(opts) < 1:
|
|
# parser.print_help()
|
|
# sys.exit(0)
|
|
return (opts, args)
|
|
|
|
def get_packagelist(myComps):
|
|
# Get the list of packages from the comps file
|
|
pkglist = []
|
|
for group in myComps.groups:
|
|
pkglist += group.packages
|
|
return pkglist
|
|
main()
|