dak/generate_packages_sources2.py

   1 #!/usr/bin/python
   2
   3 """
   4 Generate Packages/Sources files
   5
   6 @contact: Debian FTP Master <ftpmaster@debian.org>
   7 @copyright: 2011  Ansgar Burchardt <ansgar@debian.org>
   8 @copyright: Based on daklib/lists.py and dak/generate_filelist.py:
   9             2009-2011  Torsten Werner <twerner@debian.org>
  10 @copyright: Based on dak/generate_packages_sources.py:
  11             2000, 2001, 2002, 2006  James Troup <james@nocrew.org>
  12             2009  Mark Hymers <mhy@debian.org>
  13             2010  Joerg Jaspert <joerg@debian.org>
  14 @license: GNU General Public License version 2 or later
  15 """
  16
  17 # This program is free software; you can redistribute it and/or modify
  18 # it under the terms of the GNU General Public License as published by
  19 # the Free Software Foundation; either version 2 of the License, or
  20 # (at your option) any later version.
  21
  22 # This program is distributed in the hope that it will be useful,
  23 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  24 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  25 # GNU General Public License for more details.
  26
  27 # You should have received a copy of the GNU General Public License
  28 # along with this program; if not, write to the Free Software
  29 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  30
  31 from daklib.dbconn import *
  32 from daklib.config import Config
  33 from daklib import utils, daklog
  34 from daklib.dakmultiprocessing import DakProcessPool, PROC_STATUS_SUCCESS, PROC_STATUS_SIGNALRAISED
  35 from daklib.filewriter import PackagesFileWriter, SourcesFileWriter
  36
  37 import apt_pkg, os, stat, sys
  38
  39 def usage():
  40     print """Usage: dak generate-packages-sources2 [OPTIONS]
  41 Generate the Packages/Sources files
  42
  43   -s, --suite=SUITE            process this suite
  44                                Default: All suites not marked 'untouchable'
  45   -f, --force                  Allow processing of untouchable suites
  46                                CAREFUL: Only to be used at point release time!
  47   -h, --help                   show this help and exit
  48
  49 SUITE can be a space seperated list, e.g.
  50    --suite=unstable testing
  51 """
  52     sys.exit()
  53
  54 #############################################################################
  55
  56 # Here be dragons.
  57 _sources_query = R"""
  58 SELECT
  59
  60   (SELECT
  61      STRING_AGG(
  62        CASE
  63          WHEN key = 'Source' THEN 'Package\: '
  64          WHEN key = 'Files' THEN E'Files\:\n ' || f.md5sum || ' ' || f.size || ' ' || SUBSTRING(f.filename FROM E'/([^/]*)\\Z')
  65          WHEN key = 'Checksums-Sha1' THEN E'Checksums-Sha1\:\n ' || f.sha1sum || ' ' || f.size || ' ' || SUBSTRING(f.filename FROM E'/([^/]*)\\Z')
  66          WHEN key = 'Checksums-Sha256' THEN E'Checksums-Sha256\:\n ' || f.sha256sum || ' ' || f.size || ' ' || SUBSTRING(f.filename FROM E'/([^/]*)\\Z')
  67          ELSE key || '\: '
  68        END || value, E'\n' ORDER BY mk.ordering, mk.key)
  69    FROM
  70      source_metadata sm
  71      JOIN metadata_keys mk ON mk.key_id = sm.key_id
  72    WHERE s.id=sm.src_id
  73   )
  74   ||
  75   E'\nDirectory\: pool/' || SUBSTRING(f.filename FROM E'\\A(.*)/[^/]*\\Z')
  76   ||
  77   E'\nPriority\: ' || pri.priority
  78   ||
  79   E'\nSection\: ' || sec.section
  80
  81 FROM
  82
  83 source s
  84 JOIN src_associations sa ON s.id = sa.source
  85 JOIN files f ON s.file=f.id
  86 JOIN override o ON o.package = s.source
  87 JOIN section sec ON o.section = sec.id
  88 JOIN priority pri ON o.priority = pri.id
  89
  90 WHERE
  91   sa.suite = :suite
  92   AND o.suite = :overridesuite AND o.component = :component AND o.type = :dsc_type
  93
  94 ORDER BY
  95 s.source, s.version
  96 """
  97
  98 def generate_sources(suite_id, component_id):
  99     global _sources_query
 100
 101     session = DBConn().session()
 102     dsc_type = session.query(OverrideType).filter_by(overridetype='dsc').one().overridetype_id
 103
 104     suite = session.query(Suite).get(suite_id)
 105     component = session.query(Component).get(component_id)
 106
 107     overridesuite_id = suite.get_overridesuite().suite_id
 108
 109     writer = SourcesFileWriter(suite=suite.suite_name, component=component.component_name)
 110     output = writer.open()
 111
 112     # run query and write Sources
 113     r = session.execute(_sources_query, {"suite": suite_id, "component": component_id, "dsc_type": dsc_type, "overridesuite": overridesuite_id})
 114     for (stanza,) in r:
 115         print >>output, stanza
 116         print >>output, ""
 117
 118     writer.close()
 119
 120     message = ["generate sources", suite.suite_name, component.component_name]
 121     session.rollback()
 122     return (PROC_STATUS_SUCCESS, message)
 123
 124 #############################################################################
 125
 126 # Here be large dragons.
 127 _packages_query = R"""
 128 WITH
 129
 130   tmp AS (
 131     SELECT
 132       b.id AS binary_id,
 133       b.package AS package,
 134       b.version AS version,
 135       b.architecture AS architecture,
 136       b.source AS source_id,
 137       s.source AS source,
 138       f.filename AS filename,
 139       f.size AS size,
 140       f.md5sum AS md5sum,
 141       f.sha1sum AS sha1sum,
 142       f.sha256sum AS sha256sum
 143     FROM
 144       binaries b
 145       JOIN bin_associations ba ON b.id = ba.bin
 146       JOIN files f ON f.id = b.file
 147       JOIN location l ON l.id = f.location
 148       JOIN source s ON b.source = s.id
 149     WHERE
 150       (b.architecture = :arch_all OR b.architecture = :arch) AND b.type = :type_name
 151       AND ba.suite = :suite
 152       AND l.component = :component
 153   )
 154
 155 SELECT
 156   (SELECT
 157      STRING_AGG(key || '\: ' || value, E'\n' ORDER BY mk.ordering, mk.key)
 158    FROM
 159      binaries_metadata bm
 160      JOIN metadata_keys mk ON mk.key_id = bm.key_id
 161    WHERE
 162      bm.bin_id = tmp.binary_id
 163      AND key != 'Section' AND key != 'Priority'
 164   )
 165   || COALESCE(E'\n' || (SELECT
 166      STRING_AGG(key || '\: ' || value, E'\n' ORDER BY key)
 167    FROM external_overrides eo
 168    WHERE
 169      eo.package = tmp.package
 170      AND eo.suite = :overridesuite AND eo.component = :component
 171   ), '')
 172   || E'\nSection\: ' || sec.section
 173   || E'\nPriority\: ' || pri.priority
 174   || E'\nFilename\: pool/' || tmp.filename
 175   || E'\nSize\: ' || tmp.size
 176   || E'\nMD5sum\: ' || tmp.md5sum
 177   || E'\nSHA1\: ' || tmp.sha1sum
 178   || E'\nSHA256\: ' || tmp.sha256sum
 179
 180 FROM
 181   tmp
 182   JOIN override o ON o.package = tmp.package
 183   JOIN section sec ON sec.id = o.section
 184   JOIN priority pri ON pri.id = o.priority
 185
 186 WHERE
 187   (
 188       architecture <> :arch_all
 189     OR
 190       (architecture = :arch_all AND source_id IN (SELECT source_id FROM tmp WHERE architecture <> :arch_all))
 191     OR
 192       (architecture = :arch_all AND source NOT IN (SELECT DISTINCT source FROM tmp WHERE architecture <> :arch_all))
 193   )
 194   AND
 195     o.type = :type_id AND o.suite = :overridesuite AND o.component = :component
 196
 197 ORDER BY tmp.package, tmp.version
 198 """
 199
 200 def generate_packages(suite_id, component_id, architecture_id, type_name):
 201     global _packages_query
 202
 203     session = DBConn().session()
 204     arch_all_id = session.query(Architecture).filter_by(arch_string='all').one().arch_id
 205     type_id = session.query(OverrideType).filter_by(overridetype=type_name).one().overridetype_id
 206
 207     suite = session.query(Suite).get(suite_id)
 208     component = session.query(Component).get(component_id)
 209     architecture = session.query(Architecture).get(architecture_id)
 210
 211     overridesuite_id = suite.get_overridesuite().suite_id
 212
 213     writer = PackagesFileWriter(suite=suite.suite_name, component=component.component_name,
 214             architecture=architecture.arch_string, debtype=type_name)
 215     output = writer.open()
 216
 217     r = session.execute(_packages_query, {"suite": suite_id, "component": component_id,
 218         "arch": architecture_id, "type_id": type_id, "type_name": type_name, "arch_all": arch_all_id,
 219         "overridesuite": overridesuite_id})
 220     for (stanza,) in r:
 221         print >>output, stanza
 222         print >>output, ""
 223
 224     writer.close()
 225
 226     message = ["generate-packages", suite.suite_name, component.component_name, architecture.arch_string]
 227     session.rollback()
 228     return (PROC_STATUS_SUCCESS, message)
 229
 230 #############################################################################
 231
 232 def main():
 233     cnf = Config()
 234
 235     Arguments = [('h',"help","Generate-Packages-Sources::Options::Help"),
 236                  ('s',"suite","Generate-Packages-Sources::Options::Suite"),
 237                  ('f',"force","Generate-Packages-Sources::Options::Force")]
 238
 239     suite_names = apt_pkg.ParseCommandLine(cnf.Cnf, Arguments, sys.argv)
 240     try:
 241         Options = cnf.SubTree("Generate-Packages-Sources::Options")
 242     except KeyError:
 243         Options = {}
 244
 245     if Options.has_key("Help"):
 246         usage()
 247
 248     logger = daklog.Logger(cnf, 'generate-packages-sources2')
 249
 250     session = DBConn().session()
 251
 252     if Options.has_key("Suite"):
 253         suites = []
 254         for s in suite_names:
 255             suite = get_suite(s.lower(), session)
 256             if suite:
 257                 suites.append(suite)
 258             else:
 259                 print "I: Cannot find suite %s" % s
 260                 logger.log(['Cannot find suite %s' % s])
 261     else:
 262         suites = session.query(Suite).filter(Suite.untouchable == False).all()
 263
 264     force = Options.has_key("Force") and Options["Force"]
 265
 266     component_ids = [ c.component_id for c in session.query(Component).all() ]
 267
 268     def parse_results(message):
 269         # Split out into (code, msg)
 270         code, msg = message
 271         if code == PROC_STATUS_SUCCESS:
 272             logger.log([msg])
 273         elif code == PROC_STATUS_SIGNALRAISED:
 274             logger.log(['E: Subprocess recieved signal ', msg])
 275         else:
 276             logger.log(['E: ', msg])
 277
 278     pool = DakProcessPool()
 279     for s in suites:
 280         if s.untouchable and not force:
 281             utils.fubar("Refusing to touch %s (untouchable and not forced)" % s.suite_name)
 282         for c in component_ids:
 283             pool.apply_async(generate_sources, [s.suite_id, c], callback=parse_results)
 284             for a in s.architectures:
 285                 pool.apply_async(generate_packages, [s.suite_id, c, a.arch_id, 'deb'], callback=parse_results)
 286                 pool.apply_async(generate_packages, [s.suite_id, c, a.arch_id, 'udeb'], callback=parse_results)
 287
 288     pool.close()
 289     pool.join()
 290
 291     # this script doesn't change the database
 292     session.close()
 293
 294     logger.close()
 295
 296     sys.exit(pool.overall_status())
 297
 298 if __name__ == '__main__':
 299     main()