bin/pfa-signpkg

   1 #!/usr/bin/env python
   2 # vi: encoding=utf-8 ts=8 sts=4 sw=4 et
   3
   4 import sys, os
   5 import getopt
   6 sys.path.insert(0, os.environ['HOME']+'/tld-ftp-admin/modules')
   7 import ftptree
   8 import getpass
   9 from common import checkdir
  10 import ftpio
  11 from config import sign_key
  12 from sign import is_signed, signpkgs
  13
  14 os.umask(022)
  15
  16 try:
  17     opts, args = getopt.getopt(sys.argv[1:], '')
  18 except getopt.GetoptError:
  19     print >>sys.stderr, "ERR: options error"
  20     print >>sys.stderr, "sign.py tree package1 [package2...]"
  21     sys.exit(1)
  22
  23 if len(args) < 1:
  24     print >>sys.stderr, "ERR: missing tree name"
  25     print >>sys.stderr, "sign.py tree package1 [package2...]"
  26     sys.exit(1)
  27
  28 if sign_key == None:
  29     print >>sys.stderr, "ERR: sign_key not defined in config"
  30     sys.exit(1)
  31
  32 treename = args[0]
  33 packages = args[1:]
  34
  35 checkdir(treename)
  36
  37 ftpio.connect('sign')
  38
  39 if not ftpio.lock(treename, True):
  40     print >>sys.stderr, "ERR: %s tree already locked" % treename
  41     sys.exit(1)
  42
  43 files = []
  44 try:
  45     if len(packages) < 1:
  46         loadall = True
  47     else:
  48         loadall = False
  49
  50     # if no files specified, grab whole tree contents
  51     tree = ftptree.FtpTree(treename, loadall = loadall)
  52     if loadall:
  53         # this is hack, should be a param, not access private .loadedpkgs element
  54         tree.mark4moving(tree.loadedpkgs)
  55     else:
  56         tree.mark4moving(packages)
  57     files = tree.rpmfiles()
  58
  59 except ftptree.SomeError:
  60     # In case of problems we need to unlock the tree before exiting
  61     ftpio.unlock(treename)
  62     sys.exit(1)
  63
  64 ftpio.unlock(treename)
  65
  66 print "Checking signatures of %d files from %d packages" % (len(files), len(tree.loadedpkgs))
  67 sign = []
  68 n = c = 0
  69 for file in files:
  70     if not is_signed(file):
  71         sign.append(file)
  72         c += 1
  73     n += 1
  74     print "\r%d/%d %s\033[0K" % (n, c, file),
  75
  76 print ""
  77
  78 if len(sign) == 0:
  79     print "No files to sign"
  80     sys.exit(0)
  81
  82 # http://mail.python.org/pipermail/python-list/2009-February/700658.html
  83 def chunk(seq, size, pad=None):
  84      '''
  85      Slice a list into consecutive disjoint 'chunks' of
  86      length equal to size. The last chunk is padded if necessary.
  87
  88      >>> list(chunk(range(1,10),3))
  89      [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
  90      >>> list(chunk(range(1,9),3))
  91      [[1, 2, 3], [4, 5, 6], [7, 8, None]]
  92      >>> list(chunk(range(1,8),3))
  93      [[1, 2, 3], [4, 5, 6], [7, None, None]]
  94      >>> list(chunk(range(1,10),1))
  95      [[1], [2], [3], [4], [5], [6], [7], [8], [9]]
  96      >>> list(chunk(range(1,10),9))
  97      [[1, 2, 3, 4, 5, 6, 7, 8, 9]]
  98      >>> for X in chunk([],3): print X
  99      >>>
 100      '''
 101      n = len(seq)
 102      mod = n % size
 103      for i in xrange(0, n - mod, size):
 104          yield seq[i : i + size]
 105      if mod:
 106          yield seq[-mod:]
 107
 108 print "Total %d files to sign" % len(sign)
 109 password = getpass.getpass("Enter signing password: ")
 110 try:
 111     for x in chunk(sign, 512):
 112         print "Signing %d files" % len(x)
 113         signpkgs(x, password)
 114 except OSError, e:
 115     print >>sys.stderr, "ERR: %s" % e
 116     exit(1)