add pep8 check

[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 735129a..0d0419a 100644 (file)
--- a/multivimbroker/multivimbroker/pub/utils/syscomm.py
+++ b/multivimbroker/multivimbroker/pub/utils/syscomm.py
@@ -10,7 +10,52 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 
 import inspect
+import re
+
+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']
+
+
+def getHeadersKeys(response):
+    hopbyhop = HOP_BY_HOP
+    hopbyhop.extend([x.strip()
+                     for x in response.get('connection', '').split(',')])
+    return [header for header in response.keys() if header not in hopbyhop]
+
+
+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 getMultivimDriver(vimid, full_path=""):
+
+    multivim = "multivim"
+    vim = get_vim_by_id(vimid)
+    if vim["type"] and vim["version"]:
+        pass
+
+    multivimdriver = findMultivimDriver(vim=vim)
+    return re.sub(multivim, multivimdriver, full_path)