summaryrefslogtreecommitdiff
path: root/cloudinit
diff options
context:
space:
mode:
authorJoshua Harlow <harlowja@gmail.com>2012-11-22 08:49:48 -0800
committerJoshua Harlow <harlowja@gmail.com>2012-11-22 08:49:48 -0800
commitf6f6d4b961adca0a5f19ea4bbdd35bcc45955b24 (patch)
tree5c9f88d03b0c3254ee5faef0184a0f54bc3f9a7a /cloudinit
parentcc765725c6493082e8e2f72f78de9786b6e2cc2a (diff)
downloadvyos-cloud-init-f6f6d4b961adca0a5f19ea4bbdd35bcc45955b24.tar.gz
vyos-cloud-init-f6f6d4b961adca0a5f19ea4bbdd35bcc45955b24.zip
Change the yaml merge header extraction to be in a sep. function that can look in more places.
Diffstat (limited to 'cloudinit')
-rw-r--r--cloudinit/handlers/cloud_config.py21
1 files changed, 18 insertions, 3 deletions
diff --git a/cloudinit/handlers/cloud_config.py b/cloudinit/handlers/cloud_config.py
index 50fbb445..de9c5252 100644
--- a/cloudinit/handlers/cloud_config.py
+++ b/cloudinit/handlers/cloud_config.py
@@ -30,6 +30,7 @@ from cloudinit.settings import (PER_ALWAYS)
LOG = logging.getLogger(__name__)
DEF_MERGE_TYPE = "list+dict+str"
+MERGE_HEADER = 'Merge-Type'
class CloudConfigPartHandler(handlers.Handler):
@@ -46,19 +47,33 @@ class CloudConfigPartHandler(handlers.Handler):
def _write_cloud_config(self, buf):
if not self.cloud_fn:
return
+ # Write the combined & merged dictionary/yaml out
lines = ["#cloud-config", util.yaml_dumps(self.cloud_buf)]
util.write_file(self.cloud_fn, "\n".join(lines), 0600)
+ def _merge_header_extract(self, payload_yaml):
+ merge_header_yaml = ''
+ for k in [MERGE_HEADER, MERGE_HEADER.lower(),
+ MERGE_HEADER.lower().replace("-", "_")]:
+ if k in payload_yaml:
+ merge_header_yaml = str(payload_yaml[k])
+ break
+ return merge_header_yaml
+
def _merge_part(self, payload, headers, filename):
- merge_headers = headers.get("Merge-Type")
+ merge_header_headers = headers.get(MERGE_HEADER, '')
try:
payload_y = util.load_yaml(payload)
merge_how = ''
- for merge_i in [payload_y.pop("Merge-Type", ''), merge_headers]:
+ # Select either the merge-type from the content
+ # or the merge type from the headers or default to our own set
+ # if neither exists (or is empty) from the later
+ merge_header_yaml = self._merge_header_extract(payload_y)
+ for merge_i in [merge_header_yaml, merge_header_headers]:
+ merge_i = merge_i.strip().lower()
if merge_i:
merge_how = merge_i
break
- merge_how = merge_how.strip().lower()
if not merge_how:
merge_how = DEF_MERGE_TYPE
merger = mergers.construct(merge_how)