#!/usr/bin/env python
import json
import logging
import sys
from argparse import ArgumentParser, FileType, SUPPRESS
from os import fdopen, remove
from subprocess import check_call
from tempfile import mkstemp
from time import sleep
import redis
import yaml
from candolint.queue import loop
from candolint.utils import lookup_option, rel
def handle(rconn, incoming):
    payload = json.loads(incoming)
    outfd, outfn = mkstemp('.txt', '{repo}.{timestamp}.'.format(**payload))
    logging.debug('Writing output to %s', outfn)
    with fdopen(outfd, 'wb') as outf:
        outf.write(payload['output'].encode('utf-8'))
    logging.debug('Running %s incoming.py %r', sys.executable, outfn)
    check_call([sys.executable, rel('incoming.py'), outfn])
    logging.info('Processed file %s', outfn)
    remove(outfn)
def main():
    parser = ArgumentParser(argument_default=SUPPRESS)
    parser.add_argument(
        '-c', '--config', type=FileType('r'),
        help='configuration file (YAML)')
    parser.add_argument(
        '-d', '--debug', action='store_true', default=False,
        help='enable debugging output')
    group = parser.add_argument_group(
        'queue configuration',
        'these options that can also be specified in the configuration file')
    group.add_argument('--redis-host', help='(default: 127.0.0.1)')
    group.add_argument('--redis-port', type=int, help='(default: 6379)')
    group.add_argument('--redis-password')
    args = parser.parse_args()
    config = yaml.safe_load(args.config) if hasattr(args, 'config') else {}
    rhost = lookup_option('redis-host', args, config, default='127.0.0.1')
    rport = lookup_option('redis-port', args, config, default=6379)
    rpass = lookup_option('redis-password', args, config)
    logging.basicConfig(
        level=logging.DEBUG if args.debug else logging.INFO,
        format='%(asctime)s %(levelname)-8s %(message)s')
    while True:
        logging.debug('Connecting to Redis server %s:%d', rhost, rport)
        rconn = redis.StrictRedis(host=rhost, port=rport, password=rpass, db=0)
        logging.info('Connected to Redis server')
        try:
            loop(rconn, 'candolint:queue:incoming', handle)
        except redis.exceptions.ConnectionError:
            logging.warn('Connection to Redis lost, reconnecting in 1 minute')
            sleep(60)
if __name__ == '__main__':
    main()