pungi/bin/pungi-wait-for-signed-ostree-handler

#!/usr/bin/env python
# -*- coding: utf-8 -*-

from __future__ import print_function

import argparse
import datetime
import fedmsg.config
import json
import os
import sys
import time

if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument('cmd')
    opts = parser.parse_args()

    if opts.cmd != 'ostree':
        # Not an announcement of new ostree commit, nothing to do.
        sys.exit()

    try:
        data = json.load(sys.stdin)
    except ValueError:
        print('Failed to decode data', file=sys.stderr)
        sys.exit(1)

    repo = data['local_repo_path']
    commit = data['commitid']
    path = '%s/objects/%s/%s.commitmeta' % (repo, commit[:2], commit[2:])

    config = fedmsg.config.load_config()
    config['active'] = True           # Connect out to a fedmsg-relay instance
    config['cert_prefix'] = 'releng'  # Use this cert.
    fedmsg.init(**config)
    topic = 'compose.%s' % opts.cmd.replace('-', '.').lower()

    count = 0
    while not os.path.exists(path):
        print('%s: Commit not signed yet, waiting...'
              % datetime.datetime.utcnow())
        count += 1
        if count >= 60:  # Repeat every 5 minutes
            print('Repeating notification')
            fedmsg.publish(topic=topic, modname='pungi', msg=data)
            count = 0
        time.sleep(5)

    print('Found signature.')