Fix bug of interaction with artifact broker

Change-Id: I3dffcc4463fd5ec7004bbb7cc20c0dcbf1622f6d
Issue-ID: MULTICLOUD-567
Signed-off-by: Xiaohua Zhang <xiaohua.zhang@windriver.com>

1 files changed, 35 insertions, 10 deletions

diff --git a/share/starlingx_base/resource/infra_workload.py b/share/starlingx_base/resource/infra_workload.py
index fc6d7ef2..af5f5531 100644
--- a/share/starlingx_base/resource/infra_workload.py
+++ b/share/starlingx_base/resource/infra_workload.py
@@ -13,6 +13,8 @@
 # limitations under the License.
 
 import os
+import json
+
 import logging
 from django.conf import settings
 from django.http import QueryDict
@@ -27,8 +29,17 @@ from newton_base.resource import infra_workload_helper as infra_workload_helper
 
 from newton_base.util import VimDriverUtils
 
+import yaml
+NoDatesSafeLoader = yaml.SafeLoader
+NoDatesSafeLoader.yaml_implicit_resolvers = {
+    k: [r for r in v if r[0] != 'tag:yaml.org,2002:timestamp'] for
+        k, v in NoDatesSafeLoader.yaml_implicit_resolvers.items()
+}
+
 logger = logging.getLogger(__name__)
 
+
+
 # global var: Audition thread
 # the id is the workloadid, which implies post to workloadid1 followed by delete workloadid1
 # will replace the previous backlog item
@@ -416,20 +427,29 @@ class InfraWorkloadHelper(infra_workload_helper.InfraWorkloadHelper):
         # assumption: mount point: /opt/artifacts/<vfmodule_uuid>
         try:
             vfmodule_path_base = r"/opt/artifacts/%s" % vf_module_model_customization_id
+            self._logger.debug("vfmodule_path_base: %s" % vfmodule_path_base)
             vfmodule_metadata_path = r"%s/vfmodule-meta.json" % vfmodule_path_base
             service_metadata_path = r"%s/service-meta.json" % vfmodule_path_base
             with open(vfmodule_metadata_path,
-                      'r', encoding='UTF-8') as vf:
-                vfmodule_metadata = vf.read()  # assume the metadata file size is small
+                      'r') as vf:
+                vfmodule_metadata_str = vf.read()  # assume the metadata file size is small
+                vfmodule_metadata = json.loads(vfmodule_metadata_str)
+                vfmodule_metadata = [e for e in vfmodule_metadata
+                                     if e.get("vfModuleModelCustomizationUUID", None)
+                                     == vf_module_model_customization_id]
+                self._logger.debug("vfmodule_metadata: %s" % vfmodule_metadata)
                 if vfmodule_metadata and len(vfmodule_metadata) > 0:
                     # load service-metadata
                     with open(service_metadata_path,
-                              'r', encoding='UTF-8') as sf:
-                        service_metadata = sf.read()  # assume the metadata file size is small
+                              'r') as sf:
+                        service_metadata_str = sf.read()  # assume the metadata file size is small
+                        service_metadata = json.loads(service_metadata_str)
+                        self._logger.debug("service_metadata: %s" % service_metadata)
                         if service_metadata and len(service_metadata) > 0:
                             # get the artifacts uuid
-                            artifacts_uuids = vfmodule_metadata.get("artifacts", None)
-                            templatedata1 = {}.update(template_data)
+                            artifacts_uuids = vfmodule_metadata[0].get("artifacts", None)
+                            self._logger.debug("artifacts_uuids: %s" % artifacts_uuids)
+                            templatedata1 = template_data.copy()
                             for a in service_metadata["artifacts"]:
                                 artifactUUID = a.get("artifactUUID", "")
                                 if artifactUUID not in artifacts_uuids:
@@ -437,20 +457,25 @@ class InfraWorkloadHelper(infra_workload_helper.InfraWorkloadHelper):
                                 artifact_type = a.get("artifactType", "")
                                 artifact_name = a.get("artifactName", "")
                                 artifact_path = r"%s/%s" % (vfmodule_path_base, artifact_name)
+                                self._logger.debug("artifact_path: %s" % artifact_path)
 
                                 # now check the type
                                 if artifact_type.lower() == "heat":
                                     # heat template file
                                     with open(artifact_path,
-                                              'r', encoding='UTF-8') as af:
-                                        templatedata1["template"] = af.read()  # assume the template file size is small
+                                              'r') as af:
+                                        # assume the template file size is small
+                                        templatedata1["template"] = \
+                                            yaml.safe_load(af.read(), Loader=NoDatesSafeLoader)
                                     # pass
 
                                 elif artifact_type.lower() == "heat_env":
                                     # heat env file
                                     with open(artifact_path,
-                                              'r', encoding='UTF-8') as af:
-                                        templatedata1["parameters"] = af.read()  # assume the env file size is small
+                                              'r') as af:
+                                        # assume the env file size is small
+                                        templatedata1.update(yaml.safe_load(
+                                            af.read(), Loader=NoDatesSafeLoader))
                                     # pass
                                 # pass
                             return templatedata1