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 Pool
  35 import apt_pkg, os, stat, sys
  36
  37 def usage():
  38     print """Usage: dak generate-packages-sources2 [OPTIONS]
  39 Generate the Packages/Sources files
  40
  41   -s, --suite=SUITE            process this suite
  42                                Default: All suites not marked 'untouchable'
  43   -f, --force                  Allow processing of untouchable suites
  44                                CAREFUL: Only to be used at point release time!
  45   -h, --help                   show this help and exit
  46
  47 SUITE can be a space seperated list, e.g.
  48    --suite=unstable testing
  49 """
  50     sys.exit()
  51
  52 #############################################################################
  53
  54 # Here be dragons.
  55 _sources_query = R"""
  56 SELECT
  57
  58   (SELECT
  59      STRING_AGG(
  60        CASE
  61          WHEN key = 'Source' THEN 'Package\: '
  62          WHEN key = 'Files' THEN E'Files\:\n ' || f.md5sum || ' ' || f.size || ' ' || SUBSTRING(f.filename FROM E'/([^/]*)\\Z')
  63          WHEN key = 'Checksums-Sha1' THEN E'Checksums-Sha1\:\n ' || f.sha1sum || ' ' || f.size || ' ' || SUBSTRING(f.filename FROM E'/([^/]*)\\Z')
  64          WHEN key = 'Checksums-Sha256' THEN E'Checksums-Sha256\:\n ' || f.sha256sum || ' ' || f.size || ' ' || SUBSTRING(f.filename FROM E'/([^/]*)\\Z')
  65          ELSE key || '\: '
  66        END || value, E'\n' ORDER BY mk.ordering, mk.key)
  67    FROM
  68      source_metadata sm
  69      JOIN metadata_keys mk ON mk.key_id = sm.key_id
  70    WHERE s.id=sm.src_id
  71   )
  72   ||
  73   E'\nDirectory\: pool/' || SUBSTRING(f.filename FROM E'\\A(.*)/[^/]*\\Z')
  74   ||
  75   E'\nPriority\: ' || pri.priority
  76   ||
  77   E'\nSection\: ' || sec.section
  78
  79 FROM
  80
  81 source s
  82 JOIN src_associations sa ON s.id = sa.source
  83 JOIN files f ON s.file=f.id
  84 JOIN override o ON o.package = s.source
  85 JOIN section sec ON o.section = sec.id
  86 JOIN priority pri ON o.priority = pri.id
  87
  88 WHERE
  89   sa.suite = :suite
  90   AND o.suite = :suite AND o.component = :component AND o.type = :dsc_type
  91
  92 ORDER BY
  93 s.source, s.version
  94 """
  95
  96 def open_sources(suite, component):
  97     cnf = Config()
  98     dest = os.path.join(cnf['Dir::Root'], 'dists', suite.suite_name, component.component_name, 'source', 'Sources')
  99
 100     # create queue if it does not exist yet
 101     if os.path.exists(dest) and os.path.isdir(dest):
 102         dest_dir = dest
 103     else:
 104         dest_dir = os.path.dirname(dest)
 105     if not os.path.exists(dest_dir):
 106         umask = os.umask(00000)
 107         os.makedirs(dest_dir, 02775)
 108         os.umask(umask)
 109
 110     f = open(dest, 'w')
 111     return f
 112
 113 def generate_sources(suite_id, component_id):
 114     global _sources_query
 115
 116     session = DBConn().session()
 117     dsc_type = session.query(OverrideType).filter_by(overridetype='dsc').one().overridetype_id
 118
 119     suite = session.query(Suite).get(suite_id)
 120     component = session.query(Component).get(component_id)
 121
 122     output = open_sources(suite, component)
 123
 124     # run query and write Sources
 125     r = session.execute(_sources_query, {"suite": suite_id, "component": component_id, "dsc_type": dsc_type})
 126     for (stanza,) in r:
 127         print >>output, stanza
 128         print >>output, ""
 129
 130     message = ["generate sources", suite.suite_name, component.component_name]
 131     session.rollback()
 132     return message
 133
 134 #############################################################################
 135
 136 # Here be large dragons.
 137 _packages_query = R"""
 138 WITH
 139
 140   tmp AS (
 141     SELECT
 142       b.id AS binary_id,
 143       b.package AS package,
 144       b.version AS version,
 145       b.architecture AS architecture,
 146       b.source AS source_id,
 147       s.source AS source,
 148       f.filename AS filename,
 149       f.size AS size,
 150       f.md5sum AS md5sum,
 151       f.sha1sum AS sha1sum,
 152       f.sha256sum AS sha256sum
 153     FROM
 154       binaries b
 155       JOIN bin_associations ba ON b.id = ba.bin
 156       JOIN files f ON f.id = b.file
 157       JOIN location l ON l.id = f.location
 158       JOIN source s ON b.source = s.id
 159     WHERE
 160       (b.architecture = :arch_all OR b.architecture = :arch) AND b.type = :type_name
 161       AND ba.suite = :suite
 162       AND l.component = :component
 163   )
 164
 165 SELECT
 166   (SELECT
 167      STRING_AGG(key || '\: ' || value, E'\n' ORDER BY mk.ordering, mk.key)
 168    FROM
 169      binaries_metadata bm
 170      JOIN metadata_keys mk ON mk.key_id = bm.key_id
 171    WHERE
 172      bm.bin_id = tmp.binary_id
 173      AND key != 'Section' AND key != 'Priority'
 174   )
 175   || E'\nSection\: ' || sec.section
 176   || E'\nPriority\: ' || pri.priority
 177   || E'\nFilename\: pool/' || tmp.filename
 178   || E'\nSize\: ' || tmp.size
 179   || E'\nMD5sum\: ' || tmp.md5sum
 180   || E'\nSHA1\: ' || tmp.sha1sum
 181   || E'\nSHA256\: ' || tmp.sha256sum
 182
 183 FROM
 184   tmp
 185   JOIN override o ON o.package = tmp.package
 186   JOIN section sec ON sec.id = o.section
 187   JOIN priority pri ON pri.id = o.priority
 188
 189 WHERE
 190   (
 191       architecture <> :arch_all
 192     OR
 193       (architecture = :arch_all AND source_id IN (SELECT source_id FROM tmp WHERE architecture <> :arch_all))
 194     OR
 195       (architecture = :arch_all AND source NOT IN (SELECT DISTINCT source FROM tmp WHERE architecture <> :arch_all))
 196   )
 197   AND
 198     o.type = :type_id AND o.suite = :suite AND o.component = :component
 199
 200 ORDER BY tmp.package, tmp.version
 201 """
 202
 203 def open_packages(suite, component, architecture, type_name):
 204     cnf = Config()
 205     if type_name == 'udeb':
 206         dest = os.path.join(cnf['Dir::Root'], 'dists', suite.suite_name, component.component_name, 'debian-installer', 'binary-%s' % architecture.arch_string, 'Packages')
 207     else:
 208         dest = os.path.join(cnf['Dir::Root'], 'dists', suite.suite_name, component.component_name, 'binary-%s' % architecture.arch_string, 'Packages')
 209
 210     # create queue if it does not exist yet
 211     if os.path.exists(dest) and os.path.isdir(dest):
 212         dest_dir = dest
 213     else:
 214         dest_dir = os.path.dirname(dest)
 215     if not os.path.exists(dest_dir):
 216         umask = os.umask(00000)
 217         os.makedirs(dest_dir, 02775)
 218         os.umask(umask)
 219
 220     f = open(dest, 'w')
 221     return f
 222
 223 def generate_packages(suite_id, component_id, architecture_id, type_name):
 224     global _packages_query
 225
 226     session = DBConn().session()
 227     arch_all_id = session.query(Architecture).filter_by(arch_string='all').one().arch_id
 228     type_id = session.query(OverrideType).filter_by(overridetype=type_name).one().overridetype_id
 229
 230     suite = session.query(Suite).get(suite_id)
 231     component = session.query(Component).get(component_id)
 232     architecture = session.query(Architecture).get(architecture_id)
 233
 234     output = open_packages(suite, component, architecture, type_name)
 235
 236     r = session.execute(_packages_query, {"suite": suite_id, "component": component_id,
 237         "arch": architecture_id, "type_id": type_id, "type_name": type_name, "arch_all": arch_all_id})
 238     for (stanza,) in r:
 239         print >>output, stanza
 240         print >>output, ""
 241
 242     message = ["generate-packages", suite.suite_name, component.component_name, architecture.arch_string]
 243     session.rollback()
 244     return message
 245
 246 #############################################################################
 247
 248 def main():
 249     cnf = Config()
 250
 251     Arguments = [('h',"help","Generate-Packages-Sources::Options::Help"),
 252                  ('s',"suite","Generate-Packages-Sources::Options::Suite"),
 253                  ('f',"force","Generate-Packages-Sources::Options::Force")]
 254
 255     suite_names = apt_pkg.ParseCommandLine(cnf.Cnf, Arguments, sys.argv)
 256     try:
 257         Options = cnf.SubTree("Generate-Packages-Sources::Options")
 258     except KeyError:
 259         Options = {}
 260
 261     if Options.has_key("Help"):
 262         usage()
 263
 264     logger = daklog.Logger(cnf, 'generate-packages-sources2')
 265
 266     session = DBConn().session()
 267
 268     if Options.has_key("Suite"):
 269         suites = []
 270         for s in suite_names:
 271             suite = get_suite(s.lower(), session)
 272             if suite:
 273                 suites.append(suite)
 274             else:
 275                 print "I: Cannot find suite %s" % s
 276                 logger.log(['Cannot find suite %s' % s])
 277     else:
 278         suites = session.query(Suite).filter(Suite.untouchable == False).all()
 279
 280     force = Options.has_key("Force") and Options["Force"]
 281
 282     component_ids = [ c.component_id for c in session.query(Component).all() ]
 283
 284     def log(details):
 285         logger.log(details)
 286
 287     pool = Pool()
 288     for s in suites:
 289         if s.untouchable and not force:
 290             utils.fubar("Refusing to touch %s (untouchable and not forced)" % s.suite_name)
 291         for c in component_ids:
 292             pool.apply_async(generate_sources, [s.suite_id, c], callback=log)
 293             for a in s.architectures:
 294                 pool.apply_async(generate_packages, [s.suite_id, c, a.arch_id, 'deb'], callback=log)
 295                 pool.apply_async(generate_packages, [s.suite_id, c, a.arch_id, 'udeb'], callback=log)
 296
 297     pool.close()
 298     pool.join()
 299     # this script doesn't change the database
 300     session.close()
 301
 302 if __name__ == '__main__':
 303     main()