yumdownloader.py 12.2 KB
Newer Older
1
#!/usr/bin/python 
gijs's avatar
gijs committed
2 3 4 5 6 7 8 9 10 11 12 13 14 15
# 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.

Panu Matilainen's avatar
Panu Matilainen committed
16
import sys
17
sys.path.insert(0,'/usr/share/yum-cli')
Panu Matilainen's avatar
Panu Matilainen committed
18

gijs's avatar
gijs committed
19
import yum
20
from yum.misc import getCacheDir
Panu Matilainen's avatar
Panu Matilainen committed
21

22 23
from cli import *
from utils import YumUtilBase
gijs's avatar
gijs committed
24

25
from urlparse import urljoin
26 27 28
from urlgrabber.progress import TextMeter
import shutil

29
import rpmUtils
30

31 32 33 34 35 36 37 38 39 40 41 42 43 44 45
class YumDownloader(YumUtilBase):
    NAME = 'yumdownloader'
    VERSION = '1.0'
    USAGE = '"usage: yumdownloader [options] package1 [package2] [package..]'
    
    def __init__(self):
        YumUtilBase.__init__(self,
                             YumDownloader.NAME,
                             YumDownloader.VERSION,
                             YumDownloader.USAGE)
        self.logger = logging.getLogger("yum.verbose.cli.yumdownloader")                             
        self.main()

    def main(self):
        # Add util commandline options to the yum-cli ones
46
        self.optparser = self.getOptionParser() 
47
        # Add command line option specific to yumdownloader
48
        self.addCmdOptions()
49
        # Parse the commandline option and setup the basics.
50 51 52
        try:
            opts = self.doUtilConfigSetup()
        except yum.Errors.RepoError, e:
53
            self.logger.error(str(e))
54 55
            sys.exit(50)
                
56 57
        # Check if there is anything to do.
        if len(self.cmds) < 1: 
58
            self.optparser.print_help()
59 60 61 62 63 64 65 66 67
            sys.exit(0)

        # make yumdownloader work as non root user.
        if self.conf.uid != 0:
            cachedir = getCacheDir()
            if cachedir is None:
                self.logger.error("Error: Could not make cachedir, exiting")
                sys.exit(50)
            self.repos.setCacheDir(cachedir)
68

69 70 71 72 73
            # Turn off cache
            self.conf.cache = 0
            # make sure the repos know about it, too
            self.repos.setCache(0)
            
74
        # Setup yum (Ts, RPM db, Repo & Sack)
75
        self.doUtilYumSetup(opts)
76 77 78
        # Setup source repos
        if opts.source:
            self.setupSourceRepos()
79 80
        # Do the real action
        self.downloadPackages(opts)
81
        
82 83 84
    def setupSourceRepos(self):
        # enable the -source repos for enabled primary repos
        archlist = rpmUtils.arch.getArchList() + ['src']    
85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100
        # Ok, we have src and bin repos. What we want to do here is:
        #
        # 1. _enable_ source repos for which the bin repos are enabled.
        # 2. _disable_ the _other_ src repos.
        #
        # ...also we don't want to disable the src repos. for #1 and then
        # re-enable them as then we get annoying messages and call .close() on
        # them losing the primarydb data etc.

        # Get all src repos.
        src_repos = {}
        for repo in self.repos.findRepos('*-source'):
            src_repos[repo.id] = False

        #  Find the enabled bin repos, and mark their respective *-source repo.
        # as good.
101
        for repo in self.repos.listEnabled():
102
            if repo.id not in src_repos:
103
                srcrepo = '%s-source' % repo.id
104 105 106 107 108 109
                if srcrepo in src_repos:
                    src_repos[srcrepo] = True

        # Toggle src repos that are set the wrong way
        for repo in self.repos.findRepos('*-source'):
            if     repo.isEnabled() and not src_repos[repo.id]:
110 111
                repo.close()
                self.repos.disableRepo(repo.id)
112 113 114
            if not repo.isEnabled() and     src_repos[repo.id]:
                self.logger.info('Enabling %s repository' % repo.id)
                repo.enable()
115
                # Setup the repo, without a cache
116
                repo.setup(0)
117
                # Setup pkgSack with 'src' in the archlist
118
                self._getSacks(archlist=archlist, thisrepo=repo.id)
119 120
        
        
121
    def downloadPackages(self,opts):
122
        
123 124 125 126 127
        toDownload = []
    
        packages = self.cmds
        for pkg in packages:
            toActOn = []
128
            exactmatch, matched, unmatched = parsePackages(self.pkgSack.returnPackages(), [pkg])
129 130 131
            installable = yum.misc.unique(exactmatch + matched)
            if len(unmatched) > 0: # if we get back anything in unmatched, it fails
                self.logger.error('No Match for argument %s' % pkg)
gijs's avatar
gijs committed
132
                continue
133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152
            for newpkg in installable:
                # This is to fix Bug 469
                # If there are matches to the package argument given but there
                # are no source packages, this can be caused because the
                # source rpm this is built from has a different name
                # for example: nscd is built from the glibc source rpm
                # We find this name by parsing the sourcerpm filename
                # (this is ugly but it appears to work)
                # And finding a package with arch src and the same
                # ver and rel as the binary package
                # That should be the source package
                # Note we do not use the epoch to search as the epoch for the
                # source rpm might be different from the binary rpm (see
                # for example mod_ssl)
                if opts.source and newpkg.arch != 'src':
                    name = newpkg.returnSimple('sourcerpm').rsplit('-',2)[0]
                    src = self.pkgSack.searchNevra(name=name, arch = 'src',
                      ver = newpkg.version,
                      rel = newpkg.release
                    )
153 154 155
                    if src == []:
                        self.logger.error('No source RPM found for %s' % str(newpkg))
                        
156 157 158 159
                    toActOn.extend(src)
                else:
                    toActOn.append(newpkg)
            if toActOn:
160 161 162 163 164 165 166 167 168 169 170
                pkgGroups = self._groupPackages(toActOn)
                for group in pkgGroups:
                    pkgs = pkgGroups[group]
                    if opts.source:
                        toDownload.extend(self.bestPackagesFromList(pkgs, 'src'))
                    elif opts.archlist:
                        for arch in opts.archlist.split(','):
                            toDownload.extend(self.bestPackagesFromList(pkgs, arch))
                    else:
                        toDownload.extend(self.bestPackagesFromList(pkgs))
                            
171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187
        # If the user supplies to --resolve flag, resolve dependencies for
        # all packages
        # note this might require root access because the headers need to be
        # downloaded into the cachedir (is there a way around this)
        if opts.resolve:
            self.doTsSetup()
            self.localPackages = []
            # Act as if we were to install the packages in toDownload
            for po in toDownload:
                self.tsInfo.addInstall(po)
                self.localPackages.append(po)
            # Resolve dependencies
            self.resolveDeps()
            # Add newly added packages to the toDownload list
            for pkg in self.tsInfo.getMembers():
                if not pkg in toDownload:
                    toDownload.append(pkg)
188 189
        if len(toDownload) == 0:
            self.logger.error('Nothing to download')
190
            sys.exit(1)
191 192 193 194 195 196 197 198
            
        for pkg in toDownload:
            n,a,e,v,r = pkg.pkgtup
            packages =  self.pkgSack.searchNevra(n,e,v,r,a)
            for download in packages:
                repo = self.repos.getRepo(download.repoid)
                remote = download.returnSimple('relativepath')
                if opts.urls:
199
                    url = urljoin(repo.urls[0]+'/',remote)
200 201 202
                    self.logger.info('%s' % url)
                    continue
                local = os.path.basename(remote)
203 204
                if not os.path.exists(opts.destdir):
                    os.makedirs(opts.destdir)
205 206
                local = os.path.join(opts.destdir, local)
                if (os.path.exists(local) and 
207
                    os.path.getsize(local) == int(download.returnSimple('packagesize'))):
208 209 210 211 212
                    self.logger.error("%s already exists and appears to be complete" % local)
                    continue
                # Disable cache otherwise things won't download
                repo.cache = 0
                download.localpath = local # Hack: to set the localpath we want.
213 214 215 216 217 218
                try:
                    path = repo.getPackage(download)
                except IOError, e:
                    self.logger.error("Cannot write to file %s. Error was: %s" % (local, e))
                    continue
                    
219 220 221 222 223 224
    
                if not os.path.exists(local) or not os.path.samefile(path, local):
                    progress = TextMeter()
                    progress.start(basename=os.path.basename(local),
                                   size=os.stat(path).st_size)
                    shutil.copy2(path, local)
225
                    progress.end(progress.size) 
226 227 228 229 230 231 232 233 234 235
                    
    def _groupPackages(self,pkglist):
        pkgGroups = {}
        for po in pkglist:
            na = '%s.%s' % (po.name,po.arch)
            if not na in pkgGroups:
                pkgGroups[na] = [po]
            else:
                pkgGroups[na].append(po)
        return pkgGroups
236
            
237 238 239 240 241 242 243 244 245
    # sligly modified from the one in YumUtilBase    
    def doUtilYumSetup(self,opts):
        """do a default setup for all the normal/necessary yum components,
           really just a shorthand for testing"""
        try:
            self._getRepos()
            # if '--source' is used the add src to the archlist
            if opts.source:
                archlist = rpmUtils.arch.getArchList() + ['src']    
246
            elif opts.archlist:
247 248 249
                archlist = []
                for a in opts.archlist.split(','):
                    archlist.extend(rpmUtils.arch.getArchList(a))
250 251 252 253 254 255
            else:
                archlist = rpmUtils.arch.getArchList()
            self._getSacks(archlist=archlist)
        except yum.Errors.YumBaseError, msg:
            self.logger.critical(str(msg))
            sys.exit(1)
256

257 258 259 260 261 262 263
    def _removeEnabledSourceRepos(self):
        ''' Disable all enabled *-source repos.'''
        for repo in self.repos.listEnabled():
            if repo.id.endswith('-source'):
                repo.close()
                self.repos.disableRepo(repo.id)
                srcrepo = repo.id
gijs's avatar
gijs committed
264

265 266
    def addCmdOptions(self):
        self.optparser.add_option("--destdir", default=".", dest="destdir",
267
          help='destination directory (defaults to current directory)')
268
        self.optparser.add_option("--urls", default=False, dest="urls", action="store_true",
269
          help='just list the urls it would download instead of downloading')
270
        self.optparser.add_option("--resolve", default=False, dest="resolve", action="store_true",
271
          help='resolve dependencies and download required packages')
272
        self.optparser.add_option("--source", default=False, dest="source", action="store_true",
273
          help='operate on source packages')
274
        self.optparser.add_option("--archlist",
275
          help="only download packages of certain architecture(s)")        
gijs's avatar
gijs committed
276
if __name__ == '__main__':
277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292
    import locale
    # This test needs to be before locale.getpreferredencoding() as that
    # does setlocale(LC_CTYPE, "")
    try:
        locale.setlocale(locale.LC_ALL, '')
    except locale.Error, e:
        # default to C locale if we get a failure.
        print >> sys.stderr, 'Failed to set locale, defaulting to C'
        os.environ['LC_ALL'] = 'C'
        locale.setlocale(locale.LC_ALL, 'C')
        
    if True: # not sys.stdout.isatty():
        import codecs
        sys.stdout = codecs.getwriter(locale.getpreferredencoding())(sys.stdout)
        sys.stdout.errors = 'replace'

293
    util = YumDownloader()
294 295