dak/generate_releases.py

   1 #!/usr/bin/env python
   2
   3 """ Create all the Release files """
   4
   5 # Copyright (C) 2001, 2002, 2006  Anthony Towns <ajt@debian.org>
   6
   7 # This program is free software; you can redistribute it and/or modify
   8 # it under the terms of the GNU General Public License as published by
   9 # the Free Software Foundation; either version 2 of the License, or
  10 # (at your option) any later version.
  11
  12 # This program is distributed in the hope that it will be useful,
  13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15 # GNU General Public License for more details.
  16
  17 # You should have received a copy of the GNU General Public License
  18 # along with this program; if not, write to the Free Software
  19 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  20
  21 #   ``Bored now''
  22
  23 ################################################################################
  24
  25 import sys, os, stat, time, pg
  26 import gzip, bz2
  27 import apt_pkg
  28 from daklib import utils
  29 from daklib import database
  30 from daklib.dak_exceptions import *
  31
  32 ################################################################################
  33
  34 Cnf = None
  35 projectB = None
  36 out = None
  37 AptCnf = None
  38
  39 ################################################################################
  40
  41 def usage (exit_code=0):
  42     print """Usage: dak generate-releases [OPTION]... [SUITE]...
  43 Generate Release files (for SUITE).
  44
  45   -h, --help                 show this help and exit
  46   -a, --apt-conf FILE        use FILE instead of default apt.conf
  47   -f, --force-touch          ignore Untouchable directives in dak.conf
  48
  49 If no SUITE is given Release files are generated for all suites."""
  50
  51     sys.exit(exit_code)
  52
  53 ################################################################################
  54
  55 def add_tiffani (files, path, indexstem):
  56     index = "%s.diff/Index" % (indexstem)
  57     filepath = "%s/%s" % (path, index)
  58     if os.path.exists(filepath):
  59         #print "ALERT: there was a tiffani file %s" % (filepath)
  60         files.append(index)
  61
  62 def compressnames (tree,type,file):
  63     compress = AptCnf.get("%s::%s::Compress" % (tree,type), AptCnf.get("Default::%s::Compress" % (type), ". gzip"))
  64     result = []
  65     cl = compress.split()
  66     uncompress = ("." not in cl)
  67     for mode in compress.split():
  68         if mode == ".":
  69             result.append(file)
  70         elif mode == "gzip":
  71             if uncompress:
  72                 result.append("<zcat/.gz>" + file)
  73                 uncompress = 0
  74             result.append(file + ".gz")
  75         elif mode == "bzip2":
  76             if uncompress:
  77                 result.append("<bzcat/.bz2>" + file)
  78                 uncompress = 0
  79             result.append(file + ".bz2")
  80     return result
  81
  82 decompressors = { 'zcat' : gzip.GzipFile,
  83                   'bzip2' : bz2.BZ2File }
  84
  85 def print_md5sha_files (tree, files, hashop):
  86     path = Cnf["Dir::Root"] + tree + "/"
  87     for name in files:
  88         hashvalue = ""
  89         hashlen = 0
  90         try:
  91             if name[0] == "<":
  92                 j = name.index("/")
  93                 k = name.index(">")
  94                 (cat, ext, name) = (name[1:j], name[j+1:k], name[k+1:])
  95                 file_handle = decompressors[ cat ]( "%s%s%s" % (path, name, ext) )
  96                 contents = file_handle.read()
  97                 hashvalue = hashop(contents)
  98                 hashlen = len(contents)
  99             else:
 100                 try:
 101                     file_handle = utils.open_file(path + name)
 102                     hashvalue = hashop(file_handle)
 103                     hashlen = os.stat(path + name).st_size
 104                 except:
 105                     raise
 106                 else:
 107                     if file_handle:
 108                         file_handle.close()
 109
 110         except CantOpenError:
 111             print "ALERT: Couldn't open " + path + name
 112         else:
 113             out.write(" %s %8d %s\n" % (hashvalue, hashlen, name))
 114
 115 def print_md5_files (tree, files):
 116     print_md5sha_files (tree, files, apt_pkg.md5sum)
 117
 118 def print_sha1_files (tree, files):
 119     print_md5sha_files (tree, files, apt_pkg.sha1sum)
 120
 121 def print_sha256_files (tree, files):
 122     print_md5sha_files (tree, files, apt_pkg.sha256sum)
 123
 124 ################################################################################
 125
 126 def main ():
 127     global Cnf, AptCnf, projectB, out
 128     out = sys.stdout
 129
 130     Cnf = utils.get_conf()
 131
 132     Arguments = [('h',"help","Generate-Releases::Options::Help"),
 133                  ('a',"apt-conf","Generate-Releases::Options::Apt-Conf", "HasArg"),
 134                  ('f',"force-touch","Generate-Releases::Options::Force-Touch"),
 135                 ]
 136     for i in [ "help", "apt-conf", "force-touch" ]:
 137         if not Cnf.has_key("Generate-Releases::Options::%s" % (i)):
 138             Cnf["Generate-Releases::Options::%s" % (i)] = ""
 139
 140     suites = apt_pkg.ParseCommandLine(Cnf,Arguments,sys.argv)
 141     Options = Cnf.SubTree("Generate-Releases::Options")
 142
 143     if Options["Help"]:
 144         usage()
 145
 146     if not Options["Apt-Conf"]:
 147         Options["Apt-Conf"] = utils.which_apt_conf_file()
 148
 149     AptCnf = apt_pkg.newConfiguration()
 150     apt_pkg.ReadConfigFileISC(AptCnf, Options["Apt-Conf"])
 151
 152     projectB = pg.connect(Cnf["DB::Name"], Cnf["DB::Host"], int(Cnf["DB::Port"]))
 153     database.init(Cnf, projectB)
 154
 155     if not suites:
 156         suites = Cnf.SubTree("Suite").List()
 157
 158     for suite in suites:
 159         print "Processing: " + suite
 160         SuiteBlock = Cnf.SubTree("Suite::" + suite)
 161
 162         if SuiteBlock.has_key("Untouchable") and not Options["Force-Touch"]:
 163             print "Skipping: " + suite + " (untouchable)"
 164             continue
 165
 166         suite = suite.lower()
 167
 168         origin = SuiteBlock["Origin"]
 169         label = SuiteBlock.get("Label", origin)
 170         codename = SuiteBlock.get("CodeName", "")
 171
 172         version = ""
 173         description = ""
 174
 175         q = projectB.query("SELECT version, description FROM suite WHERE suite_name = '%s'" % (suite))
 176         qs = q.getresult()
 177         if len(qs) == 1:
 178             if qs[0][0] != "-": version = qs[0][0]
 179             if qs[0][1]: description = qs[0][1]
 180
 181         architectures = database.get_suite_architectures(suite)
 182         if architectures == None:
 183             architectures = []
 184
 185         if SuiteBlock.has_key("NotAutomatic"):
 186             notautomatic = "yes"
 187         else:
 188             notautomatic = ""
 189
 190         if SuiteBlock.has_key("Components"):
 191             components = SuiteBlock.ValueList("Components")
 192         else:
 193             components = []
 194
 195         suite_suffix = Cnf.Find("Dinstall::SuiteSuffix")
 196         if components and suite_suffix:
 197             longsuite = suite + "/" + suite_suffix
 198         else:
 199             longsuite = suite
 200
 201         tree = SuiteBlock.get("Tree", "dists/%s" % (longsuite))
 202
 203         if AptCnf.has_key("tree::%s" % (tree)):
 204             pass
 205         elif AptCnf.has_key("bindirectory::%s" % (tree)):
 206             pass
 207         else:
 208             aptcnf_filename = os.path.basename(utils.which_apt_conf_file())
 209             print "ALERT: suite %s not in %s, nor untouchable!" % (suite, aptcnf_filename)
 210             continue
 211
 212         print Cnf["Dir::Root"] + tree + "/Release"
 213         out = open(Cnf["Dir::Root"] + tree + "/Release", "w")
 214
 215         out.write("Origin: %s\n" % (origin))
 216         out.write("Label: %s\n" % (label))
 217         out.write("Suite: %s\n" % (suite))
 218         if version != "":
 219             out.write("Version: %s\n" % (version))
 220         if codename != "":
 221             out.write("Codename: %s\n" % (codename))
 222         out.write("Date: %s\n" % (time.strftime("%a, %d %b %Y %H:%M:%S UTC", time.gmtime(time.time()))))
 223
 224         if SuiteBlock.has_key("ValidTime"):
 225             validtime=float(SuiteBlock["ValidTime"])
 226             out.write("Valid-Until: %s\n" % (time.strftime("%a, %d %b %Y %H:%M:%S UTC", time.gmtime(time.time()+validtime))))
 227
 228         if notautomatic != "":
 229             out.write("NotAutomatic: %s\n" % (notautomatic))
 230         out.write("Architectures: %s\n" % (" ".join(filter(utils.real_arch, architectures))))
 231         if components:
 232             out.write("Components: %s\n" % (" ".join(components)))
 233
 234         if description:
 235             out.write("Description: %s\n" % (description))
 236
 237         files = []
 238
 239         if AptCnf.has_key("tree::%s" % (tree)):
 240             if AptCnf.has_key("tree::%s::Contents" % (tree)):
 241                 pass
 242             else:
 243                 for x in os.listdir("%s/%s" % (Cnf["Dir::Root"], tree)):
 244                     if x.startswith('Contents-'):
 245                         files.append(x)
 246
 247             for sec in AptCnf["tree::%s::Sections" % (tree)].split():
 248                 for arch in AptCnf["tree::%s::Architectures" % (tree)].split():
 249                     if arch == "source":
 250                         filepath = "%s/%s/Sources" % (sec, arch)
 251                         for cfile in compressnames("tree::%s" % (tree), "Sources", filepath):
 252                             files.append(cfile)
 253                         add_tiffani(files, Cnf["Dir::Root"] + tree, filepath)
 254                     else:
 255                         disks = "%s/disks-%s" % (sec, arch)
 256                         diskspath = Cnf["Dir::Root"]+tree+"/"+disks
 257                         if os.path.exists(diskspath):
 258                             for dir in os.listdir(diskspath):
 259                                 if os.path.exists("%s/%s/md5sum.txt" % (diskspath, dir)):
 260                                     files.append("%s/%s/md5sum.txt" % (disks, dir))
 261
 262                         filepath = "%s/binary-%s/Packages" % (sec, arch)
 263                         for cfile in compressnames("tree::%s" % (tree), "Packages", filepath):
 264                             files.append(cfile)
 265                         add_tiffani(files, Cnf["Dir::Root"] + tree, filepath)
 266
 267                     if arch == "source":
 268                         rel = "%s/%s/Release" % (sec, arch)
 269                     else:
 270                         rel = "%s/binary-%s/Release" % (sec, arch)
 271                     relpath = Cnf["Dir::Root"]+tree+"/"+rel
 272
 273                     try:
 274                         if os.access(relpath, os.F_OK):
 275                             if os.stat(relpath).st_nlink > 1:
 276                                 os.unlink(relpath)
 277                         release = open(relpath, "w")
 278                         #release = open(longsuite.replace("/","_") + "_" + arch + "_" + sec + "_Release", "w")
 279                     except IOError:
 280                         utils.fubar("Couldn't write to " + relpath)
 281
 282                     release.write("Archive: %s\n" % (suite))
 283                     if version != "":
 284                         release.write("Version: %s\n" % (version))
 285                     if suite_suffix:
 286                         release.write("Component: %s/%s\n" % (suite_suffix,sec))
 287                     else:
 288                         release.write("Component: %s\n" % (sec))
 289                     release.write("Origin: %s\n" % (origin))
 290                     release.write("Label: %s\n" % (label))
 291                     if notautomatic != "":
 292                         release.write("NotAutomatic: %s\n" % (notautomatic))
 293                     release.write("Architecture: %s\n" % (arch))
 294                     release.close()
 295                     files.append(rel)
 296
 297             if AptCnf.has_key("tree::%s/main" % (tree)):
 298                 for dis in ["main", "contrib", "non-free"]:
 299                     if not AptCnf.has_key("tree::%s/%s" % (tree, dis)): continue
 300                     sec = AptCnf["tree::%s/%s::Sections" % (tree,dis)].split()[0]
 301                     if sec != "debian-installer":
 302                         print "ALERT: weird non debian-installer section in %s" % (tree)
 303
 304                     for arch in AptCnf["tree::%s/%s::Architectures" % (tree,dis)].split():
 305                         if arch != "source":  # always true
 306                             for cfile in compressnames("tree::%s/%s" % (tree,dis),
 307                                 "Packages",
 308                                 "%s/%s/binary-%s/Packages" % (dis, sec, arch)):
 309                                 files.append(cfile)
 310             elif AptCnf.has_key("tree::%s::FakeDI" % (tree)):
 311                 usetree = AptCnf["tree::%s::FakeDI" % (tree)]
 312                 sec = AptCnf["tree::%s/main::Sections" % (usetree)].split()[0]
 313                 if sec != "debian-installer":
 314                     print "ALERT: weird non debian-installer section in %s" % (usetree)
 315
 316                 for arch in AptCnf["tree::%s/main::Architectures" % (usetree)].split():
 317                     if arch != "source":  # always true
 318                         for cfile in compressnames("tree::%s/main" % (usetree), "Packages", "main/%s/binary-%s/Packages" % (sec, arch)):
 319                             files.append(cfile)
 320
 321         elif AptCnf.has_key("bindirectory::%s" % (tree)):
 322             for cfile in compressnames("bindirectory::%s" % (tree), "Packages", AptCnf["bindirectory::%s::Packages" % (tree)]):
 323                 files.append(cfile.replace(tree+"/","",1))
 324             for cfile in compressnames("bindirectory::%s" % (tree), "Sources", AptCnf["bindirectory::%s::Sources" % (tree)]):
 325                 files.append(cfile.replace(tree+"/","",1))
 326         else:
 327             print "ALERT: no tree/bindirectory for %s" % (tree)
 328
 329         out.write("MD5Sum:\n")
 330         print_md5_files(tree, files)
 331         out.write("SHA1:\n")
 332         print_sha1_files(tree, files)
 333         out.write("SHA256:\n")
 334         print_sha256_files(tree, files)
 335
 336         out.close()
 337         if Cnf.has_key("Dinstall::SigningKeyring"):
 338             keyring = "--secret-keyring \"%s\"" % Cnf["Dinstall::SigningKeyring"]
 339             if Cnf.has_key("Dinstall::SigningPubKeyring"):
 340                 keyring += " --keyring \"%s\"" % Cnf["Dinstall::SigningPubKeyring"]
 341
 342             arguments = "--no-options --batch --no-tty --armour"
 343             if Cnf.has_key("Dinstall::SigningKeyIds"):
 344                 signkeyids = Cnf["Dinstall::SigningKeyIds"].split()
 345             else:
 346                 signkeyids = [""]
 347
 348             dest = Cnf["Dir::Root"] + tree + "/Release.gpg"
 349             if os.path.exists(dest):
 350                 os.unlink(dest)
 351
 352             for keyid in signkeyids:
 353                 if keyid != "": defkeyid = "--default-key %s" % keyid
 354                 else: defkeyid = ""
 355                 os.system("gpg %s %s %s --detach-sign <%s >>%s" %
 356                         (keyring, defkeyid, arguments,
 357                         Cnf["Dir::Root"] + tree + "/Release", dest))
 358
 359 #######################################################################################
 360
 361 if __name__ == '__main__':
 362     main()