src/emuvim/cli/rest/compute.py

   1 """
   2 Copyright (c) 2015 SONATA-NFV and Paderborn University
   3 ALL RIGHTS RESERVED.
   4
   5 Licensed under the Apache License, Version 2.0 (the "License");
   6 you may not use this file except in compliance with the License.
   7 You may obtain a copy of the License at
   8
   9     http://www.apache.org/licenses/LICENSE-2.0
  10
  11 Unless required by applicable law or agreed to in writing, software
  12 distributed under the License is distributed on an "AS IS" BASIS,
  13 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14 See the License for the specific language governing permissions and
  15 limitations under the License.
  16
  17 Neither the name of the SONATA-NFV [, ANY ADDITIONAL AFFILIATION]
  18 nor the names of its contributors may be used to endorse or promote
  19 products derived from this software without specific prior written
  20 permission.
  21
  22 This work has been performed in the framework of the SONATA project,
  23 funded by the European Commission under Grant number 671517 through
  24 the Horizon 2020 and 5G-PPP programmes. The authors would like to
  25 acknowledge the contributions of their colleagues of the SONATA
  26 partner consortium (www.sonata-nfv.eu).
  27 """
  28 from requests import get,put
  29 from tabulate import tabulate
  30 import pprint
  31 import argparse
  32 import json
  33
  34 pp = pprint.PrettyPrinter(indent=4)
  35
  36 class RestApiClient():
  37
  38     def __init__(self):
  39         self.cmds = {}
  40
  41     def execute_command(self, args):
  42         if getattr(self, args["command"]) is not None:
  43             # call the local method with the same name as the command arg
  44             getattr(self, args["command"])(args)
  45         else:
  46             print("Command not implemented.")
  47
  48     def start(self, args):
  49
  50         nw_list = list()
  51         if args.get("network") is not None:
  52             nw_list = self._parse_network(args.get("network"))
  53         req = {'image':args.get("image"),
  54                'command':args.get("docker_command"),
  55                'network':nw_list}
  56
  57         response = put("%s/restapi/compute/%s/%s/start" %
  58                        (args.get("endpoint"),
  59                         args.get("datacenter"),
  60                         args.get("name")),
  61                        json = json.dumps(req))
  62         pp.pprint(response.json())
  63
  64     def stop(self, args):
  65
  66         response = get("%s/restapi/compute/%s/%s/stop" %
  67                        (args.get("endpoint"),
  68                         args.get("datacenter"),
  69                         args.get("name")))
  70         pp.pprint(response.json())
  71
  72     def list(self,args):
  73
  74         list = get('%s/restapi/compute/%s' % (args.get("endpoint"),args.get('datacenter'))).json()
  75
  76         table = []
  77         for c in list:
  78             # for each container add a line to the output table
  79             if len(c) > 1:
  80                 name = c[0]
  81                 status = c[1]
  82                 eth0ip = None
  83                 eth0status = "down"
  84                 if len(status.get("network")) > 0:
  85                     eth0ip = status.get("network")[0].get("ip")
  86                     eth0status = "up" if status.get(
  87                         "network")[0].get("up") else "down"
  88                 table.append([status.get("datacenter"),
  89                               name,
  90                               status.get("image"),
  91                               eth0ip,
  92                               eth0status,
  93                               status.get("state").get("Status")])
  94
  95         headers = ["Datacenter",
  96                    "Container",
  97                    "Image",
  98                    "eth0 IP",
  99                    "eth0 status",
 100                    "Status"]
 101         print(tabulate(table, headers=headers, tablefmt="grid"))
 102
 103     def status(self,args):
 104
 105         list = get("%s/restapi/compute/%s/%s" %
 106                    (args.get("endpoint"),
 107                     args.get("datacenter"),
 108                     args.get("name"))).json()
 109         pp.pprint(list)
 110
 111
 112
 113     def _parse_network(self, network_str):
 114         '''
 115         parse the options for all network interfaces of the vnf
 116         :param network_str: (id=x,ip=x.x.x.x/x), ...
 117         :return: list of dicts [{"id":x,"ip":"x.x.x.x/x"}, ...]
 118         '''
 119         nw_list = list()
 120         networks = network_str[1:-1].split('),(')
 121         for nw in networks:
 122             nw_dict = dict(tuple(e.split('=')) for e in nw.split(','))
 123             nw_list.append(nw_dict)
 124
 125         return nw_list
 126
 127
 128 parser = argparse.ArgumentParser(description='son-emu datacenter')
 129 parser.add_argument(
 130     "command",
 131     choices=['start', 'stop', 'list', 'status'],
 132     help="Action to be executed.")
 133 parser.add_argument(
 134     "--datacenter", "-d", dest="datacenter",
 135     help="Data center to which the command should be applied.")
 136 parser.add_argument(
 137     "--name", "-n", dest="name",
 138     help="Name of compute instance e.g. 'vnf1'.")
 139 parser.add_argument(
 140     "--image","-i", dest="image",
 141     help="Name of container image to be used e.g. 'ubuntu:trusty'")
 142 parser.add_argument(
 143     "--dcmd", "-c", dest="docker_command",
 144     help="Startup command of the container e.g. './start.sh'")
 145 parser.add_argument(
 146     "--net", dest="network",
 147     help="Network properties of a compute instance e.g. \
 148           '(id=input,ip=10.0.10.3/24),(id=output,ip=10.0.10.4/24)' for multiple interfaces.")
 149 parser.add_argument(
 150     "--endpoint", "-e", dest="endpoint",
 151     default="http://127.0.0.1:5001",
 152     help="UUID of the plugin to be manipulated.")
 153
 154 def main(argv):
 155     args = vars(parser.parse_args(argv))
 156     c = RestApiClient()
 157     c.execute_command(args)