Updates in status and pipeline debugging

[moe.git] / t / moe / status.py

diff --git a/t/moe/status.py b/t/moe/status.py
index 3d72db476de48df25caf8c566cb1f14a429f922d..320d7c9308d566d5082c0217462f05fd9a6fdbd8 100644 (file)
--- a/t/moe/status.py
+++ b/t/moe/status.py
@@ -37,6 +37,21 @@ class MoeStatus:
            m[k] = [m[k]]
        return m[k]
 
+    def dump(self, prefix=""):
+       "Return a list of lines, `prefix` is indent prefix."
+       l = []
+        for k,v in self.stat.items():
+           if type(v) == types.ListType: vals = v
+           else: vals = [v]
+           for w in vals:
+               if isinstance(w, MoeStatus):
+                   l.append(prefix + k + "(\n")
+                   l.extend(self.str_lines(prefix+"  "))
+                   l.append(prefix + ")\n")
+               else:
+                   l.append(prefix + k + ":" + str(w) + "\n")
+       return l
+       
     def write(self, file=None, name=None):
        if file is None:
            if name is not None: