gca-client

#!/usr/bin/env python3
from __future__ import print_function

import argparse
import sys

from gca.core import Session
from gca.auth import NetRCAuth


def write_json(json_data):
    data = Session.to_json(json_data)
    sys.stdout.write(data)
    return data

if __name__ == '__main__':
    parser = argparse.ArgumentParser(description='GCA Client')
    parser.add_argument('url', type=str)
    parser.add_argument('-a', '--auth', dest='auth', action='store_true', default=False, help='force authentication')
    subparsers = parser.add_subparsers(help='commands')

    parser_abs = subparsers.add_parser("abstracts", help='list abstracts')
    parser_abs.add_argument('conference', type=str, help='for which conference')
    parser_abs.add_argument('-f', '--full', help='fetch all data', action='store_true', default=False)
    parser_abs.add_argument('-p', '--public', help='fetch only public', action='store_true', default=False)
    parser_abs.set_defaults(cmd='abstracts')

    parser_abs = subparsers.add_parser("image", help='get figure images')
    parser_abs.add_argument('uuid', type=str, nargs='+', help='figure uuid')
    parser_abs.add_argument('--path', type=str, default=None, help='path to store the figures in')
    parser_abs.set_defaults(cmd='image')

    parser_abs = subparsers.add_parser("owners", help='get owner of an object')
    parser_abs.add_argument('-t', '--type', type=str, help='type to fetch [abstract]', default='abstracts', dest='otype')
    parser_abs.add_argument('uuid', type=str, nargs='+', help='object uuid')
    parser_abs.set_defaults(cmd='owners')

    parser_abs = subparsers.add_parser("log", help='get state log of an abstract')
    parser_abs.add_argument('uuid', type=str, help='abstract uuid to fetch log for')
    parser_abs.set_defaults(cmd='state_log')

    parser_abs = subparsers.add_parser("put", help='put abstracts')
    parser_abs.add_argument('conference', type=str, help='for which conference')
    parser_abs.add_argument('abstract', type=str, help='abstract to upload')
    parser_abs.set_defaults(cmd='put')

    parser_abs = subparsers.add_parser("conference", help='fetch conference info')
    parser_abs.add_argument('conference', type=str, help='conf to fetch')
    parser_abs.set_defaults(cmd='conf')

    parser_abs = subparsers.add_parser("patch", help='patch abstracts')
    parser_abs.add_argument('file', type=str, default='-', help="json file to read abstracts from")
    parser_abs.add_argument('fields', type=str, nargs='+', help='fields to patch online in the abstract')
    parser_abs.set_defaults(cmd='patch')

    parser_abs = subparsers.add_parser("set-state", help='set the state if abstracts')
    parser_abs.add_argument('state', type=str, help='new state for the abstract')
    parser_abs.add_argument('uuid', type=str, nargs='+', help='abstracts to set the state')
    parser_abs.add_argument('--note', type=str, dest='note', default='', help='note to attach to the state change log')
    parser_abs.set_defaults(cmd='set-state')

    args = parser.parse_args()

    auth = NetRCAuth()
    session = Session(args.url, auth)

    if args.auth:
        session.authenticate()

    if args.cmd == 'abstracts':
        abstracts = session.get_all_abstracts(args.conference, raw=True, full=args.full, public=args.public)
        write_json(abstracts)
    elif args.cmd == 'image':
        paths = [session.get_figure_image(uuid, path=args.path) for uuid in args.uuid]
        for p in paths:
            print(p)
    elif args.cmd == 'owners':
        import itertools
        owners = itertools.chain.from_iterable(session.get_owners(uuid, args.otype, raw=True) for uuid in args.uuid)
        write_json(list(owners))
    elif args.cmd == 'state_log':
        log = session.get_state_log(args.uuid, raw=True)
        write_json(log)
    elif args.cmd == 'put':
        import json
        import codecs
        conference = args.conference
        fd = codecs.open(args.abstract, 'r', encoding='utf-8') if args.abstract != '-' else sys.stdin
        data = json.load(fd)
        if isinstance(data, list):
            res = [session.upload_abstract(a, conference, raw=True) for a in data]
        else:
            res = session.upload_abstract(data, conference, raw=True)
        write_json(res)
    elif args.cmd == 'conf':
        conf = session.get_conference(args.conference, raw=True)
        write_json(conf)
    elif args.cmd == 'patch':
        import codecs
        import json
        import urllib2
        fd = codecs.open(args.file, 'r', encoding='utf-8') if args.file != '-' else sys.stdin
        data = json.load(fd)
        abstracts = []
        failed = []
        for abstract in data:
            uuid = abstract['uuid']
            try:
                sys.stderr.write('[P] %s\n' % uuid)
                patched = session.patch_abstract(abstract, args.fields, raw=True)
                abstracts.append(patched)
            except (urllib2.HTTPError) as error:
                sys.stderr.write('[E] Failed: %s \n' % str(error.code))
                failed.append((uuid, str(error.code)))

        if len(failed):
            sys.stderr.write('Following abstracts could not be patched [N = %d of %d]:\n' %
                             (len(failed), len(data)))
            for f in failed:
                sys.stderr.write('\t%s [%s]\n' % (f[0], f[1]))

        write_json(abstracts)
    elif args.cmd == 'set-state':
        import urllib2
        failed = []
        logs = []
        for uuid in args.uuid:
            try:
                log = session.set_state(uuid, args.state, args.note, raw=True)
                logs.append(log)
            except (urllib2.HTTPError) as error:
                sys.stderr.write('[E] Failed: %s \n' % str(error.code))
                failed.append((uuid, str(error.code)))

        if len(failed):
            sys.stderr.write('Following abstracts had an error while setting the state [N = %d of %d]:\n' %
                             (len(failed), len(args.uuid)))
        for f in failed:
            sys.stderr.write('\t%s [%s]\n' % (f[0], f[1]))
        write_json(logs)