_yaml.py: Make _yaml.dump() optionally dump to stdout

1 files changed, 8 insertions, 4 deletions

diff --git a/buildstream/_yaml.py b/buildstream/_yaml.py
index f9d49343d..2a1073fa9 100644
--- a/buildstream/_yaml.py
+++ b/buildstream/_yaml.py
@@ -18,9 +18,11 @@
 #  Authors:
 #        Tristan Van Berkom <tristan.vanberkom@codethink.co.uk>
 
+import sys
 import collections
 import copy
 from enum import Enum
+from contextlib import ExitStack
 
 from ruamel import yaml
 from ruamel.yaml.representer import SafeRepresenter, RoundTripRepresenter
@@ -227,10 +229,12 @@ def load(filename, shortname=None, copy_tree=False):
 #    node (dict): A node previously loaded with _yaml.load() above
 #    filename (str): The YAML file to load
 #
-#
-def dump(node, filename):
-
-    with open(filename, 'w') as f:
+def dump(node, filename=None):
+    with ExitStack() as stack:
+        if filename:
+            f = stack.enter_context(open(filename, 'w'))
+        else:
+            f = sys.stdout
         yaml.round_trip_dump(node, f)