]> git.decadent.org.uk Git - dak.git/blob - dak/override_disparity.py
Merge remote-tracking branch 'drkranz/misc' into merge
[dak.git] / 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()