src/emuvim/cli/rest/network.py

   1 """
   2 Copyright (c) 2015 SONATA-NFV
   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, delete
  29 import argparse
  30
  31
  32 class RestApiClient():
  33
  34     def __init__(self):
  35         self.cmds = {}
  36
  37     def execute_command(self, args):
  38         if getattr(self, args["command"]) is not None:
  39             # call the local method with the same name as the command arg
  40             getattr(self, args["command"])(args)
  41         else:
  42             print("Command not implemented.")
  43
  44     def add(self, args):
  45         params = self._create_dict(
  46             vnf_src_name=self._parse_vnf_name(args.get("source")),
  47             vnf_dst_name = self._parse_vnf_name(args.get("destination")),
  48             vnf_src_interface=self._parse_vnf_interface(args.get("source")),
  49             vnf_dst_interface=self._parse_vnf_interface(args.get("destination")),
  50             weight=args.get("weight"),
  51             match=args.get("match"),
  52             bidirectional=args.get("bidirectional"),
  53             cookie=args.get("cookie"),
  54             priority=args.get("priority"))
  55
  56         response = put("{0}/restapi/network".format(args.get("endpoint")),
  57                        params=params)
  58         print(self._nice_print(response.text))
  59
  60     def remove(self, args):
  61         params = self._create_dict(
  62             vnf_src_name = self._parse_vnf_name(args.get("source")),
  63             vnf_dst_name = self._parse_vnf_name(args.get("destination")),
  64             vnf_src_interface=self._parse_vnf_interface(args.get("source")),
  65             vnf_dst_interface=self._parse_vnf_interface(args.get("destination")),
  66             weight=args.get("weight"),
  67             match=args.get("match"),
  68             bidirectional=args.get("bidirectional"),
  69             cookie=args.get("cookie"),
  70             priority=args.get("priority"))
  71
  72         response = delete("{0}/restapi/network".format(args.get("endpoint")),
  73                           params=params)
  74         print(self._nice_print(response.text))
  75
  76     def _parse_vnf_name(self, vnf_name_str):
  77         vnf_name = vnf_name_str.split(':')[0]
  78         return vnf_name
  79
  80     def _parse_vnf_interface(self, vnf_name_str):
  81         try:
  82             vnf_interface = vnf_name_str.split(':')[1]
  83         except:
  84             vnf_interface = None
  85
  86         return vnf_interface
  87
  88     def _create_dict(self, **kwargs):
  89         return kwargs
  90
  91     def _nice_print(self, text):
  92         # some modules seem to return unicode strings where newlines, other special characters are escaped
  93         text = str(text).replace('\\n', '\n')
  94         text = str(text).replace('\\"', '"')
  95         return text
  96
  97 parser = argparse.ArgumentParser(description='son-emu-cli network')
  98 parser.add_argument(
  99     "command",
 100     choices=['add', 'remove'],
 101     help="Action to be executed.")
 102 parser.add_argument(
 103     "--datacenter", "-d", dest="datacenter",
 104     help="Data center to in which the network action should be initiated")
 105 parser.add_argument(
 106     "--source", "-src", dest="source",
 107     help="vnf name of the source of the chain")
 108 parser.add_argument(
 109     "--destination", "-dst", dest="destination",
 110     help="vnf name of the destination of the chain")
 111 parser.add_argument(
 112     "--weight", "-w", dest="weight",
 113     help="weight edge attribute to calculate the path")
 114 parser.add_argument(
 115     "--priority", "-p", dest="priority", default="1000",
 116     help="priority of flow rule")
 117 parser.add_argument(
 118     "--match", "-m", dest="match",
 119     help="string holding extra matches for the flow entries")
 120 parser.add_argument(
 121     "--bidirectional", "-b", dest="bidirectional", action='store_true',
 122     help="add/remove the flow entries from src to dst and back")
 123 parser.add_argument(
 124     "--cookie", "-c", dest="cookie", default="10",
 125     help="cookie for this flow, as easy to use identifier (eg. per tenant/service)")
 126 parser.add_argument(
 127     "--endpoint", "-e", dest="endpoint",
 128     default="http://127.0.0.1:5001",
 129     help="REST API endpoint of son-emu (default:http://127.0.0.1:5001)")
 130
 131 def main(argv):
 132     args = vars(parser.parse_args(argv))
 133     c = RestApiClient()
 134     c.execute_command(args)