]> git.decadent.org.uk Git - dak.git/commitdiff
generate Packages/Sources directly from database
authorAnsgar Burchardt <ansgar@debian.org>
Fri, 25 Mar 2011 22:39:40 +0000 (22:39 +0000)
committerAnsgar Burchardt <ansgar@debian.org>
Fri, 25 Mar 2011 22:39:40 +0000 (22:39 +0000)
We can now write Packages/Sources outselves without using
apt-ftparchive.  Before this is usable we still need to compress
the output.

Signed-off-by: Ansgar Burchardt <ansgar@debian.org>
dak/dak.py
dak/generate_packages_sources2.py [new file with mode: 0755]

index ad99a5a06c80b5c4e383dad854583d5ae67b2a15..304af5f11203d542d2d2f0c63175a7e59b3b2502 100755 (executable)
@@ -84,6 +84,8 @@ def init():
          "Generate Release files"),
         ("generate-packages-sources",
          "Generate Packages/Sources files"),
+        ("generate-packages-sources2",
+         "Generate Packages/Sources files [directly from database]"),
         ("contents",
          "Generate content files"),
         ("metadata",
diff --git a/dak/generate_packages_sources2.py b/dak/generate_packages_sources2.py
new file mode 100755 (executable)
index 0000000..5a1008c
--- /dev/null
@@ -0,0 +1,295 @@
+#!/usr/bin/python
+
+"""
+Generate Packages/Sources files
+
+@contact: Debian FTP Master <ftpmaster@debian.org>
+@copyright: 2011  Ansgar Burchardt <ansgar@debian.org>
+@copyright: Based on daklib/lists.py and dak/generate_filelist.py:
+            2009-2011  Torsten Werner <twerner@debian.org>
+@copyright: Based on dak/generate_packages_sources.py:
+            2000, 2001, 2002, 2006  James Troup <james@nocrew.org>
+            2009  Mark Hymers <mhy@debian.org>
+            2010  Joerg Jaspert <joerg@debian.org>
+@license: GNU General Public License version 2 or later
+"""
+
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+
+from daklib.dbconn import *
+from daklib.config import Config
+from daklib import utils, daklog
+from multiprocessing import Pool
+import apt_pkg, os, stat, sys
+
+from daklib.lists import getSources, getBinaries, getArchAll
+
+def usage():
+    print """Usage: dak generate-packages-sources2 [OPTIONS]
+Generate the Packages/Sources files
+
+  -s, --suite=SUITE            process this suite
+                               Default: All suites not marked 'untouchable'
+  -f, --force                  Allow processing of untouchable suites
+                               CAREFUL: Only to be used at point release time!
+  -h, --help                   show this help and exit
+
+SUITE can be a space seperated list, e.g.
+   --suite=unstable testing
+"""
+    sys.exit()
+
+#############################################################################
+
+# Here be dragons.
+_sources_query = R"""
+SELECT
+
+  (SELECT
+     STRING_AGG(
+       CASE
+         WHEN key = 'Source' THEN 'Package\: '
+         WHEN key = 'Files' THEN E'Files\:\n ' || f.md5sum || ' ' || f.size || ' ' || SUBSTRING(f.filename FROM E'/([^/]*)\\Z')
+         WHEN key = 'Checksums-Sha1' THEN E'Checksums-Sha1\:\n ' || f.sha1sum || ' ' || f.size || ' ' || SUBSTRING(f.filename FROM E'/([^/]*)\\Z')
+         WHEN key = 'Checksums-Sha256' THEN E'Checksums-Sha256\:\n ' || f.sha256sum || ' ' || f.size || ' ' || SUBSTRING(f.filename FROM E'/([^/]*)\\Z')
+         ELSE key || '\: '
+       END || value, E'\n' ORDER BY mk.order)
+   FROM
+     source_metadata sm
+     JOIN metadata_keys mk ON mk.key_id = sm.key_id
+   WHERE s.id=sm.src_id
+  )
+  ||
+  E'\nDirectory\: pool/' || SUBSTRING(f.filename FROM E'\\A(.*)/[^/]*\\Z')
+  ||
+  E'\nPriority\: ' || pri.priority
+  ||
+  E'\nSection\: ' || sec.section
+
+FROM
+
+source s
+JOIN files f ON s.file=f.id
+JOIN override o ON o.package = s.source
+JOIN section sec ON o.section = sec.id
+JOIN priority pri ON o.priority = pri.id
+
+WHERE
+o.suite = :suite AND o.component = :component AND o.type = :dsc_type
+
+ORDER BY
+s.source, s.version
+"""
+
+def open_sources(suite, component):
+    cnf = Config()
+    dest = os.path.join(cnf['Dir::Root'], 'dists', suite.suite_name, component.component_name, 'source', 'Sources')
+
+    # create queue if it does not exist yet
+    if os.path.exists(dest) and os.path.isdir(dest):
+        dest_dir = dest
+    else:
+        dest_dir = os.path.dirname(dest)
+    if not os.path.exists(dest_dir):
+        umask = os.umask(00000)
+        os.makedirs(dest_dir, 02775)
+        os.umask(umask)
+
+    f = open(dest, 'w')
+    return f
+
+def generate_sources(suite_id, component_id):
+    global _sources_query
+
+    session = DBConn().session()
+    dsc_type = session.query(OverrideType).filter_by(overridetype='dsc').one().overridetype_id
+
+    suite = session.query(Suite).get(suite_id)
+    component = session.query(Component).get(component_id)
+
+    output = open_sources(suite, component)
+
+    # run query and write Sources
+    r = session.execute(_sources_query, {"suite": suite_id, "component": component_id, "dsc_type": dsc_type})
+    for (stanza,) in r:
+        print >>output, stanza
+        print >>output, ""
+
+    return ["generate sources", suite.suite_name, component.component_name]
+
+#############################################################################
+
+# Here be large dragons.
+_packages_query = R"""
+WITH
+
+  tmp AS (
+    SELECT
+      b.id AS binary_id,
+      b.package AS package,
+      b.version AS version,
+      b.architecture AS architecture,
+      b.source AS source_id,
+      s.source AS source,
+      f.filename AS filename,
+      f.size AS size,
+      f.md5sum AS md5sum,
+      f.sha1sum AS sha1sum,
+      f.sha256sum AS sha256sum
+    FROM
+      binaries b
+      JOIN bin_associations ba ON b.id = ba.bin
+      JOIN files f ON f.id = b.file
+      JOIN location l ON l.id = f.location
+      JOIN source s ON b.source = s.id
+    WHERE
+      (b.architecture = :arch_all OR b.architecture = :all) AND b.type = :type_name
+      AND ba.suite = :suite
+      AND l.component = :component
+  )
+
+SELECT
+  (SELECT
+     STRING_AGG(key || '\: ' || value, E'\n' ORDER BY mk.order)
+   FROM
+     binaries_metadata bm
+     JOIN metadata_keys mk ON mk.key_id = bm.key_id
+   WHERE
+     bm.bin_id = tmp.binary_id
+     AND key != 'Section' AND key != 'Priority'
+  )
+  || E'\nSection\: ' || sec.section
+  || E'\nPriority\: ' || pri.priority
+  || E'\nFilename\: ' || tmp.filename
+  || E'\nSize\: ' || tmp.size
+  || E'\nMD5sum\: ' || tmp.md5sum
+  || E'\nSHA1\: ' || tmp.sha1sum
+  || E'\nSHA256\: ' || tmp.sha256sum
+
+FROM
+  tmp
+  JOIN override o ON o.package = tmp.package
+  JOIN section sec ON sec.id = o.section
+  JOIN priority pri ON pri.id = o.priority
+
+WHERE
+  (
+      architecture <> :arch_all
+    OR
+      (architecture = :arch_all AND source_id IN (SELECT source_id FROM tmp WHERE architecture <> :arch_all))
+    OR
+      (architecture = :arch_all AND source NOT IN (SELECT DISTINCT source FROM tmp WHERE architecture <> :arch_all))
+  )
+  AND
+    o.type = :type_id AND o.suite = :suite AND o.component = :component
+"""
+
+def open_packages(suite, component, architecture, type_name):
+    cnf = Config()
+    if type_name == 'udeb':
+        dest = os.path.join(cnf['Dir::Root'], 'dists', suite.suite_name, component.component_name, 'debian-installer', 'binary-%s' % architecture.arch_string, 'Packages')
+    else:
+        dest = os.path.join(cnf['Dir::Root'], 'dists', suite.suite_name, component.component_name, 'binary-%s' % architecture.arch_string, 'Packages')
+
+    # create queue if it does not exist yet
+    if os.path.exists(dest) and os.path.isdir(dest):
+        dest_dir = dest
+    else:
+        dest_dir = os.path.dirname(dest)
+    if not os.path.exists(dest_dir):
+        umask = os.umask(00000)
+        os.makedirs(dest_dir, 02775)
+        os.umask(umask)
+
+    f = open(dest, 'w')
+    return f
+
+def generate_packages(suite_id, component_id, architecture_id, type_name):
+    global _packages_query
+
+    session = DBConn().session()
+    arch_all_id = session.query(Architecture).filter_by(arch_string='all').one().arch_id
+    type_id = session.query(OverrideType).filter_by(overridetype=type_name).one().overridetype_id
+
+    suite = session.query(Suite).get(suite_id)
+    component = session.query(Component).get(component_id)
+    architecture = session.query(Architecture).get(architecture_id)
+
+    output = open_packages(suite, component, architecture, type_name)
+
+    r = session.execute(_packages_query, {"suite": suite_id, "component": component_id,
+        "type_id": type_id, "type_name": type_name, "arch_all": arch_all_id})
+    for (stanza,) in r:
+        print >>output, stanza
+
+    session.close()
+
+    return ["generate-packages", suite.suite_name, component.component_name, architecture.arch_string]
+
+#############################################################################
+
+def main():
+    cnf = Config()
+
+    Arguments = [('h',"help","Generate-Packages-Sources::Options::Help"),
+                 ('s',"suite","Generate-Packages-Sources::Options::Suite"),
+                 ('f',"force","Generate-Packages-Sources::Options::Force")]
+
+    suite_names = apt_pkg.ParseCommandLine(cnf.Cnf, Arguments, sys.argv)
+    try:
+        Options = cnf.SubTree("Generate-Packages-Sources::Options")
+    except KeyError:
+        Options = {}
+
+    if Options.has_key("Help"):
+        usage()
+
+    logger = daklog.Logger(cnf, 'generate-packages-sources2')
+
+    session = DBConn().session()
+
+    if Options.has_key("Suite"):
+        suites = []
+        for s in suite_names:
+            suite = get_suite(s.lower(), session)
+            if suite:
+                suites.append(suite)
+            else:
+                print "I: Cannot find suite %s" % s
+                logger.log(['Cannot find suite %s' % s])
+    else:
+        suites = session.query(Suite).filter(Suite.untouchable == False).all()
+
+    component_ids = [ c.component_id for c in session.query(Component).all() ]
+
+    def log(details):
+        logger.log(details)
+
+    pool = Pool()
+    for s in suites:
+        for c in component_ids:
+            pool.apply_async(generate_sources, [s.suite_id, c], callback=log)
+            for a in s.architectures:
+                #pool.apply_async(generate_packages, [s.suite_id, c, a.arch_id, 'deb'], callback=log)
+                apply(generate_packages, [s.suite_id, c, a.arch_id, 'deb'])
+                #pool.apply_async(generate_packages, [s.suite_id, c, a.arch_id, 'udeb'], callback=log)
+
+    pool.close()
+    pool.join()
+    # this script doesn't change the database
+    session.close()
+
+if __name__ == '__main__':
+    main()