dak/override_disparity.py

   1 #!/usr/bin/env python
   2
   3 """
   4 Generate a list of override disparities
   5
   6 @contact: Debian FTP Master <ftpmaster@debian.org>
   7 @copyright: 2010 Luca Falavigna <dktrkranz@debian.org>
   8 @license: GNU General Public License version 2 or later
   9 """
  10
  11 # This program is free software; you can redistribute it and/or modify
  12 # it under the terms of the GNU General Public License as published by
  13 # the Free Software Foundation; either version 2 of the License, or
  14 # (at your option) any later version.
  15
  16 # This program is distributed in the hope that it will be useful,
  17 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  18 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  19 # GNU General Public License for more details.
  20
  21 # You should have received a copy of the GNU General Public License
  22 # along with this program; if not, write to the Free Software
  23 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  24
  25 ################################################################################
  26
  27 # <adsb> Yay bugzilla *sigh*
  28 # <phil> :)
  29 # <Ganneff> quick, replace the bts with it
  30 # * jcristau replaces dak with soyuz
  31 # <adsb> and expects Ganneff to look after it?
  32 # <jcristau> nah, elmo can do that
  33 # * jcristau hides
  34
  35 ################################################################################
  36
  37 import os
  38 import sys
  39 import apt_pkg
  40 import yaml
  41
  42 from daklib.config import Config
  43 from daklib.dbconn import *
  44 from daklib import utils
  45
  46 ################################################################################
  47
  48 def usage (exit_code=0):
  49     print """Generate a list of override disparities
  50
  51        Usage:
  52        dak override-disparity [-f <file>] [ -p <package> ] [ -s <suite> ]
  53
  54 Options:
  55
  56   -h, --help                show this help and exit
  57   -f, --file                store output into given file
  58   -p, --package             limit check on given package only
  59   -s, --suite               choose suite to look for (default: unstable)"""
  60
  61     sys.exit(exit_code)
  62
  63 def main():
  64     cnf = Config()
  65     Arguments = [('h','help','Override-Disparity::Options::Help'),
  66                  ('f','file','Override-Disparity::Options::File','HasArg'),
  67                  ('s','suite','Override-Disparity::Options::Suite','HasArg'),
  68                  ('p','package','Override-Disparity::Options::Package','HasArg')]
  69
  70     for i in ['help', 'package']:
  71         if not cnf.has_key('Override-Disparity::Options::%s' % (i)):
  72             cnf['Override-Disparity::Options::%s' % (i)] = ''
  73     if not cnf.has_key('Override-Disparity::Options::Suite'):
  74         cnf['Override-Disparity::Options::Suite'] = 'unstable'
  75
  76     apt_pkg.ParseCommandLine(cnf.Cnf, Arguments, sys.argv)
  77     Options = cnf.SubTree('Override-Disparity::Options')
  78
  79     if Options['help']:
  80         usage()
  81
  82     depends = {}
  83     session = DBConn().session()
  84     suite = Options['suite']
  85     components = get_component_names(session)
  86     arches = set([x.arch_string for x in get_suite_architectures(suite)])
  87     arches -= set(['source', 'all'])
  88     for arch in arches:
  89         for component in components:
  90             Packages = utils.get_packages_from_ftp(cnf['Dir::Root'], suite, component, arch)
  91             while Packages.Step():
  92                 package = Packages.Section.Find('Package')
  93                 dep_list = Packages.Section.Find('Depends')
  94                 if Options['package'] and package != Options['package']:
  95                     continue
  96                 if dep_list:
  97                     for d in apt_pkg.ParseDepends(dep_list):
  98                         for i in d:
  99                             if not depends.has_key(package):
 100                                 depends[package] = set()
 101                             depends[package].add(i[0])
 102
 103     priorities = {}
 104     query = """SELECT DISTINCT o.package, p.level, p.priority, m.name
 105                FROM override o
 106                JOIN suite s ON s.id = o.suite
 107                JOIN priority p ON p.id = o.priority
 108                JOIN binaries b ON b.package = o.package
 109                JOIN maintainer m ON m.id = b.maintainer
 110                JOIN bin_associations ba ON ba.bin = b.id
 111                WHERE s.suite_name = '%s'
 112                AND ba.suite = s.id
 113                AND p.level <> 0""" % suite
 114     packages = session.execute(query)
 115
 116     out = {}
 117     if Options.has_key('file'):
 118         outfile = file(os.path.expanduser(Options['file']), 'w')
 119     else:
 120         outfile = sys.stdout
 121     for p in packages:
 122         priorities[p[0]] = [p[1], p[2], p[3], True]
 123     for d in sorted(depends.keys()):
 124         for p in depends[d]:
 125             if priorities.has_key(d) and priorities.has_key(p):
 126                 if priorities[d][0] < priorities[p][0]:
 127                      if priorities[d][3]:
 128                          if not out.has_key(d):
 129                              out[d] = {}
 130                          out[d]['priority'] = priorities[d][1]
 131                          out[d]['maintainer'] = unicode(priorities[d][2], 'utf-8')
 132                          out[d]['priority'] = priorities[d][1]
 133                          priorities[d][3] = False
 134                      if not out[d].has_key('dependency'):
 135                          out[d]['dependency'] = {}
 136                      out[d]['dependency'][p] = priorities[p][1]
 137     yaml.safe_dump(out, outfile, default_flow_style=False)
 138     if Options.has_key('file'):
 139         outfile.close()
 140
 141 if __name__ == '__main__':
 142     main()