add plot command to widgets

chriddyp · chriddyp · commit c5677c82e8cf · 2015-02-16T22:32:22.000-05:00
diff --git a/plotly/widgets/graph_widget.py b/plotly/widgets/graph_widget.py
@@ -1,14 +1,13 @@
 from collections import deque
 import json
-import os
 import uuid
 
 # TODO: protected imports?
 from IPython.html import widgets
 from IPython.utils.traitlets import Unicode
 from IPython.display import Javascript, display
 
-from plotly import utils
+from plotly import utils, tools
 from pkg_resources import resource_string
 
 # Load JS widget code
@@ -247,6 +246,18 @@ def message_handler(widget, ranges):
         """
         self._handle_registration('zoom', callback, remove)
 
+    def plot(self, figure_or_data, validate=True):
+        """Plot figure_or_data in the Plotly graph.
+        """
+        figure = tools.return_figure_from_figure_or_data(figure_or_data, validate)
+        message = {
+            'task': 'newPlot',
+            'data': figure['data'],
+            'layout': figure.get('layout', {}),
+            'graphId': self._graphId
+        }
+        self._handle_outgoing_message(message)
+
     def restyle(self, data, indices=None):
         """Update the style of existing traces in the Plotly graph.
 
