src/emuvim/cli/datacenter.py

"""
son-emu datacenter CLI
(c) 2016 by Manuel Peuster <manuel.peuster@upb.de>
"""

import argparse
import pprint
from tabulate import tabulate
import zerorpc


pp = pprint.PrettyPrinter(indent=4)


class ZeroRpcClient(object):

    def __init__(self):
        self.c = zerorpc.Client()
        self.c.connect("tcp://127.0.0.1:4242")  # TODO hard coded for now. we'll change this later
        self.cmds = {}

    def execute_command(self, args):
        if getattr(self, args["command"]) is not None:
            # call the local method with the same name as the command arg
            getattr(self, args["command"])(args)
        else:
            print("Command not implemented.")

    def list(self, args):
        r = self.c.datacenter_list()
        table = []
        for d in r:
            # for each dc add a line to the output table
            if len(d) > 0:
                table.append([d.get("label"),
                              d.get("internalname"),
                              d.get("switch"),
                              d.get("n_running_containers"),
                              len(d.get("metadata"))])
        headers = ["Label",
                   "Internal Name",
                   "Switch",
                   "# Containers",
                   "# Metadata Items"]
        print(tabulate(table, headers=headers, tablefmt="grid"))

    def status(self, args):
        r = self.c.datacenter_status(
            args.get("datacenter"))
        pp.pprint(r)


parser = argparse.ArgumentParser(description='son-emu datacenter')
parser.add_argument(
    "command",
    choices=['list', 'status'],
    help="Action to be executed.")
parser.add_argument(
    "--datacenter", "-d", dest="datacenter",
    help="Data center to which the command should be applied.")


def main(argv):
    args = vars(parser.parse_args(argv))
    c = ZeroRpcClient()
    c.execute_command(args)