dak/make_overrides.py

   1 #!/usr/bin/env python
   2
   3 """
   4 Output override files for apt-ftparchive and indices/
   5 @contact: Debian FTP Master <ftpmaster@debian.org>
   6 @copyright: 2000, 2001, 2002, 2004, 2006  James Troup <james@nocrew.org>
   7 @license: GNU General Public License version 2 or later
   8 """
   9
  10 # This program is free software; you can redistribute it and/or modify
  11 # it under the terms of the GNU General Public License as published by
  12 # the Free Software Foundation; either version 2 of the License, or
  13 # (at your option) any later version.
  14
  15 # This program is distributed in the hope that it will be useful,
  16 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  17 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18 # GNU General Public License for more details.
  19
  20 # You should have received a copy of the GNU General Public License
  21 # along with this program; if not, write to the Free Software
  22 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  23
  24 ################################################################################
  25
  26 # This is seperate because it's horribly Debian specific and I don't
  27 # want that kind of horribleness in the otherwise generic 'dak
  28 # make-overrides'.  It does duplicate code tho.
  29
  30 ################################################################################
  31
  32 import os
  33 import sys
  34 import apt_pkg
  35
  36 from daklib.dbconn import *
  37 from daklib.config import Config
  38 from daklib import utils
  39
  40 ################################################################################
  41
  42 def usage(exit_code=0):
  43     print """Usage: dak make-overrides
  44 Outputs the override tables to text files.
  45
  46   -h, --help                show this help and exit."""
  47     sys.exit(exit_code)
  48
  49 ################################################################################
  50
  51 def do_list(output_file, suite, component, otype, session):
  52     """
  53     Fetch override data for suite from the database and dump it.
  54
  55     @type output_file: fileobject
  56     @param output_file: where to write the overrides to
  57
  58     @type suite: Suite object
  59     @param suite: A suite object describing the Suite
  60
  61     @type component: Component object
  62     @param component: The name of the component
  63
  64     @type otype: OverrideType object
  65     @param otype: object of type of override. deb/udeb/dsc
  66
  67     @type session: SQLA Session
  68     @param session: the database session in use
  69
  70     """
  71     # Here's a nice example of why the object API isn't always the
  72     # right answer.  On my laptop, the object version of the code
  73     # takes 1:45, the 'dumb' tuple-based one takes 0:16 - mhy
  74
  75     if otype.overridetype == "dsc":
  76         #q = session.query(Override).filter_by(suite_id = suite.suite_id)
  77         #q = q.filter_by(component_id = component.component_id)
  78         #q = q.filter_by(overridetype_id = otype.overridetype_id)
  79         #q = q.join(Section).order_by(Section.section, Override.package)
  80         #for o in q.all():
  81         #    dat = (o.package, o.section.section, o.maintainer)
  82         #    output_file.write(utils.result_join(dat) + '\n')
  83         q = session.execute("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.suite_id, component.component_id, otype.overridetype_id))
  84         for i in q.fetchall():
  85             output_file.write(utils.result_join(i) + '\n')
  86
  87     else:
  88         #q = session.query(Override).filter_by(suite_id = suite.suite_id)
  89         #q = q.filter_by(component_id = component.component_id)
  90         #q = q.filter_by(overridetype_id = otype.overridetype_id)
  91         #q = q.join(Priority).join(Section).order_by(Section.section, Priority.level, Override.package)
  92         #for o in q.all():
  93         #    dat = (o.package, o.priority.priority, o.section.section, o.maintainer)
  94         #    output_file.write(utils.result_join(dat) + '\n')
  95         q = session.execute("SELECT o.package, p.priority, s.section, o.maintainer 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.suite_id, component.component_id, otype.overridetype_id))
  96         for i in q.fetchall():
  97             output_file.write(utils.result_join(i) + '\n')
  98
  99 ################################################################################
 100
 101 def main ():
 102     cnf = Config()
 103     Arguments = [('h',"help","Make-Overrides::Options::Help")]
 104     for i in [ "help" ]:
 105         if not cnf.has_key("Make-Overrides::Options::%s" % (i)):
 106             cnf["Make-Overrides::Options::%s" % (i)] = ""
 107     apt_pkg.ParseCommandLine(cnf.Cnf, Arguments, sys.argv)
 108     Options = cnf.SubTree("Make-Overrides::Options")
 109     if Options["Help"]:
 110         usage()
 111
 112     d = DBConn()
 113     session = d.session()
 114
 115     for suite in session.query(Suite).filter(Suite.overrideprocess==True):
 116         if suite.untouchable:
 117             print "Skipping %s as it is marked as untouchable" % suite.suite_name
 118             continue
 119
 120         sys.stderr.write("Processing %s...\n" % (suite.suite_name))
 121         override_suite = suite.overridecodename
 122
 123         for component in session.query(Component).all():
 124             for otype in session.query(OverrideType).all():
 125                 otype_name = otype.overridetype
 126                 cname = component.component_name
 127
 128                 # TODO: Stick suffix info in database (or get rid of it)
 129                 if otype_name == "deb":
 130                     suffix = ""
 131                 elif otype_name == "udeb":
 132                     if cname == "contrib":
 133                         continue # Ick2
 134                     suffix = ".debian-installer"
 135                 elif otype_name == "dsc":
 136                     suffix = ".src"
 137                 else:
 138                     utils.fubar("Don't understand OverrideType %s" % otype.overridetype)
 139
 140                 cname = cname.replace('/', '_')
 141                 filename = os.path.join(cnf["Dir::Override"], "override.%s.%s%s" % (override_suite, cname, suffix))
 142
 143                 output_file = utils.open_file(filename, 'w')
 144                 do_list(output_file, suite, component, otype, session)
 145                 output_file.close()
 146
 147 ################################################################################
 148
 149 if __name__ == '__main__':
 150     main()