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"),
40 cookie
=args
.get("cookie"))
42 # note zerorpc does not support named arguments
43 r
= self
.c
.network_action_start(
44 #args.get("datacenter"),
50 def remove(self
, args
):
51 vnf_src_name
= self
._parse
_vnf
_name
(args
.get("source"))
52 vnf_dst_name
= self
._parse
_vnf
_name
(args
.get("destination"))
54 params
= self
._create
_dict
(
55 vnf_src_interface
=self
._parse
_vnf
_interface
(args
.get("source")),
56 vnf_dst_interface
=self
._parse
_vnf
_interface
(args
.get("destination")),
57 weight
=args
.get("weight"),
58 match
=args
.get("match"),
59 bidirectional
=args
.get("bidirectional"),
60 cookie
=args
.get("cookie"))
62 r
= self
.c
.network_action_stop(
63 #args.get("datacenter"),
69 def _parse_vnf_name(self
, vnf_name_str
):
70 vnf_name
= vnf_name_str
.split(':')[0]
73 def _parse_vnf_interface(self
, vnf_name_str
):
75 vnf_interface
= vnf_name_str
.split(':')[1]
81 def _create_dict(self
, **kwargs
):
84 parser
= argparse
.ArgumentParser(description
='son-emu network')
87 help="Action to be executed: add|remove")
89 "--datacenter", "-d", dest
="datacenter",
90 help="Data center to in which the network action should be initiated")
92 "--source", "-src", dest
="source",
93 help="vnf name of the source of the chain")
95 "--destination", "-dst", dest
="destination",
96 help="vnf name of the destination of the chain")
98 "--weight", "-w", dest
="weight",
99 help="weight metric to calculate the path")
101 "--match", "-m", dest
="match",
102 help="string holding extra matches for the flow entries")
104 "--bidirectional", "-b", dest
="bidirectional",
106 help="add/remove the flow entries in 2 directions")
108 "--cookie", "-c", dest
="cookie",
109 help="cookie for this flow")
112 args
= vars(parser
.parse_args(argv
))
114 c
.execute_command(args
)