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