4 Create all the Release files
6 @contact: Debian FTPMaster <ftpmaster@debian.org>
7 @copyright: 2011 Joerg Jaspert <joerg@debian.org>
8 @copyright: 2011 Mark Hymers <mhy@debian.org>
9 @license: GNU General Public License version 2 or later
13 # This program is free software; you can redistribute it and/or modify
14 # it under the terms of the GNU General Public License as published by
15 # the Free Software Foundation; either version 2 of the License, or
16 # (at your option) any later version.
18 # This program is distributed in the hope that it will be useful,
19 # but WITHOUT ANY WARRANTY; without even the implied warranty of
20 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
21 # GNU General Public License for more details.
23 # You should have received a copy of the GNU General Public License
24 # along with this program; if not, write to the Free Software
25 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
27 ################################################################################
29 # <mhy> I wish they wouldnt leave biscuits out, thats just tempting. Damnit.
31 ################################################################################
41 from tempfile import mkstemp, mkdtemp
43 from multiprocessing import Pool, TimeoutError
44 from sqlalchemy.orm import object_session
46 from daklib import utils, daklog
47 from daklib.regexes import re_gensubrelease, re_includeinrelease
48 from daklib.dak_exceptions import *
49 from daklib.dbconn import *
50 from daklib.config import Config
52 ################################################################################
53 Logger = None #: Our logging object
54 results = [] #: Results of the subprocesses
56 ################################################################################
58 def usage (exit_code=0):
59 """ Usage information"""
61 print """Usage: dak generate-releases [OPTIONS]
62 Generate the Release files
64 -s, --suite=SUITE(s) process this suite
65 Default: All suites not marked 'untouchable'
66 -f, --force Allow processing of untouchable suites
67 CAREFUL: Only to be used at (point) release time!
68 -h, --help show this help and exit
70 SUITE can be a space seperated list, e.g.
71 --suite=unstable testing
75 ########################################################################
82 def sign_release_dir(dirname):
85 if cnf.has_key("Dinstall::SigningKeyring"):
86 keyring = "--secret-keyring \"%s\"" % cnf["Dinstall::SigningKeyring"]
87 if cnf.has_key("Dinstall::SigningPubKeyring"):
88 keyring += " --keyring \"%s\"" % cnf["Dinstall::SigningPubKeyring"]
90 arguments = "--no-options --batch --no-tty --armour"
91 signkeyids = cnf.signingkeyids.split()
93 relname = os.path.join(dirname, 'Release')
95 dest = os.path.join(dirname, 'Release.gpg')
96 if os.path.exists(dest):
99 inlinedest = os.path.join(dirname, 'InRelease')
100 if os.path.exists(inlinedest):
101 os.unlink(inlinedest)
103 for keyid in signkeyids:
105 defkeyid = "--default-key %s" % keyid
109 os.system("gpg %s %s %s --detach-sign <%s >>%s" %
110 (keyring, defkeyid, arguments, relname, dest))
112 os.system("gpg %s %s %s --clearsign <%s >>%s" %
113 (keyring, defkeyid, arguments, relname, inlinedest))
115 class ReleaseWriter(object):
116 def __init__(self, suite):
119 def generate_release_files(self):
121 Generate Release files for the given suite
124 @param suite: Suite name
128 session = object_session(suite)
130 architectures = get_suite_architectures(suite.suite_name, skipall=True, skipsrc=True, session=session)
132 # Attribs contains a tuple of field names and the database names to use to
134 attribs = ( ('Origin', 'origin'),
136 ('Suite', 'suite_name'),
137 ('Version', 'version'),
138 ('Codename', 'codename') )
140 # A "Sub" Release file has slightly different fields
141 subattribs = ( ('Origin', 'origin'),
143 ('Archive', 'suite_name'),
144 ('Version', 'version') )
146 # Boolean stuff. If we find it true in database, write out "yes" into the release file
147 boolattrs = ( ('NotAutomatic', 'notautomatic'),
148 ('ButAutomaticUpgrades', 'butautomaticupgrades') )
152 outfile = os.path.join(cnf["Dir::Root"], 'dists', suite.suite_name, "Release")
154 out = open(outfile, "w")
156 for key, dbfield in attribs:
157 if getattr(suite, dbfield) is not None:
158 out.write("%s: %s\n" % (key, getattr(suite, dbfield)))
160 out.write("Date: %s\n" % (time.strftime("%a, %d %b %Y %H:%M:%S UTC", time.gmtime(time.time()))))
163 validtime=float(suite.validtime)
164 out.write("Valid-Until: %s\n" % (time.strftime("%a, %d %b %Y %H:%M:%S UTC", time.gmtime(time.time()+validtime))))
166 for key, dbfield in boolattrs:
167 if getattr(suite, dbfield, False):
168 out.write("%s: yes\n" % (key))
170 out.write("Architectures: %s\n" % (" ".join([a.arch_string for a in architectures])))
172 suite_suffix = "%s" % (cnf.Find("Dinstall::SuiteSuffix"))
174 ## FIXME: Components need to be adjusted to whatever will be in the db
175 ## Needs putting in the DB
176 components = ['main', 'contrib', 'non-free']
178 out.write("Components: %s\n" % ( " ".join(map(lambda x: "%s%s" % (suite_suffix, x), components ))))
180 # For exact compatibility with old g-r, write out Description here instead
181 # of with the rest of the DB fields above
182 if getattr(suite, 'description') is not None:
183 out.write("Description: %s\n" % suite.description)
185 for comp in components:
186 for dirpath, dirnames, filenames in os.walk("%sdists/%s/%s" % (cnf["Dir::Root"], suite.suite_name, comp), topdown=True):
187 if not re_gensubrelease.match(dirpath):
190 subfile = os.path.join(dirpath, "Release")
191 subrel = open(subfile, "w")
193 for key, dbfield in subattribs:
194 if getattr(suite, dbfield) is not None:
195 subrel.write("%s: %s\n" % (key, getattr(suite, dbfield)))
197 for key, dbfield in boolattrs:
198 if getattr(suite, dbfield, False):
199 subrel.write("%s: yes\n" % (key))
201 subrel.write("Component: %s%s\n" % (suite_suffix, comp))
204 # Now that we have done the groundwork, we want to get off and add the files with
205 # their checksums to the main Release file
208 os.chdir("%sdists/%s" % (cnf["Dir::Root"], suite.suite_name))
210 hashfuncs = { 'MD5Sum' : apt_pkg.md5sum,
211 'SHA1' : apt_pkg.sha1sum,
212 'SHA256' : apt_pkg.sha256sum }
218 for dirpath, dirnames, filenames in os.walk(".", followlinks=True, topdown=True):
219 for entry in filenames:
220 # Skip things we don't want to include
221 if not re_includeinrelease.match(entry):
224 if dirpath == '.' and entry in ["Release", "Release.gpg", "InRelease"]:
227 filename = os.path.join(dirpath.lstrip('./'), entry)
228 fileinfo[filename] = {}
229 contents = open(filename, 'r').read()
231 # If we find a file for which we have a compressed version and
232 # haven't yet seen the uncompressed one, store the possibility
234 if entry.endswith(".gz") and entry[:-3] not in uncompnotseen.keys():
235 uncompnotseen[filename[:-3]] = (gzip.GzipFile, filename)
236 elif entry.endswith(".bz2") and entry[:-4] not in uncompnotseen.keys():
237 uncompnotseen[filename[:-4]] = (bz2.BZ2File, filename)
239 fileinfo[filename]['len'] = len(contents)
241 for hf, func in hashfuncs.items():
242 fileinfo[filename][hf] = func(contents)
244 for filename, comp in uncompnotseen.items():
245 # If we've already seen the uncompressed file, we don't
246 # need to do anything again
247 if filename in fileinfo.keys():
250 # Skip uncompressed Contents files as they're huge, take ages to
251 # checksum and we checksum the compressed ones anyways
252 if os.path.basename(filename).startswith("Contents"):
255 fileinfo[filename] = {}
257 # File handler is comp[0], filename of compressed file is comp[1]
258 contents = comp[0](comp[1], 'r').read()
260 fileinfo[filename]['len'] = len(contents)
262 for hf, func in hashfuncs.items():
263 fileinfo[filename][hf] = func(contents)
266 for h in sorted(hashfuncs.keys()):
267 out.write('%s:\n' % h)
268 for filename in sorted(fileinfo.keys()):
269 out.write(" %s %8d %s\n" % (fileinfo[filename][h], fileinfo[filename]['len'], filename))
273 sign_release_dir(os.path.dirname(outfile))
281 global Logger, results
285 for i in ["Help", "Suite", "Force"]:
286 if not cnf.has_key("Generate-Releases::Options::%s" % (i)):
287 cnf["Generate-Releases::Options::%s" % (i)] = ""
289 Arguments = [('h',"help","Generate-Releases::Options::Help"),
290 ('s',"suite","Generate-Releases::Options::Suite"),
291 ('f',"force","Generate-Releases::Options::Force")]
293 suite_names = apt_pkg.ParseCommandLine(cnf.Cnf, Arguments, sys.argv)
294 Options = cnf.SubTree("Generate-Releases::Options")
299 Logger = daklog.Logger(cnf, 'generate-releases')
301 session = DBConn().session()
305 for s in suite_names:
306 suite = get_suite(s.lower(), session)
310 print "cannot find suite %s" % s
311 Logger.log(['cannot find suite %s' % s])
313 suites = session.query(Suite).filter(Suite.untouchable == False).all()
316 # For each given suite, run one process
322 # Setup a multiprocessing Pool. As many workers as we have CPU cores.
323 if s.untouchable and not Options["Force"]:
324 print "Skipping %s (untouchable)" % s.suite_name
327 print "Processing %s" % s.suite_name
328 Logger.log(['Processing release file for Suite: %s' % (s.suite_name)])
329 pool.apply_async(generate_helper, (s.suite_id, ), callback=get_result)
331 # No more work will be added to our pool, close it and then wait for all to finish
338 Logger.log(['Release file generation broken: %s' % (results)])
339 print "Release file generation broken:\n", '\n'.join(results)
346 def generate_helper(suite_id):
348 This function is called in a new subprocess.
350 session = DBConn().session()
351 suite = Suite.get(suite_id, session)
353 rw = ReleaseWriter(suite)
354 rw.generate_release_files()
360 #######################################################################################
362 if __name__ == '__main__':