From 5bec6b0e2a2ce5fd03bb04f441536fc130e67997 Mon Sep 17 00:00:00 2001
From: Scott Moser <smoser@brickies.net>
Date: Thu, 24 Oct 2019 20:02:15 +0000
Subject: Fix usages of yaml, and move yaml_dump to safeyaml.dumps.

Here we replace uses of the pyyaml module directly with functions
provided by cloudinit.safeyaml.  Also, change/move
  cloudinit.util.yaml_dumps
to
  cloudinit.safeyaml.dumps

LP: #1849640
---
 cloudinit/safeyaml.py | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

(limited to 'cloudinit/safeyaml.py')

diff --git a/cloudinit/safeyaml.py b/cloudinit/safeyaml.py
index 3bd5e03d..d6f5f95b 100644
--- a/cloudinit/safeyaml.py
+++ b/cloudinit/safeyaml.py
@@ -6,6 +6,8 @@
 
 import yaml
 
+YAMLError = yaml.YAMLError
+
 
 class _CustomSafeLoader(yaml.SafeLoader):
     def construct_python_unicode(self, node):
@@ -27,4 +29,17 @@ class NoAliasSafeDumper(yaml.dumper.SafeDumper):
 def load(blob):
     return(yaml.load(blob, Loader=_CustomSafeLoader))
 
+
+def dumps(obj, explicit_start=True, explicit_end=True, noalias=False):
+    """Return data in nicely formatted yaml."""
+
+    return yaml.dump(obj,
+                     line_break="\n",
+                     indent=4,
+                     explicit_start=explicit_start,
+                     explicit_end=explicit_end,
+                     default_flow_style=False,
+                     Dumper=(NoAliasSafeDumper
+                             if noalias else yaml.dumper.Dumper))
+
 # vi: ts=4 expandtab
-- 
cgit v1.2.3