src/emuvim/cli/network.py

"""
son-emu network 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()
        # 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 add(self, args):
        vnf_src_name = self._parse_vnf_name(args.get("source"))
        vnf_dst_name = self._parse_vnf_name(args.get("destination"))

        params = self._create_dict(
            vnf_src_interface=self._parse_vnf_interface(args.get("source")),
            vnf_dst_interface=self._parse_vnf_interface(args.get("destination")),
            weight=args.get("weight"),
            match=args.get("match"),
            bidirectional=args.get("bidirectional"),
            cookie=args.get("cookie"))

        # note zerorpc does not support named arguments
        r = self.c.network_action_start(
            #args.get("datacenter"),
            vnf_src_name,
            vnf_dst_name,
            params)
        pp.pprint(r)

    def remove(self, args):
        vnf_src_name = self._parse_vnf_name(args.get("source"))
        vnf_dst_name = self._parse_vnf_name(args.get("destination"))

        params = self._create_dict(
            vnf_src_interface=self._parse_vnf_interface(args.get("source")),
            vnf_dst_interface=self._parse_vnf_interface(args.get("destination")),
            weight=args.get("weight"),
            match=args.get("match"),
            cookie=args.get("cookie"))

        r = self.c.network_action_stop(
            #args.get("datacenter"),
            vnf_src_name,
            vnf_dst_name,
            params)
        pp.pprint(r)

    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

    def _create_dict(self, **kwargs):
        return kwargs

parser = argparse.ArgumentParser(description='son-emu network')
parser.add_argument(
    "command",
    help="Action to be executed: add|remove")
parser.add_argument(
    "--datacenter", "-d", dest="datacenter",
    help="Data center to in which the network action should be initiated")
parser.add_argument(
    "--source", "-src", dest="source",
    help="vnf name of the source of the chain")
parser.add_argument(
    "--destination", "-dst", dest="destination",
    help="vnf name of the destination of the chain")
parser.add_argument(
    "--weight", "-w", dest="weight",
    help="weight metric to calculate the path")
parser.add_argument(
    "--match", "-m", dest="match",
    help="string holding extra matches for the flow entries")
parser.add_argument(
    "--bidirectional", "-b", dest="bidirectional",
    action='store_true',
    help="add/remove the flow entries in 2 directions")
parser.add_argument(
    "--cookie", "-c", dest="cookie",
    help="cookie for this flow")

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