summaryrefslogtreecommitdiffstats
path: root/share/starlingx_base/resource/infra_workload.py
diff options
context:
space:
mode:
authorXiaohua Zhang <xiaohua.zhang@windriver.com>2019-04-22 08:47:40 +0000
committerXiaohua Zhang <xiaohua.zhang@windriver.com>2019-04-22 08:47:40 +0000
commita0db0636571ff8bc39d16ac76fa9fde98f90173c (patch)
treea0e598f46d12faa18ca9b407e4448bd64a11313a /share/starlingx_base/resource/infra_workload.py
parentdf8a90f400d85bdf910d25b988886c110d051de2 (diff)
Fix bug of interaction with artifact broker
Change-Id: I3dffcc4463fd5ec7004bbb7cc20c0dcbf1622f6d Issue-ID: MULTICLOUD-567 Signed-off-by: Xiaohua Zhang <xiaohua.zhang@windriver.com>
Diffstat (limited to 'share/starlingx_base/resource/infra_workload.py')
-rw-r--r--share/starlingx_base/resource/infra_workload.py45
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