elements: Allow exporting to DMS YAML

This allows for exporting the experiment to a YAML format which is
parsable by the DMS.

1 files changed, 9 insertions, 0 deletions

diff --git a/rumba/elements/experimentation.py b/rumba/elements/experimentation.py
index 9890975..2c6f478 100644
--- a/rumba/elements/experimentation.py
+++ b/rumba/elements/experimentation.py
@@ -521,6 +521,15 @@ class Experiment(object):
         end = time.time()
         logger.info("Swap-out took %.2f seconds", end - start)
 
+    def to_dms_yaml(self, filename, clean=True):
+        mode = 'w'
+        if not clean:
+            mode += '+'
+        with open(filename, mode) as f:
+            for node in self.nodes:
+                f.write("---\n")
+                node.to_dms_yaml(f)
+            f.write("...\n")
 
 class Executor:
     __metaclass__ = abc.ABCMeta