emuvim/cli/compute.py

   1 """
   2 son-emu compute CLI
   3 (c) 2016 by Manuel Peuster <manuel.peuster@upb.de>
   4 """
   5
   6 import argparse
   7 import pprint
   8 from tabulate import tabulate
   9 import zerorpc
  10
  11
  12 pp = pprint.PrettyPrinter(indent=4)
  13
  14
  15 class ZeroRpcClient(object):
  16
  17     def __init__(self):
  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
  20         self.cmds = {}
  21
  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)
  26         else:
  27             print "Command not implemented."
  28
  29     def start(self, args):
  30         network = {}
  31         if args.get("network") is not None:
  32             network = {"ip": args.get("network")}
  33         r = self.c.compute_action_start(
  34             args.get("datacenter"),
  35             args.get("name"),
  36             args.get("image"),
  37             args.get("docker_command"),
  38             network)
  39         pp.pprint(r)
  40
  41     def stop(self, args):
  42         r = self.c.compute_action_stop(
  43             args.get("datacenter"), args.get("name"))
  44         pp.pprint(r)
  45
  46     def list(self, args):
  47         r = self.c.compute_list(
  48             args.get("datacenter"))
  49         table = []
  50         for c in r:
  51             # for each container add a line to the output table
  52             if len(c) > 1:
  53                 name = c[0]
  54                 status = c[1]
  55                 eth0ip = None
  56                 eth0status = "down"
  57                 if len(status.get("network")) > 0:
  58                     eth0ip = status.get("network")[0][1]
  59                     eth0status = "up" if status.get(
  60                         "network")[0][3] else "down"
  61                 table.append([status.get("datacenter"),
  62                               name,
  63                               status.get("image"),
  64                               eth0ip,
  65                               eth0status,
  66                               status.get("state").get("Status")])
  67         headers = ["Datacenter",
  68                    "Container",
  69                    "Image",
  70                    "eth0 IP",
  71                    "eth0 status",
  72                    "Status"]
  73         print tabulate(table, headers=headers, tablefmt="grid")
  74
  75     def status(self, args):
  76         r = self.c.compute_status(
  77             args.get("datacenter"), args.get("name"))
  78         pp.pprint(r)
  79
  80
  81 parser = argparse.ArgumentParser(description='son-emu compute')
  82 parser.add_argument(
  83     "command",
  84     choices=['start', 'stop', 'list', 'status'],
  85     help="Action to be executed.")
  86 parser.add_argument(
  87     "--datacenter", "-d", dest="datacenter",
  88     help="Data center to in which the compute instance should be executed")
  89 parser.add_argument(
  90     "--name", "-n", dest="name",
  91     help="Name of compute instance e.g. 'vnf1'")
  92 parser.add_argument(
  93     "--image","-i", dest="image",
  94     help="Name of container image to be used e.g. 'ubuntu'")
  95 parser.add_argument(
  96     "--dcmd", "-c", dest="docker_command",
  97     help="Startup command of the container e.g. './start.sh'")
  98 parser.add_argument(
  99     "--net", dest="network",
 100     help="Network properties of compute instance e.g. '10.0.0.123/8'")
 101
 102
 103 def main(argv):
 104     args = vars(parser.parse_args(argv))
 105     c = ZeroRpcClient()
 106     c.execute_command(args)