rscli/cli.py

#!/usr/bin/env python3
import sys
from pprint import pprint
import requests
import argparse

try:
    from colors import color
except ImportError:
    def color(text, *args, **kwargs):
        return text


def req(args, location, data=None):
    kwargs = {}
    if data is not None:
        kwargs['json'] = data
    r = requests.post(args.endpoint + location,
        auth=tuple(args.auth.split(":", 2)),
                         **kwargs
                        )
    return r

def main_channel_list(args):
    r = req(args, "/rsGxsChannels/getChannelsSummaries")
    channels = r.json()['channels']
    for chan in channels:
        print(color(chan['mGroupName'], style='bold', fg='green'))
        print('   ' + color(chan['mGroupId'], style='underline'))

def main_channel_show(args):
    r = req(
        args, "/rsGxsChannels/getChannelsInfo",
        {
            'chanIds': [args.channel_id]
        }
    )
    data = r.json()
    channels = data['channelsInfo']
    for chan in channels:
        print(color(chan['mMeta']['mGroupName'], style='bold', fg='green'))
        print('   ' + color(chan['mMeta']['mGroupId'], style='underline'))
        print('   ' + chan['mDescription'])
        print(color('Last Post:', style='bold') + ' \t%s' % chan['mMeta']['mLastPost'])


def main_channel_post(args):
    chid = args.channel_id
    r = req(
        args, "/rsGxsChannels/createPostV2",
        {
            'channelId': args.channel_id,
            'title': args.post_title,
            'mBody': args.post_body,
        }
    )
    ret = r.json()
    if ret.get('retval', True) is False:
        print(color('ERROR: could not create post', fg='red', style='bold'))
        print(ret['errorMessage'])
        pprint(ret)
        sys.exit(1)
    pprint(ret)


def get_parser():
    p = argparse.ArgumentParser()
    p.add_argument("--endpoint", default="http://127.0.0.1:9092")
    p.add_argument("-u", "--auth", dest="auth")

    p_sub = p.add_subparsers()

    ch = p_sub.add_parser("channel")
    ch.add_argument("--channel-id")
    ch_sub = ch.add_subparsers()
    ch_list = ch_sub.add_parser("list")
    ch_list.set_defaults(mainfunc=main_channel_list)

    ch_show = ch_sub.add_parser("show")
    ch_show.set_defaults(mainfunc=main_channel_show)

    ch_post = ch_sub.add_parser("post")
    ch_post.set_defaults(mainfunc=main_channel_post)
    ch_post.add_argument('--post-title')
    ch_post.add_argument('--post-body')

    return p


def main():
    p = get_parser()
    args = p.parse_args()
    if getattr(args, "mainfunc", None) is None:
        print("Not a complete command")
        sys.exit(2)
    args.mainfunc(args)


if __name__ == "__main__":
    main()