generate-releases

[dak.git] / dak / generate_releases.py

diff --git a/dak/generate_releases.py b/dak/generate_releases.py
index 5644b957646e5280dbeedfc5230b199ad2e4f77d..3d42c51cc40df89d976068b2dd46390d3931f331 100755 (executable)
--- a/dak/generate_releases.py
+++ b/dak/generate_releases.py
@@ -108,6 +108,10 @@ def compressnames (tree,type,file):
 decompressors = { 'zcat' : gzip.GzipFile,
                   'bzip2' : bz2.BZ2File }
 
+hashfuncs = { 'MD5' : apt_pkg.md5sum,
+              'SHA1' : apt_pkg.sha1sum,
+              'SHA256' : apt_pkg.sha256sum }
+
 def print_hash_files (tree, files, hashop):
     path = Cnf["Dir::Root"] + tree + "/"
     for name in files:
@@ -120,12 +124,12 @@ def print_hash_files (tree, files, hashop):
                 (cat, ext, name) = (name[1:j], name[j+1:k], name[k+1:])
                 file_handle = decompressors[ cat ]( "%s%s%s" % (path, name, ext) )
                 contents = file_handle.read()
-                hashvalue = hashop(contents)
+                hashvalue = hashfuncs[ hashop ](contents)
                 hashlen = len(contents)
             else:
                 try:
                     file_handle = utils.open_file(path + name)
-                    hashvalue = hashop(file_handle)
+                    hashvalue = hashfuncs[ hashop ](file_handle)
                     hashlen = os.stat(path + name).st_size
                 except:
                     raise
@@ -141,12 +145,6 @@ def print_hash_files (tree, files, hashop):
         else:
             out.write(" %s %8d %s\n" % (hashvalue, hashlen, name))
 
-def print_sha1_files (tree, files):
-    print_hash_files (tree, files, apt_pkg.sha1sum)
-
-def print_sha256_files (tree, files):
-    print_hash_files (tree, files, apt_pkg.sha256sum)
-
 def write_release_file (relpath, suite, component, origin, label, arch, version="", suite_suffix="", notautomatic="", butautomaticupgrades=""):
     try:
         if os.access(relpath, os.F_OK):
@@ -368,10 +366,10 @@ def main ():
         else:
             print "ALERT: no tree/bindirectory for %s" % (tree)
 
-        out.write("SHA1:\n")
-        print_sha1_files(tree, files)
-        out.write("SHA256:\n")
-        print_sha256_files(tree, files)
+        for hashvalue in cnf.SubTree("Generate-Releases").List():
+            if suite in [ i.lower() for i in cnf.ValueList("Generate-Releases::%s" % (hashvalue)) ]:
+                out.write("%s:\n" % (hashvalue))
+                print_hash_files(tree, files, hashvalue)
 
         out.close()
         if Cnf.has_key("Dinstall::SigningKeyring"):