yum-debug-restore.py 7.35 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54
#!/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.
## (c) 2008 Red Hat. Written by skvidal@fedoraproject.org
##                                james@fedoraproject.org

import os
import sys
import gzip
import tempfile

from optparse import OptionParser

import yum
import rpmUtils.miscutils

sections = ['%%%%SYSTEM INFO\n', '%%%%YUM INFO\n',
            '%%%%RPMDB PROBLEMS\n', '%%%%RPMDB\n',
            '%%%%REPOS\n']

def cmd_line():
    parser = OptionParser()
    parser.set_usage("yum-debug-restore [options]")
    parser.add_option("-C", "--cache", action="store_true",
                      help="run from cache only")
    parser.add_option("-c", dest="conffile", help="config file location")
    parser.add_option("--enablerepo", action="append", dest="enablerepos",
                      help="specify additional repoids to query, can be specified multiple times")
    parser.add_option("--disablerepo", action="append", dest="disablerepos",
                      help="specify repoids to disable, can be specified multiple times")                      
    parser.add_option("-y", dest="assumeyes", action="store_true",
                      help="answer yes for all questions")
    parser.add_option("--skip-broken", action="store_true",
                      help="skip packages with depsolving problems")

    parser.add_option("--output", action="store_true",
                      help="output the yum shell commands")
    parser.add_option("--shell", 
                      help="output the yum shell commands to a file")

    parser.add_option("--install-latest", action="store_true",
                      help="install the latest instead of specific versions")
55 56
    parser.add_option("--ignore-arch", action="store_true",
                      help="ignore arch of packages, so you can dump on .i386 and restore on .x86_64")
57 58 59 60 61 62

    parser.add_option("--filter-types", 
                      help="Limit to: install, remove, update, downgrade")

    (opts, args) = parser.parse_args()
    if not args:
Seth Vidal's avatar
Seth Vidal committed
63
        parser.print_usage()
64
        sys.exit(1)
65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96
    return (opts, args)

class OtherRpmDB:

    def __init__(self, fn):
        self.pkgtups = []

        if fn.endswith(".gz"):
            fo = gzip.GzipFile(fn)
        else:
            fo = open(fn)

        if fo.readline() != 'yum-debug-dump version 1\n':
            print >>sys.stderr, "Bad yum debug file:", fn
            sys.exit(1)

        skip = sections[:-1]
        for line in fo:
            if skip: # Header stuff
                if line == skip[0]:
                    skip.pop(0)
                continue

            if not line or line[0] != ' ':
                break

            pkgtup = rpmUtils.miscutils.splitFilename(line.strip())
            n,v,r,e,a = pkgtup # grrr...
            pkgtup = (n,a,e,v,r)
            self.pkgtups.append(pkgtup)

def naevr2str(n,a,e,v,r):
97 98
    if a is None: # Assume epoch doesn't change without release changing
        return "%s-%s-%s" % (n,v,r)
99 100 101 102 103 104 105
    if e in (None, '', '0'):
        return "%s-%s-%s.%s" % (n,v,r,a)
    return "%s-%s:%s-%s.%s" % (n,e,v,r,a)
def pkgtup2str(pkgtup):
    n,a,e,v,r = pkgtup
    return naevr2str(n,a,e,v,r)

106
def pkg_data2list(yb, opkgtups, opkgmaps, install_latest, ignore_arch):
107 108 109 110
    ret = []
    npkgtups = set()
    npkgmaps = {}
    for po in sorted(yb.rpmdb.returnPackages()):
111 112 113
        arch = po.arch
        if ignore_arch:
            arch = None
114
        if False: pass
115
        elif (po.name, arch) not in opkgmaps:
116 117
            ret.append(("remove", str(po)))
        elif po.pkgtup not in opkgtups:
118
            n,a,e,v,r = opkgmaps[(po.name, arch)]
119
            pinstEVR = yum.packages.PackageEVR(e, v, r)
120 121 122 123
            if po.EVR == pinstEVR:
                assert ignore_arch and po.arch != a
            elif po.EVR < pinstEVR:
                ret.append(("upgrade",   naevr2str(n,arch,e,v,r)))
124
            else:
125
                ret.append(("downgrade", naevr2str(n,arch,e,v,r)))
126 127
        npkgtups.add(po.pkgtup)
        npkgmaps[(po.name, po.arch)] = po
128 129
        if ignore_arch:
            npkgmaps[(po.name, None)] = po
130 131

    for name, arch in sorted(opkgmaps):
132 133
        if ignore_arch and arch is not None:
            continue
134 135
        if (name, arch) in npkgmaps:
            continue
136 137 138
        if install_latest and ignore_arch:
            ret.append(("install", name))
        elif install_latest:
139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178
            ret.append(("install", "%s.%s" % (name, arch)))
        else:
            ret.append(("install", pkgtup2str(opkgmaps[(name, arch)])))
    return ret

def main():
    (opts, args) = cmd_line()
    yb = yum.YumBase()
    yb.preconf.init_plugins = True
    if opts.conffile:
        yb.preconf.fn = opts.conffile

    yb.conf
    if opts.cache:
        yb.conf.cache = True

    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()

    xtra_args = []
    if opts.skip_broken:
        xtra_args.append('--skip-broken')

    if opts.assumeyes:
        xtra_args.append('-y')

    fn = args[0]
    print "Reading from: %s" % fn
    orpmdb = OtherRpmDB(fn)

    opkgmaps = {}
    for pkgtup in orpmdb.pkgtups:
        opkgmaps[(pkgtup[0], pkgtup[1])] = pkgtup
179 180 181
        if opts.ignore_arch:
            n,a,e,v,r = pkgtup
            opkgmaps[(pkgtup[0], None)] = n,None,e,v,r
182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199

    if opts.output:
        fo = sys.stdout
    elif opts.shell:
        try:
            fo = open(opts.shell, "wb")
        except OSError, e:
            print >>sys.stderr, "open(%s): %s" % (opts.shell, e)
            sys.exit(1)
    else:
        fo = tempfile.NamedTemporaryFile()

    fT = None
    if opts.filter_types:
        fT = set(opts.filter_types.replace(",", " ").split())

    counts = {}
    for T, pkg in pkg_data2list(yb, set(orpmdb.pkgtups), opkgmaps,
200
                                opts.install_latest, opts.ignore_arch):
201 202 203
        if fT is not None and T not in fT:
            continue
        counts[T] = counts.get(T, 0) + 1
204 205 206 207 208 209
        try:
            print >>fo, "%-9s %s" % (T, pkg)
        except IOError:
            if opts.output: # mainly due to |
                sys.exit(1)
            raise
210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234

    if opts.output:
        sys.exit(0)

    print >>fo, "run"
    fo.flush()

    if opts.shell:
        if counts:
            print "Statistics:"
        for T in sorted(counts):
            print "    %9s %6u" % (T, counts[T])
        print "Done"
        sys.exit(0)

    # Want to do the transaction, hacky method
    if xtra_args:
        os.system("yum shell %s %s" % (" ".join(xtra_args), fo.name))
    else:
        os.system("yum shell %s" % fo.name)

if __name__ == "__main__":
    main()