diff options
| -rwxr-xr-x | src/build-data/scripts/dist.py | 168 |
1 files changed, 168 insertions, 0 deletions
diff --git a/src/build-data/scripts/dist.py b/src/build-data/scripts/dist.py new file mode 100755 index 000000000..8a69e7aa6 --- /dev/null +++ b/src/build-data/scripts/dist.py @@ -0,0 +1,168 @@ +#!/usr/bin/python + +import optparse +import subprocess +import logging +import os +import sys +import shutil +import tarfile +import errno + +def check_subprocess_results(subproc, name): + (stdout, stderr) = subproc.communicate() + + stdout = stdout.strip() + stderr = stderr.strip() + + if subproc.returncode != 0: + if stdout != '': + logging.error(stdout) + if stderr != '': + logging.error(stderr) + raise Exception('Running %s failed' % (name)) + else: + if stderr != '': + logging.debug(stderr) + + return stdout + +def run_monotone(db, args): + mtn = subprocess.Popen(['mtn', '--db', db] + args, + stdout=subprocess.PIPE, + stderr=subprocess.PIPE) + + return check_subprocess_results(mtn, 'mtn') + +def gpg_sign(file, keyid): + print file + gpg = subprocess.Popen(['gpg', '--armor', '--detach-sign', + '--local-user', keyid, file], + stdout=subprocess.PIPE, + stderr=subprocess.PIPE) + + check_subprocess_results(gpg, 'gpg') + +def parse_args(args): + parser = optparse.OptionParser() + parser.add_option('--verbose', action='store_true', + default=False, help='Extra debug output') + + parser.add_option('--output-dir', metavar='DIR', + default='.', + help='Where to place output (default %default)') + + parser.add_option('--mtn-db', metavar='DB', + default=os.getenv('BOTAN_MTN_DB', ''), + help='Set monotone db (default \'%default\')') + + parser.add_option('--pgp-key-id', metavar='KEYID', + default='EFBADFBC', + help='PGP signing key (default %default)') + + return parser.parse_args(args) + +def remove_file_if_exists(fspath): + try: + os.unlink(fspath) + except OSError, e: + if e.errno != errno.ENOENT: + raise + +def main(args = None): + if args is None: + args = sys.argv[1:] + + (options, args) = parse_args(args) + + def log_level(): + if options.verbose: + return logging.DEBUG + return logging.INFO + + logging.basicConfig(stream = sys.stdout, + format = '%(levelname) 7s: %(message)s', + level = log_level()) + + if options.mtn_db == '': + logging.error('No monotone db set (use --mtn-db)') + return 1 + + if not os.access(options.mtn_db, os.R_OK): + logging.error('Monotone db %s not found' % (options.mtn_db)) + return 1 + + if len(args) != 1: + logging.error('Usage: %s version' % (sys.argv[0])) + return 1 + + try: + version = args[0] + + rev_id = run_monotone(options.mtn_db, + ['automate', 'select', 't:' + version]) + + if rev_id == '': + logging.error('No revision for %s found' % (version)) + return 2 + + output_basename = 'Botan-' + version + output_name = os.path.join(options.output_dir, output_basename) + + output_tgz = output_name + '.tgz' + output_tbz = output_name + '.tbz' + + logging.info('Found revision id %s' % (rev_id)) + + if os.access(output_name, os.X_OK): + shutil.rmtree(output_name) + + run_monotone(options.mtn_db, + ['checkout', '-r', rev_id, output_name]) + + shutil.rmtree(os.path.join(output_name, '_MTN')) + remove_file_if_exists(os.path.join(output_name, '.mtn-ignore')) + + version_file = os.path.join(output_name, 'botan_version.py') + if os.access(version_file, os.R_OK): + # rewrite botan_version.py + + contents = open(version_file).readlines() + + def content_rewriter(): + for line in contents: + if line == 'release_vc_rev = None\n': + yield 'release_vc_rev = \'mtn:%s\'\n' % (rev_id) + else: + yield line + + open(version_file, 'w').write(''.join(list(content_rewriter()))) + + os.chdir(options.output_dir) + + remove_file_if_exists(output_tgz) + remove_file_if_exists(output_tgz + '.asc') + archive = tarfile.open(output_tgz, 'w:gz') + archive.add(output_basename) + archive.close() + if options.pgp_key_id != '': + gpg_sign(output_tgz, options.pgp_key_id) + + remove_file_if_exists(output_tbz) + remove_file_if_exists(output_tbz + '.asc') + archive = tarfile.open(output_tbz, 'w:bz2') + archive.add(output_basename) + archive.close() + if options.pgp_key_id != '': + gpg_sign(output_tbz, options.pgp_key_id) + + shutil.rmtree(output_name) + + except Exception, e: + import traceback + traceback.print_exc(file=sys.stderr) + logging.error(str(e)) + return 1 + +if __name__ == '__main__': + sys.exit(main()) |