import rpm def dEBUG(str): print str def addNewPackageToUpgSet(pkgDict, pkg): """Check to see if there's already a pkg by the name of pkg already in our dictionary. If not, add this one. If there is, see if this one is 'newer' or has a 'better' arch.""" name = pkg[rpm.RPMTAG_NAME] if not pkgDict.has_key(name): # nope pkgDict[name] = pkg else: # first check version val = rpm.versionCompare(pkgDict[name], pkg) if val < 0: # we're newer, add this one pkgDict[name] = pkg elif val == 0: # same version, so check the architecture newscore = rpm.archscore(pkg[rpm.RPMTAG_ARCH]) oldscore = pkgDict[name][rpm.RPMTAG_ARCH] if newscore and newscore < oldscore: # if the score is less, we're "better" pkgDict[name] = pkg def findpackageset(hdrlist, dbPath='/'): ts = rpm.TransactionSet(dbPath) ts.setVSFlags(rpm.RPMVSF_NORSA|rpm.RPMVSF_NODSA) ts.setFlags(rpm.RPMTRANS_FLAG_NOMD5) pkgDict = {} # go through and figure out which packages in the header list are # actually applicable for our architecture pkgDict = {} for h in hdrlist: score1 = rpm.archscore(h[rpm.RPMTAG_ARCH]) if (score1): name = h[rpm.RPMTAG_NAME] if pkgDict.has_key(name): score2 = rpm.archscore(pkgDict[name][rpm.RPMTAG_ARCH]) if (score1 < score2): pkgDict[name] = h else: pkgDict[name] = h hdlist = pkgDict.values() pkgDict = {} # loop through packages and find ones which are a newer # version than what we have for pkg in hdlist: mi = ts.dbMatch('name', pkg[rpm.RPMTAG_NAME]) for h in mi: val = rpm.versionCompare(h, pkg) if (val > 0): # dEBUG("found older version of %(name)s" % h) pass elif (val < 0): # dEBUG("found newer version of %(name)s" % h) # check if we already have this package in our dictionary addNewPackageToUpgSet(pkgDict, pkg) else: # dEBUG("found same verison of %(name)s" % h) pass # handle obsoletes for pkg in hdlist: if pkg[rpm.RPMTAG_NAME] in pkgDict.keys(): # dEBUG("%(name)s is already selected" % pkg) continue if pkg[rpm.RPMTAG_OBSOLETENAME] is not None: for obs in pkg[rpm.RPMTAG_OBSOLETENAME]: mi = ts.dbMatch('name', obs) # FIXME: I should really iterate over all matches and verify # versioned obsoletes, but nothing in Red Hat Linux uses # them, so I'll optimize for h in mi: # dEBUG("adding %(name)s to the upgrade set for obsoletes" % pkg) addNewPackageToUpgSet(pkgDict, pkg) break return pkgDict.values() if __name__ == "__main__": import sys, os if len(sys.argv) < 2: print "Usage: %s /path/to/tree [rootpath]" %(sys.argv[0],) sys.exit(0) tree = sys.argv[1] if len(sys.argv) >= 3: instPath = sys.argv[2] else: instPath = "/" fd = os.open("%s/RedHat/base/hdlist" %(tree,), os.O_RDONLY) hdlist = rpm.readHeaderListFromFD(fd) os.close(fd) packages = findpackageset(hdlist, instPath) for pkg in packages: print pkg[rpm.RPMTAG_NAME]