src/emuvim/cli/monitor.py

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

import argparse
import pprint
from tabulate import tabulate
import zerorpc
import time


pp = pprint.PrettyPrinter(indent=4)

class ZeroRpcClient(object):

    def __init__(self):
        self.c = zerorpc.Client()
        # TODO connect to DCNetwork API
        #self.c.connect("tcp://127.0.0.1:4242")  # TODO hard coded for now. we'll change this later
        self.c.connect("tcp://127.0.0.1:5151")
        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 get_rate(self, args):
        vnf_name = self._parse_vnf_name(args.get("vnf_name"))
        vnf_interface = self._parse_vnf_interface(args.get("vnf_name"))
        self.c.monitor_setup_rate_measurement(
            vnf_name,
            vnf_interface,
            args.get("direction"),
            args.get("metric"))
        while True:
            r = self.c.monitor_get_rate(
                vnf_name,
                vnf_interface,
                args.get("direction"),
                args.get("metric"))
            pp.pprint(r)
            time.sleep(1)

    def _parse_vnf_name(self, vnf_name_str):
        vnf_name = vnf_name_str.split(':')[0]
        return vnf_name

    def _parse_vnf_interface(self, vnf_name_str):
        try:
            vnf_interface = vnf_name_str.split(':')[1]
        except:
            vnf_interface = None

        return vnf_interface

parser = argparse.ArgumentParser(description='son-emu network')
parser.add_argument(
    "command",
    help="Action to be executed: get_rate")
parser.add_argument(
    "--vnf_name", "-vnf", dest="vnf_name",
    help="vnf name to be monitored")
parser.add_argument(
    "--direction", "-d", dest="direction",
    help="rx (ingress rate) or tx (egress rate)")
parser.add_argument(
    "--metric", "-m", dest="metric",
    help="bytes (byte rate), packets (packet rate)")

def main(argv):
    print "This is the son-emu monitor CLI."
    print "Arguments: %s" % str(argv)
    args = vars(parser.parse_args(argv))
    c = ZeroRpcClient()
    c.execute_command(args)