3 (c) 2016 by Manuel Peuster <manuel.peuster@upb.de>
8 from tabulate
import tabulate
12 pp
= pprint
.PrettyPrinter(indent
=4)
14 class ZeroRpcClient(object):
17 self
.c
= zerorpc
.Client()
18 # TODO connect to DCNetwork API
19 #self.c.connect("tcp://127.0.0.1:4242") # TODO hard coded for now. we'll change this later
20 self
.c
.connect("tcp://127.0.0.1:5151")
23 def execute_command(self
, args
):
24 if getattr(self
, args
["command"]) is not None:
25 # call the local method with the same name as the command arg
26 getattr(self
, args
["command"])(args
)
28 print "Command not implemented."
31 vnf_src_name
= self
._parse
_vnf
_name
(args
.get("source"))
32 vnf_dst_name
= self
._parse
_vnf
_name
(args
.get("destination"))
34 params
= self
._create
_dict
(
35 vnf_src_interface
=self
._parse
_vnf
_interface
(args
.get("source")),
36 vnf_dst_interface
=self
._parse
_vnf
_interface
(args
.get("destination")),
37 weight
=args
.get("weight"),
38 match
=args
.get("match"),
39 bidirectional
=args
.get("bidirectional"))
41 # note zerorpc does not support named arguments
42 r
= self
.c
.network_action_start(
43 #args.get("datacenter"),
49 def remove(self
, args
):
50 vnf_src_name
= self
._parse
_vnf
_name
(args
.get("source"))
51 vnf_dst_name
= self
._parse
_vnf
_name
(args
.get("destination"))
53 params
= self
._create
_dict
(
54 vnf_src_interface
=self
._parse
_vnf
_interface
(args
.get("source")),
55 vnf_dst_interface
=self
._parse
_vnf
_interface
(args
.get("destination")),
56 weight
=args
.get("weight"),
57 match
=args
.get("match"))
59 r
= self
.c
.network_action_stop(
60 #args.get("datacenter"),
66 def _parse_vnf_name(self
, vnf_name_str
):
67 vnf_name
= vnf_name_str
.split(':')[0]
70 def _parse_vnf_interface(self
, vnf_name_str
):
72 vnf_interface
= vnf_name_str
.split(':')[1]
78 def _create_dict(self
, **kwargs
):
81 parser
= argparse
.ArgumentParser(description
='son-emu network')
84 help="Action to be executed: add|remove")
86 "--datacenter", "-d", dest
="datacenter",
87 help="Data center to in which the network action should be initiated")
89 "--source", "-src", dest
="source",
90 help="vnf name of the source of the chain")
92 "--destination", "-dst", dest
="destination",
93 help="vnf name of the destination of the chain")
95 "--weight", "-w", dest
="weight",
96 help="weight metric to calculate the path")
98 "--match", "-m", dest
="match",
99 help="string holding extra matches for the flow entries")
101 "--bidirectional", "-b", dest
="bidirectional",
103 help="add/remove the flow entries in 2 directions")
107 args
= vars(parser
.parse_args(argv
))
109 c
.execute_command(args
)