- catch signpkg errors

[projects/pld-ftp-admin.git] / bin / pfa-signpkg

#!/usr/bin/env python
# vi: encoding=utf-8 ts=8 sts=4 sw=4 et

import sys, os
sys.path.insert(0, os.environ['HOME']+'/pld-ftp-admin/modules')
import ftptree
from common import checkdir
import ftpio
from config import sign_key
from sign import is_signed, signpkgs

if len(sys.argv) < 3:
    print >>sys.stderr, "ERR: not enough parameters given"
    print >>sys.stderr, "sign.py tree package1 [package2...]"
    sys.exit(1)

if sign_key == None:
    print >>sys.stderr, "ERR: sign_key not defined in config"
    sys.exit(1)

checkdir(sys.argv[1])

ftpio.connect('sign')

if not ftpio.lock(sys.argv[1], True):
    print >>sys.stderr, "ERR: %s tree already locked" % sys.argv[1]
    sys.exit(1)

files = []
try:
    tree = ftptree.FtpTree(sys.argv[1])
    tree.mark4moving(sys.argv[2:])
    files = tree.rpmfiles()

except ftptree.SomeError:
    # In case of problems we need to unlock the tree before exiting
    ftpio.unlock(sys.argv[1])
    sys.exit(1)

ftpio.unlock(sys.argv[1])

print "Checking signatures of %d files from %d packages" % (len(files), len(tree.loadedpkgs))
sign = []
for file in files:
    if not is_signed(file):
        sign.append(file)

if len(sign) == 0:
    print "No files to sign"
    sys.exit(0)

print "Signing %d files" % len(sign)
try:
    signpkgs(sign)
except OSError, e:
    print >>sys.stderr, "ERR: %s" % e
    exit(1)