dak/check_proposed_updates.py

   1 #!/usr/bin/env python
   2
   3 """ Dependency check proposed-updates """
   4 # Copyright (C) 2001, 2002, 2004, 2006  James Troup <james@nocrew.org>
   5
   6 # This program is free software; you can redistribute it and/or modify
   7 # it under the terms of the GNU General Public License as published by
   8 # the Free Software Foundation; either version 2 of the License, or
   9 # (at your option) any later version.
  10
  11 # This program is distributed in the hope that it will be useful,
  12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 # GNU General Public License for more details.
  15
  16 # You should have received a copy of the GNU General Public License
  17 # along with this program; if not, write to the Free Software
  18 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19
  20 ################################################################################
  21
  22 # | > amd64 is more mature than even some released architectures
  23 # |
  24 # | This might be true of the architecture, unfortunately it seems to be the
  25 # | exact opposite for most of the people involved with it.
  26 #
  27 # <1089213290.24029.6.camel@descent.netsplit.com>
  28
  29 ################################################################################
  30
  31 import pg, sys, os
  32 import apt_pkg, apt_inst
  33 from daklib import database
  34 from daklib import utils
  35 from daklib.regexes import re_no_epoch
  36
  37 ################################################################################
  38
  39 Cnf = None
  40 projectB = None
  41 Options = None
  42 stable = {}
  43 stable_virtual = {}
  44 architectures = None
  45
  46 ################################################################################
  47
  48 def usage (exit_code=0):
  49     print """Usage: dak check-proposed-updates [OPTION] <CHANGES FILE | DEB FILE | ADMIN FILE>[...]
  50 (Very) Basic dependency checking for proposed-updates.
  51
  52   -q, --quiet                be quieter about what is being done
  53   -v, --verbose              be more verbose about what is being done
  54   -h, --help                 show this help and exit
  55
  56 Need either changes files, deb files or an admin.txt file with a '.joey' suffix."""
  57     sys.exit(exit_code)
  58
  59 ################################################################################
  60
  61 def d_test (dict, key, positive, negative):
  62     if not dict:
  63         return negative
  64     if dict.has_key(key):
  65         return positive
  66     else:
  67         return negative
  68
  69 ################################################################################
  70
  71 def check_dep (depends, dep_type, check_archs, filename, files):
  72     pkg_unsat = 0
  73     for arch in check_archs:
  74         for parsed_dep in apt_pkg.ParseDepends(depends):
  75             unsat = []
  76             for atom in parsed_dep:
  77                 (dep, version, constraint) = atom
  78                 # As a real package?
  79                 if stable.has_key(dep):
  80                     if stable[dep].has_key(arch):
  81                         if apt_pkg.CheckDep(stable[dep][arch], constraint, version):
  82                             if Options["debug"]:
  83                                 print "Found %s as a real package." % (utils.pp_deps(parsed_dep))
  84                             unsat = 0
  85                             break
  86                 # As a virtual?
  87                 if stable_virtual.has_key(dep):
  88                     if stable_virtual[dep].has_key(arch):
  89                         if not constraint and not version:
  90                             if Options["debug"]:
  91                                 print "Found %s as a virtual package." % (utils.pp_deps(parsed_dep))
  92                             unsat = 0
  93                             break
  94                 # As part of the same .changes?
  95                 epochless_version = re_no_epoch.sub('', version)
  96                 dep_filename = "%s_%s_%s.deb" % (dep, epochless_version, arch)
  97                 if files.has_key(dep_filename):
  98                     if Options["debug"]:
  99                         print "Found %s in the same upload." % (utils.pp_deps(parsed_dep))
 100                     unsat = 0
 101                     break
 102                 # Not found...
 103                 # [FIXME: must be a better way ... ]
 104                 error = "%s not found. [Real: " % (utils.pp_deps(parsed_dep))
 105                 if stable.has_key(dep):
 106                     if stable[dep].has_key(arch):
 107                         error += "%s:%s:%s" % (dep, arch, stable[dep][arch])
 108                     else:
 109                         error += "%s:-:-" % (dep)
 110                 else:
 111                     error += "-:-:-"
 112                 error += ", Virtual: "
 113                 if stable_virtual.has_key(dep):
 114                     if stable_virtual[dep].has_key(arch):
 115                         error += "%s:%s" % (dep, arch)
 116                     else:
 117                         error += "%s:-"
 118                 else:
 119                     error += "-:-"
 120                 error += ", Upload: "
 121                 if files.has_key(dep_filename):
 122                     error += "yes"
 123                 else:
 124                     error += "no"
 125                 error += "]"
 126                 unsat.append(error)
 127
 128             if unsat:
 129                 sys.stderr.write("MWAAP! %s: '%s' %s can not be satisifed:\n" % (filename, utils.pp_deps(parsed_dep), dep_type))
 130                 for error in unsat:
 131                     sys.stderr.write("  %s\n" % (error))
 132                 pkg_unsat = 1
 133
 134     return pkg_unsat
 135
 136 def check_package(filename, files):
 137     try:
 138         control = apt_pkg.ParseSection(apt_inst.debExtractControl(utils.open_file(filename)))
 139     except:
 140         utils.warn("%s: debExtractControl() raised %s." % (filename, sys.exc_type))
 141         return 1
 142     Depends = control.Find("Depends")
 143     Pre_Depends = control.Find("Pre-Depends")
 144     #Recommends = control.Find("Recommends")
 145     pkg_arch = control.Find("Architecture")
 146     base_file = os.path.basename(filename)
 147     if pkg_arch == "all":
 148         check_archs = architectures
 149     else:
 150         check_archs = [pkg_arch]
 151
 152     pkg_unsat = 0
 153     if Pre_Depends:
 154         pkg_unsat += check_dep(Pre_Depends, "pre-dependency", check_archs, base_file, files)
 155
 156     if Depends:
 157         pkg_unsat += check_dep(Depends, "dependency", check_archs, base_file, files)
 158     #if Recommends:
 159     #pkg_unsat += check_dep(Recommends, "recommendation", check_archs, base_file, files)
 160
 161     return pkg_unsat
 162
 163 ################################################################################
 164
 165 def pass_fail (filename, result):
 166     if not Options["quiet"]:
 167         print "%s:" % (os.path.basename(filename)),
 168         if result:
 169             print "FAIL"
 170         else:
 171             print "ok"
 172
 173 ################################################################################
 174
 175 def check_changes (filename):
 176     try:
 177         changes = utils.parse_changes(filename)
 178         files = utils.build_file_list(changes)
 179     except ChangesUnicodeError:
 180         utils.warn("Improperly encoded changes file, not utf-8")
 181         return
 182     except:
 183         utils.warn("Error parsing changes file '%s'" % (filename))
 184         return
 185
 186     result = 0
 187
 188     # Move to the pool directory
 189     cwd = os.getcwd()
 190     f = files.keys()[0]
 191     pool_dir = Cnf["Dir::Pool"] + '/' + utils.poolify(changes["source"], files[f]["component"])
 192     os.chdir(pool_dir)
 193
 194     changes_result = 0
 195     for f in files.keys():
 196         if f.endswith(".deb"):
 197             result = check_package(f, files)
 198             if Options["verbose"]:
 199                 pass_fail(f, result)
 200             changes_result += result
 201
 202     pass_fail (filename, changes_result)
 203
 204     # Move back
 205     os.chdir(cwd)
 206
 207 ################################################################################
 208
 209 def check_deb (filename):
 210     result = check_package(filename, {})
 211     pass_fail(filename, result)
 212
 213
 214 ################################################################################
 215
 216 def check_joey (filename):
 217     f = utils.open_file(filename)
 218
 219     cwd = os.getcwd()
 220     os.chdir("%s/dists/proposed-updates" % (Cnf["Dir::Root"]))
 221
 222     for line in f.readlines():
 223         line = line.rstrip()
 224         if line.find('install') != -1:
 225             split_line = line.split()
 226             if len(split_line) != 2:
 227                 utils.fubar("Parse error (not exactly 2 elements): %s" % (line))
 228             install_type = split_line[0]
 229             if install_type not in [ "install", "install-u", "sync-install" ]:
 230                 utils.fubar("Unknown install type ('%s') from: %s" % (install_type, line))
 231             changes_filename = split_line[1]
 232             if Options["debug"]:
 233                 print "Processing %s..." % (changes_filename)
 234             check_changes(changes_filename)
 235     f.close()
 236
 237     os.chdir(cwd)
 238
 239 ################################################################################
 240
 241 def parse_packages():
 242     global stable, stable_virtual, architectures
 243
 244     # Parse the Packages files (since it's a sub-second operation on auric)
 245     suite = "stable"
 246     stable = {}
 247     components = Cnf.ValueList("Suite::%s::Components" % (suite))
 248     architectures = filter(utils.real_arch, database.get_suite_architectures(suite))
 249     for component in components:
 250         for architecture in architectures:
 251             filename = "%s/dists/%s/%s/binary-%s/Packages" % (Cnf["Dir::Root"], suite, component, architecture)
 252             packages = utils.open_file(filename, 'r')
 253             Packages = apt_pkg.ParseTagFile(packages)
 254             while Packages.Step():
 255                 package = Packages.Section.Find('Package')
 256                 version = Packages.Section.Find('Version')
 257                 provides = Packages.Section.Find('Provides')
 258                 if not stable.has_key(package):
 259                     stable[package] = {}
 260                 stable[package][architecture] = version
 261                 if provides:
 262                     for virtual_pkg in provides.split(","):
 263                         virtual_pkg = virtual_pkg.strip()
 264                         if not stable_virtual.has_key(virtual_pkg):
 265                             stable_virtual[virtual_pkg] = {}
 266                         stable_virtual[virtual_pkg][architecture] = "NA"
 267             packages.close()
 268
 269 ################################################################################
 270
 271 def main ():
 272     global Cnf, projectB, Options
 273
 274     Cnf = utils.get_conf()
 275
 276     Arguments = [('d', "debug", "Check-Proposed-Updates::Options::Debug"),
 277                  ('q',"quiet","Check-Proposed-Updates::Options::Quiet"),
 278                  ('v',"verbose","Check-Proposed-Updates::Options::Verbose"),
 279                  ('h',"help","Check-Proposed-Updates::Options::Help")]
 280     for i in [ "debug", "quiet", "verbose", "help" ]:
 281         if not Cnf.has_key("Check-Proposed-Updates::Options::%s" % (i)):
 282             Cnf["Check-Proposed-Updates::Options::%s" % (i)] = ""
 283
 284     arguments = apt_pkg.ParseCommandLine(Cnf,Arguments,sys.argv)
 285     Options = Cnf.SubTree("Check-Proposed-Updates::Options")
 286
 287     if Options["Help"]:
 288         usage(0)
 289     if not arguments:
 290         utils.fubar("need at least one package name as an argument.")
 291
 292     projectB = pg.connect(Cnf["DB::Name"], Cnf["DB::Host"], int(Cnf["DB::Port"]))
 293     database.init(Cnf, projectB)
 294
 295     print "Parsing packages files...",
 296     parse_packages()
 297     print "done."
 298
 299     for f in arguments:
 300         if f.endswith(".changes"):
 301             check_changes(f)
 302         elif f.endswith(".deb"):
 303             check_deb(f)
 304         elif f.endswith(".joey"):
 305             check_joey(f)
 306         else:
 307             utils.fubar("Unrecognised file type: '%s'." % (f))
 308
 309 #######################################################################################
 310
 311 if __name__ == '__main__':
 312     main()