Merge "Add test_get_res_from_aai"

[multicloud/framework.git] / multivimbroker / multivimbroker / pub / utils / syscomm.py

diff --git a/multivimbroker/multivimbroker/pub/utils/syscomm.py b/multivimbroker/multivimbroker/pub/utils/syscomm.py
index 07606bf..337a1bd 100644 (file)
--- a/multivimbroker/multivimbroker/pub/utils/syscomm.py
+++ b/multivimbroker/multivimbroker/pub/utils/syscomm.py
@@ -1,4 +1,5 @@
 # Copyright (c) 2017 Wind River Systems, Inc.
+# Copyright (c) 2017-2018 VMware, Inc.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -10,49 +11,70 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 
 import inspect
+import json
+import os
 import re
 
-import multivimbroker.pub.exceptions as  exceptions
+import multivimbroker.pub.exceptions as exceptions
 from multivimbroker.pub.msapi.extsys import get_vim_by_id
+
+
 def fun_name():
     return inspect.stack()[1][3]
 
 
+# Which headers are hop-by-hop headers by default
+HOP_BY_HOP = ['connection', 'keep-alive', 'proxy-authenticate',
+              'proxy-authorization', 'te', 'trailers',
+              'transfer-encoding', 'upgrade']
 
 
-# Which headers are hop-by-hop headers by default
-HOP_BY_HOP = ['connection', 'keep-alive', 'proxy-authenticate', 'proxy-authorization', 'te', 'trailers', 'transfer-encoding', 'upgrade']
 def getHeadersKeys(response):
     hopbyhop = HOP_BY_HOP
-    hopbyhop.extend([x.strip() for x in response.get('connection', '').split(',')])
+    hopbyhop.extend([x.strip()
+                     for x in response.get('connection', '').split(',')])
     return [header for header in response.keys() if header not in hopbyhop]
 
 
+# trim out 'HTTP_' prefix part and replace "_" wiht "-".
+def originHeaders(request):
+    regex = re.compile('^HTTP_')
+    return dict((regex.sub('', header).replace("_", "-"), value)
+                for (header, value) in request.META.items()
+                if header.startswith('HTTP_'))
 
-def findMultivimDriver(vim=None):
 
-    if vim and vim["type"] == "openstack":
-        if vim["version"] == "kilo":
-            multivimdriver = "multivim-kilo"
-        elif vim["version"] == "newton":
-            multivimdriver = "multivim-newton"
-        else:
-            # if vim type is openstack, use latest "newton" version as default
-            multivimdriver = "multivim-newton"
-    elif vim and vim["type"] == "vmware":
-        multivimdriver = "multivim-vio"
-    else:
-        raise  exceptions.NotFound("Not support VIM type")
-    return  multivimdriver
+def findMultivimDriver(vim=None):
+    json_file = os.path.join(os.path.dirname(__file__),
+                             '../config/provider-plugin.json')
+    with open(json_file, "r") as f:
+        plugins = json.load(f)
+    if not vim or vim.get("type") not in plugins.keys():
+        raise exceptions.NotFound("Not support VIM type")
+    plugin = plugins[vim["type"]]
+    if vim.get("version") in plugin["versions"].keys():
+        return plugin["versions"][vim["version"]]["provider_plugin"]
+    return plugin["provider_plugin"]
 
 
+def getMultivimDriver(vimid, full_path=""):
+    multcloud = "multicloud"
+    vim = get_vim_by_id(vimid)
+    multclouddriver = findMultivimDriver(vim=vim)
+    return re.sub(multcloud, multclouddriver, full_path)
 
-def  getMultivimDriver(vimid,full_path=""):
 
-    multivim = "multivim"
-    vim = get_vim_by_id(vimid)
-    if vim["type"] and vim["version"]:
-        pass
+def getVIMTypes():
+        # Fix here unless we have plugin registry
+        json_file = os.path.join(os.path.dirname(__file__),
+                                 '../config/provider-plugin.json')
+        with open(json_file, "r") as f:
+            plugins = json.load(f)
+        ret = []
+        for k, v in plugins.items():
+            item = {}
+            item["vim_type"] = v.get("vim_type")
+            item["versions"] = [k for k in v.get('versions', {})]
+            ret.append(item)
 
-    multivimdriver = findMultivimDriver(vim=vim)
-    return re.sub(multivim, multivimdriver, full_path)
+        return ret