]> git.decadent.org.uk Git - dak.git/blob - dak/generate_packages_sources2.py
Merge remote-tracking branch 'drkranz/misc' into merge
[dak.git] / 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 import apt_pkg, sys
32
33 def usage():
34     print """Usage: dak generate-packages-sources2 [OPTIONS]
35 Generate the Packages/Sources files
36
37   -s, --suite=SUITE            process this suite
38                                Default: All suites not marked 'untouchable'
39   -f, --force                  Allow processing of untouchable suites
40                                CAREFUL: Only to be used at point release time!
41   -h, --help                   show this help and exit
42
43 SUITE can be a space seperated list, e.g.
44    --suite=unstable testing
45 """
46     sys.exit()
47
48 #############################################################################
49
50 # Here be dragons.
51 _sources_query = R"""
52 SELECT
53
54   (SELECT
55      STRING_AGG(
56        CASE
57          WHEN key = 'Source' THEN 'Package\: '
58          WHEN key = 'Files' THEN E'Files\:\n ' || f.md5sum || ' ' || f.size || ' ' || SUBSTRING(f.filename FROM E'/([^/]*)\\Z')
59          WHEN key = 'Checksums-Sha1' THEN E'Checksums-Sha1\:\n ' || f.sha1sum || ' ' || f.size || ' ' || SUBSTRING(f.filename FROM E'/([^/]*)\\Z')
60          WHEN key = 'Checksums-Sha256' THEN E'Checksums-Sha256\:\n ' || f.sha256sum || ' ' || f.size || ' ' || SUBSTRING(f.filename FROM E'/([^/]*)\\Z')
61          ELSE key || '\: '
62        END || value, E'\n' ORDER BY mk.ordering, mk.key)
63    FROM
64      source_metadata sm
65      JOIN metadata_keys mk ON mk.key_id = sm.key_id
66    WHERE s.id=sm.src_id
67   )
68   ||
69   E'\nDirectory\: pool/' || SUBSTRING(f.filename FROM E'\\A(.*)/[^/]*\\Z')
70   ||
71   E'\nPriority\: ' || pri.priority
72   ||
73   E'\nSection\: ' || sec.section
74
75 FROM
76
77 source s
78 JOIN src_associations sa ON s.id = sa.source
79 JOIN files f ON s.file=f.id
80 JOIN override o ON o.package = s.source
81 JOIN section sec ON o.section = sec.id
82 JOIN priority pri ON o.priority = pri.id
83
84 WHERE
85   sa.suite = :suite
86   AND o.suite = :overridesuite AND o.component = :component AND o.type = :dsc_type
87
88 ORDER BY
89 s.source, s.version
90 """
91
92 def generate_sources(suite_id, component_id):
93     global _sources_query
94     from daklib.filewriter import SourcesFileWriter
95     from daklib.dbconn import Component, DBConn, OverrideType, Suite
96     from daklib.dakmultiprocessing import PROC_STATUS_SUCCESS
97
98     session = DBConn().session()
99     dsc_type = session.query(OverrideType).filter_by(overridetype='dsc').one().overridetype_id
100
101     suite = session.query(Suite).get(suite_id)
102     component = session.query(Component).get(component_id)
103
104     overridesuite_id = suite.get_overridesuite().suite_id
105
106     writer = SourcesFileWriter(suite=suite.suite_name, component=component.component_name)
107     output = writer.open()
108
109     # run query and write Sources
110     r = session.execute(_sources_query, {"suite": suite_id, "component": component_id, "dsc_type": dsc_type, "overridesuite": overridesuite_id})
111     for (stanza,) in r:
112         print >>output, stanza
113         print >>output, ""
114
115     writer.close()
116
117     message = ["generate sources", suite.suite_name, component.component_name]
118     session.rollback()
119     return (PROC_STATUS_SUCCESS, message)
120
121 #############################################################################
122
123 # Here be large dragons.
124 _packages_query = R"""
125 WITH
126
127   tmp AS (
128     SELECT
129       b.id AS binary_id,
130       b.package AS package,
131       b.version AS version,
132       b.architecture AS architecture,
133       b.source AS source_id,
134       s.source AS source,
135       f.filename AS filename,
136       f.size AS size,
137       f.md5sum AS md5sum,
138       f.sha1sum AS sha1sum,
139       f.sha256sum AS sha256sum
140     FROM
141       binaries b
142       JOIN bin_associations ba ON b.id = ba.bin
143       JOIN files f ON f.id = b.file
144       JOIN location l ON l.id = f.location
145       JOIN source s ON b.source = s.id
146     WHERE
147       (b.architecture = :arch_all OR b.architecture = :arch) AND b.type = :type_name
148       AND ba.suite = :suite
149       AND l.component = :component
150   )
151
152 SELECT
153   (SELECT
154      STRING_AGG(key || '\: ' || value, E'\n' ORDER BY ordering, key)
155    FROM
156      (SELECT key, ordering,
157         CASE WHEN :include_long_description = 'false' AND key = 'Description'
158           THEN SUBSTRING(value FROM E'\\A[^\n]*')
159           ELSE value
160         END AS value
161       FROM
162         binaries_metadata bm
163         JOIN metadata_keys mk ON mk.key_id = bm.key_id
164       WHERE
165         bm.bin_id = tmp.binary_id
166         AND key != ALL (:metadata_skip)
167      ) AS metadata
168   )
169   || COALESCE(E'\n' || (SELECT
170      STRING_AGG(key || '\: ' || value, E'\n' ORDER BY key)
171    FROM external_overrides eo
172    WHERE
173      eo.package = tmp.package
174      AND eo.suite = :overridesuite AND eo.component = :component
175   ), '')
176   || E'\nSection\: ' || sec.section
177   || E'\nPriority\: ' || pri.priority
178   || E'\nFilename\: pool/' || tmp.filename
179   || E'\nSize\: ' || tmp.size
180   || E'\nMD5sum\: ' || tmp.md5sum
181   || E'\nSHA1\: ' || tmp.sha1sum
182   || E'\nSHA256\: ' || tmp.sha256sum
183
184 FROM
185   tmp
186   JOIN override o ON o.package = tmp.package
187   JOIN section sec ON sec.id = o.section
188   JOIN priority pri ON pri.id = o.priority
189
190 WHERE
191   (
192       architecture <> :arch_all
193     OR
194       (architecture = :arch_all AND source_id IN (SELECT source_id FROM tmp WHERE architecture <> :arch_all))
195     OR
196       (architecture = :arch_all AND source NOT IN (SELECT DISTINCT source FROM tmp WHERE architecture <> :arch_all))
197   )
198   AND
199     o.type = :type_id AND o.suite = :overridesuite AND o.component = :component
200
201 ORDER BY tmp.source, tmp.package, tmp.version
202 """
203
204 def generate_packages(suite_id, component_id, architecture_id, type_name):
205     global _packages_query
206     from daklib.filewriter import PackagesFileWriter
207     from daklib.dbconn import Architecture, Component, DBConn, OverrideType, Suite
208     from daklib.dakmultiprocessing import PROC_STATUS_SUCCESS
209
210     session = DBConn().session()
211     arch_all_id = session.query(Architecture).filter_by(arch_string='all').one().arch_id
212     type_id = session.query(OverrideType).filter_by(overridetype=type_name).one().overridetype_id
213
214     suite = session.query(Suite).get(suite_id)
215     component = session.query(Component).get(component_id)
216     architecture = session.query(Architecture).get(architecture_id)
217
218     overridesuite_id = suite.get_overridesuite().suite_id
219     include_long_description = suite.include_long_description
220
221     # We currently filter out the "Tag" line. They are set by external
222     # overrides and NOT by the maintainer. And actually having it set by
223     # maintainer means we output it twice at the moment -> which breaks
224     # dselect.
225     metadata_skip = ["Section", "Priority", "Tag"]
226     if include_long_description:
227         metadata_skip.append("Description-md5")
228
229     writer = PackagesFileWriter(suite=suite.suite_name, component=component.component_name,
230             architecture=architecture.arch_string, debtype=type_name)
231     output = writer.open()
232
233     r = session.execute(_packages_query, {"suite": suite_id, "component": component_id,
234         "arch": architecture_id, "type_id": type_id, "type_name": type_name, "arch_all": arch_all_id,
235         "overridesuite": overridesuite_id, "metadata_skip": metadata_skip,
236         "include_long_description": 'true' if include_long_description else 'false'})
237     for (stanza,) in r:
238         print >>output, stanza
239         print >>output, ""
240
241     writer.close()
242
243     message = ["generate-packages", suite.suite_name, component.component_name, architecture.arch_string]
244     session.rollback()
245     return (PROC_STATUS_SUCCESS, message)
246
247 #############################################################################
248
249 _translations_query = """
250 WITH
251   override_suite AS
252     (SELECT
253       s.id AS id,
254       COALESCE(os.id, s.id) AS overridesuite_id
255       FROM suite AS s LEFT JOIN suite AS os ON s.overridesuite = os.suite_name)
256
257 SELECT
258      'Package\: ' || b.package
259   || E'\nDescription-md5\: ' || bm_description_md5.value
260   || E'\nDescription-en\: ' || bm_description.value
261   || E'\n'
262 FROM binaries b
263   -- join tables for suite and component
264   JOIN bin_associations ba ON b.id = ba.bin
265   JOIN override_suite os ON os.id = ba.suite
266   JOIN override o ON b.package = o.package AND o.suite = os.overridesuite_id AND o.type = (SELECT id FROM override_type WHERE type = 'deb')
267
268   -- join tables for Description and Description-md5
269   JOIN binaries_metadata bm_description ON b.id = bm_description.bin_id AND bm_description.key_id = (SELECT key_id FROM metadata_keys WHERE key = 'Description')
270   JOIN binaries_metadata bm_description_md5 ON b.id = bm_description_md5.bin_id AND bm_description_md5.key_id = (SELECT key_id FROM metadata_keys WHERE key = 'Description-md5')
271
272   -- we want to sort by source name
273   JOIN source s ON b.source = s.id
274
275 WHERE ba.suite = :suite AND o.component = :component
276 GROUP BY s.source, b.package, bm_description_md5.value, bm_description.value
277 ORDER BY s.source, b.package, bm_description_md5.value
278 """
279
280 def generate_translations(suite_id, component_id):
281     global _translations_query
282     from daklib.filewriter import TranslationFileWriter
283     from daklib.dbconn import DBConn, Suite, Component
284     from daklib.dakmultiprocessing import PROC_STATUS_SUCCESS
285
286     session = DBConn().session()
287     suite = session.query(Suite).get(suite_id)
288     component = session.query(Component).get(component_id)
289
290     writer = TranslationFileWriter(suite=suite.suite_name, component=component.component_name, language="en")
291     output = writer.open()
292
293     r = session.execute(_translations_query, {"suite": suite_id, "component": component_id})
294     for (stanza,) in r:
295         print >>output, stanza
296
297     writer.close()
298
299     message = ["generate-translations", suite.suite_name, component.component_name]
300     session.rollback()
301     return (PROC_STATUS_SUCCESS, message)
302
303 #############################################################################
304
305 def main():
306     from daklib.config import Config
307     from daklib import daklog
308
309     cnf = Config()
310
311     Arguments = [('h',"help","Generate-Packages-Sources::Options::Help"),
312                  ('s',"suite","Generate-Packages-Sources::Options::Suite"),
313                  ('f',"force","Generate-Packages-Sources::Options::Force"),
314                  ('o','option','','ArbItem')]
315
316     suite_names = apt_pkg.ParseCommandLine(cnf.Cnf, Arguments, sys.argv)
317     try:
318         Options = cnf.SubTree("Generate-Packages-Sources::Options")
319     except KeyError:
320         Options = {}
321
322     if Options.has_key("Help"):
323         usage()
324
325     from daklib.dakmultiprocessing import DakProcessPool, PROC_STATUS_SUCCESS, PROC_STATUS_SIGNALRAISED
326     pool = DakProcessPool()
327
328     logger = daklog.Logger('generate-packages-sources2')
329
330     from daklib.dbconn import Component, DBConn, get_suite, Suite
331     session = DBConn().session()
332     session.execute("SELECT add_missing_description_md5()")
333     session.commit()
334
335     if Options.has_key("Suite"):
336         suites = []
337         for s in suite_names:
338             suite = get_suite(s.lower(), session)
339             if suite:
340                 suites.append(suite)
341             else:
342                 print "I: Cannot find suite %s" % s
343                 logger.log(['Cannot find suite %s' % s])
344     else:
345         suites = session.query(Suite).filter(Suite.untouchable == False).all()
346
347     force = Options.has_key("Force") and Options["Force"]
348
349     component_ids = [ c.component_id for c in session.query(Component).all() ]
350
351     def parse_results(message):
352         # Split out into (code, msg)
353         code, msg = message
354         if code == PROC_STATUS_SUCCESS:
355             logger.log([msg])
356         elif code == PROC_STATUS_SIGNALRAISED:
357             logger.log(['E: Subprocess recieved signal ', msg])
358         else:
359             logger.log(['E: ', msg])
360
361     for s in suites:
362         if s.untouchable and not force:
363             import utils
364             utils.fubar("Refusing to touch %s (untouchable and not forced)" % s.suite_name)
365         for c in component_ids:
366             pool.apply_async(generate_sources, [s.suite_id, c], callback=parse_results)
367             if not s.include_long_description:
368                 pool.apply_async(generate_translations, [s.suite_id, c], callback=parse_results)
369             for a in s.architectures:
370                 if a == 'source':
371                     continue
372                 pool.apply_async(generate_packages, [s.suite_id, c, a.arch_id, 'deb'], callback=parse_results)
373                 pool.apply_async(generate_packages, [s.suite_id, c, a.arch_id, 'udeb'], callback=parse_results)
374
375     pool.close()
376     pool.join()
377
378     # this script doesn't change the database
379     session.close()
380
381     logger.close()
382
383     sys.exit(pool.overall_status())
384
385 if __name__ == '__main__':
386     main()