denise

   1 #!/usr/bin/env python
   2
   3 # Output override files for apt-ftparchive and indices/
   4 # Copyright (C) 2000, 2001, 2002, 2004  James Troup <james@nocrew.org>
   5 # $Id: denise,v 1.17 2004-11-27 17:56:42 troup Exp $
   6
   7 # This program is free software; you can redistribute it and/or modify
   8 # it under the terms of the GNU General Public License as published by
   9 # the Free Software Foundation; either version 2 of the License, or
  10 # (at your option) any later version.
  11
  12 # This program is distributed in the hope that it will be useful,
  13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15 # GNU General Public License for more details.
  16
  17 # You should have received a copy of the GNU General Public License
  18 # along with this program; if not, write to the Free Software
  19 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  20
  21 ################################################################################
  22
  23 # This is seperate because it's horribly Debian specific and I don't
  24 # want that kind of horribleness in the otherwise generic natalie.  It
  25 # does duplicate code tho.
  26
  27 ################################################################################
  28
  29 import pg, sys;
  30 import utils, db_access;
  31 import apt_pkg;
  32
  33 ################################################################################
  34
  35 Cnf = None;
  36 projectB = None;
  37 override = {}
  38
  39 ################################################################################
  40
  41 def usage(exit_code=0):
  42     print """Usage: denise
  43 Outputs the override tables to text files.
  44
  45   -h, --help                show this help and exit."""
  46     sys.exit(exit_code)
  47
  48 ################################################################################
  49
  50 def do_list(output_file, suite, component, otype):
  51     global override;
  52
  53     suite_id = db_access.get_suite_id(suite);
  54     if suite_id == -1:
  55         utils.fubar("Suite '%s' not recognised." % (suite));
  56
  57     component_id = db_access.get_component_id(component);
  58     if component_id == -1:
  59         utils.fubar("Component '%s' not recognised." % (component));
  60
  61     otype_id = db_access.get_override_type_id(otype);
  62     if otype_id == -1:
  63         utils.fubar("Type '%s' not recognised. (Valid types are deb, udeb and dsc)" % (otype));
  64
  65     override.setdefault(suite, {});
  66     override[suite].setdefault(component, {});
  67     override[suite][component].setdefault(otype, {});
  68
  69     if otype == "dsc":
  70         q = projectB.query("SELECT o.package, s.section, o.maintainer FROM override o, section s WHERE o.suite = %s AND o.component = %s AND o.type = %s AND o.section = s.id ORDER BY s.section, o.package" % (suite_id, component_id, otype_id));
  71         for i in q.getresult():
  72             override[suite][component][otype][i[0]] = i;
  73             output_file.write(utils.result_join(i)+'\n');
  74     else:
  75         q = projectB.query("SELECT o.package, p.priority, s.section, o.maintainer, p.level FROM override o, priority p, section s WHERE o.suite = %s AND o.component = %s AND o.type = %s AND o.priority = p.id AND o.section = s.id ORDER BY s.section, p.level, o.package" % (suite_id, component_id, otype_id));
  76         for i in q.getresult():
  77             i = i[:-1]; # Strip the priority level
  78             override[suite][component][otype][i[0]] = i;
  79             output_file.write(utils.result_join(i)+'\n');
  80
  81 ################################################################################
  82
  83 def main ():
  84     global Cnf, projectB, override;
  85
  86     Cnf = utils.get_conf()
  87     Arguments = [('h',"help","Denise::Options::Help")];
  88     for i in [ "help" ]:
  89         if not Cnf.has_key("Denise::Options::%s" % (i)):
  90             Cnf["Denise::Options::%s" % (i)] = "";
  91     apt_pkg.ParseCommandLine(Cnf,Arguments,sys.argv);
  92     Options = Cnf.SubTree("Denise::Options")
  93     if Options["Help"]:
  94         usage();
  95
  96     projectB = pg.connect(Cnf["DB::Name"], Cnf["DB::Host"], int(Cnf["DB::Port"]));
  97     db_access.init(Cnf, projectB);
  98
  99     for suite in [ "stable", "unstable" ]:
 100         sys.stderr.write("Processing %s...\n" % (suite));
 101         override_suite = Cnf["Suite::%s::OverrideCodeName" % (suite)];
 102         for component in Cnf.SubTree("Component").List():
 103             if component == "mixed":
 104                 continue; # Ick
 105             for otype in Cnf.ValueList("OverrideType"):
 106                 if otype == "deb":
 107                     suffix = "";
 108                 elif otype == "udeb":
 109                     if suite != "unstable" or component != "main":
 110                         continue; # Ick2
 111                     suffix = ".debian-installer";
 112                 elif otype == "dsc":
 113                     suffix = ".src";
 114                 filename = "%s/override.%s.%s%s" % (Cnf["Dir::Override"], override_suite, component.replace("non-US/", ""), suffix);
 115                 output_file = utils.open_file(filename, 'w');
 116                 do_list(output_file, suite, component, otype);
 117                 output_file.close();
 118
 119     # Munge the override file for testing by using unstable's where
 120     # possible and falling back on stable's where it's not.
 121
 122     sys.stderr.write("Processing testing...\n");
 123     suite = "testing";
 124     suite_id = db_access.get_suite_id(suite);
 125     override_suite = Cnf["Suite::%s::OverrideCodeName" % (suite)];
 126     for component in Cnf.SubTree("Component").List():
 127         if component == "mixed":
 128             continue;
 129         component_id = db_access.get_component_id(component);
 130         for otype in Cnf.ValueList("OverrideType"):
 131             if otype == "deb" or otype == "udeb":
 132                 if otype == "deb":
 133                     suffix = "";
 134                 elif otype == "udeb":
 135                     if component != "main":
 136                         continue;
 137                     suffix = ".debian-installer";
 138                 q = projectB.query("SELECT DISTINCT b.package FROM bin_associations ba, binaries b, files f, location l WHERE ba.suite = %s AND l.component = %s AND b.type = '%s' AND ba.bin = b.id AND b.file = f.id AND f.location = l.id" % (suite_id, component_id, otype));
 139             elif otype == "dsc":
 140                 q = projectB.query("SELECT DISTINCT s.source FROM src_associations sa, source s, files f, location l WHERE sa.suite = %s AND l.component = %s AND sa.source = s.id AND s.file = f.id AND f.location = l.id" % (suite_id, component_id));
 141                 suffix = ".src";
 142             filename = "%s/override.%s.%s%s" % (Cnf["Dir::Override"], override_suite, component.replace("non-US/", ""), suffix);
 143             output_file = utils.open_file(filename, 'w');
 144             for i in q.getresult():
 145                 package = i[0];
 146                 if otype == "deb" or otype == "dsc":
 147                     if override["unstable"][component][otype].has_key(package):
 148                         output_file.write(utils.result_join(override["unstable"][component][otype][package])+'\n');
 149                     elif override["stable"][component][otype].has_key(package):
 150                         output_file.write(utils.result_join(override["stable"][component][otype][package])+'\n');
 151                     else:
 152                         if otype == "dsc" and (override["unstable"][component]["deb"].has_key(package) or override["stable"][component]["deb"].has_key(package)):
 153                             continue; # source falls back on binary; so accept silently
 154                         utils.warn("Can't find override entry for testing package '%s' (component %s, type %s)." % (package, component, otype));
 155                 elif otype == "udeb":
 156                     if component == "main" and override["unstable"][component][otype].has_key(package):
 157                         output_file.write(utils.result_join(override["unstable"][component][otype][package])+'\n');
 158                     else:
 159                         utils.warn("Can't find override entry for testing package '%s' (component %s, type %s)." % (package, component, otype));
 160
 161             output_file.close();
 162
 163 ################################################################################
 164
 165 if __name__ == '__main__':
 166     main();