Tiny export tuning

Author: blokhin

mpds_client/export_MPDS.py

@@ -39,10 +39,10 @@ def _gen_basename(cls):
 
     @classmethod
     def _get_title(cls, term):
-        return cls.human_names.get(term, term)
+        return cls.human_names.get(term, term.capitalize())
 
     @classmethod
-    def save_plot(cls, data, columns, plottype, fmt='json'):
+    def save_plot(cls, data, columns, plottype, fmt='json', **kwargs):
         """
         Exports the data in the following formats for plotting:
 
@@ -103,6 +103,9 @@ def save_plot(cls, data, columns, plottype, fmt='json'):
                         plot["payload"]["meshes"][-1]["y"].append(row[pointers[1]])
                         plot["payload"]["meshes"][-1]["z"].append(row[pointers[2]])
 
+            if kwargs:
+                plot["payload"].update(kwargs)
+
             else: raise RuntimeError("\r\nError: %s is an unknown plot type" % plottype)
 
             f_export.write(json.dumps(plot, escape_forward_slashes=False, indent=4))