Standardize Formatting

[osm/osmclient.git] / osmclient / common / utils.py

diff --git a/osmclient/common/utils.py b/osmclient/common/utils.py
index ec0e0b0..aed7d29 100644 (file)
--- a/osmclient/common/utils.py
+++ b/osmclient/common/utils.py
@@ -59,8 +59,7 @@ def get_key_val_from_pkg(descriptor_file):
     tar = tarfile.open(descriptor_file)
     yamlfile = None
     for member in tar.getmembers():
-        if (re.match('.*.yaml', member.name) and
-                len(member.name.split('/')) == 2):
+        if re.match(".*.yaml", member.name) and len(member.name.split("/")) == 2:
             yamlfile = member.name
             break
     if yamlfile is None:
@@ -68,23 +67,27 @@ def get_key_val_from_pkg(descriptor_file):
 
     dict = yaml.safe_load(tar.extractfile(yamlfile))
     result = {}
-    for k1, v1 in list(dict.items()):
-        if not k1.endswith('-catalog'):
-            continue
-        for k2, v2 in v1.items():
-            if not k2.endswith('nsd') and not k2.endswith('vnfd'):
+    for k in dict:
+        if "nsd" in k:
+            result["type"] = "nsd"
+        else:
+            result["type"] = "vnfd"
+    if "type" not in result:
+        for k1, v1 in list(dict.items()):
+            if not k1.endswith("-catalog"):
                 continue
+            for k2, v2 in v1.items():
+                if not k2.endswith("nsd") and not k2.endswith("vnfd"):
+                    continue
+                if "nsd" in k2:
+                    result["type"] = "nsd"
+                else:
+                    result["type"] = "vnfd"
+                for entry in v2:
+                    for k3, v3 in list(entry.items()):
+                        # strip off preceeding chars before :
+                        key_name = k3.split(":").pop()
+                        result[key_name] = v3
 
-            if 'nsd' in k2:
-                result['type'] = 'nsd'
-            else:
-                result['type'] = 'vnfd'
-
-            for entry in v2:
-                for k3, v3 in list(entry.items()):
-                    # strip off preceeding chars before :
-                    key_name = k3.split(':').pop()
-
-                    result[key_name] = v3
     tar.close()
     return result