642391775a1b533797343e1616a043e3959d53d6
3 (c) 2016 by Manuel Peuster <manuel.peuster@upb.de>
8 from tabulate
import tabulate
12 pp
= pprint
.PrettyPrinter(indent
=4)
15 class ZeroRpcClient(object):
18 self
.c
= zerorpc
.Client()
19 self
.c
.connect("tcp://127.0.0.1:4242") # TODO hard coded for now. we'll change this later
22 def execute_command(self
, args
):
23 if getattr(self
, args
["command"]) is not None:
24 # call the local method with the same name as the command arg
25 getattr(self
, args
["command"])(args
)
27 print "Command not implemented."
29 def start(self
, args
):
31 if args
.get("network") is not None:
32 nw_list
= self
._parse
_network
(args
.get("network"))
34 pp
.pprint('nwlist1: {0}'.format(nw_list
))
36 r
= self
.c
.compute_action_start(
37 args
.get("datacenter"),
41 args
.get("docker_command")
46 r
= self
.c
.compute_action_stop(
47 args
.get("datacenter"), args
.get("name"))
51 r
= self
.c
.compute_list(
52 args
.get("datacenter"))
55 # for each container add a line to the output table
61 if len(status
.get("network")) > 0:
62 eth0ip
= status
.get("network")[0][1]
63 eth0status
= "up" if status
.get(
64 "network")[0][3] else "down"
65 table
.append([status
.get("datacenter"),
70 status
.get("state").get("Status")])
71 headers
= ["Datacenter",
77 print tabulate(table
, headers
=headers
, tablefmt
="grid")
79 def status(self
, args
):
80 r
= self
.c
.compute_status(
81 args
.get("datacenter"), args
.get("name"))
84 def profile(self
, args
):
86 if args
.get("network") is not None:
87 nw_list
= self
._parse
_network
(args
.get("network"))
88 logging
.info('nwlist: {0}'.format(nw_list
))
89 r
= self
.c
.compute_profile(
90 args
.get("datacenter"),
94 command
=args
.get("docker_command"),
95 input=args
.get("input"),
96 output
=args
.get("output")
100 def _parse_network(self
, network_str
):
102 parse the options for all network interfaces of the vnf
103 :param network_str: (id=x,ip=x.x.x.x/x), ...
104 :return: list of dicts [{"id":x,"ip":"x.x.x.x/x"}, ...]
107 networks
= network_str
[1:-1].split('),(')
109 nw_dict
= dict(tuple(e
.split('=')) for e
in nw
.split(','))
110 nw_list
.append(nw_dict
)
116 parser
= argparse
.ArgumentParser(description
='son-emu compute')
119 choices
=['start', 'stop', 'list', 'status'],
120 help="Action to be executed.")
122 "--datacenter", "-d", dest
="datacenter",
123 help="Data center to in which the compute instance should be executed")
125 "--name", "-n", dest
="name",
126 help="Name of compute instance e.g. 'vnf1'")
128 "--image","-i", dest
="image",
129 help="Name of container image to be used e.g. 'ubuntu:trusty'")
131 "--dcmd", "-c", dest
="docker_command",
132 help="Startup command of the container e.g. './start.sh'")
134 "--net", dest
="network",
135 help="Network properties of compute instance e.g. \
136 '10.0.0.123/8' or '10.0.0.123/8,11.0.0.123/24' for multiple interfaces.")
138 "--input", "-in", dest
="input",
139 help="input interface of the vnf to profile")
141 "--output", "-out", dest
="output",
142 help="output interface of the vnf to profile")
146 args
= vars(parser
.parse_args(argv
))
148 c
.execute_command(args
)