rewrote create_animations to take frames - still beta

Author: Kully

plotly/plotly/plotly.py

@@ -1407,8 +1407,7 @@ def _send_to_plotly(figure, **plot_options):
                              fileopt=plot_options['fileopt'],
                              world_readable=plot_options['world_readable'],
                              sharing=plot_options['sharing'],
-                             layout=fig['layout'] if 'layout' in fig else {},
-                             frames=fig['frames'] if 'frames' in fig else {}),
+                             layout=fig['layout'] if 'layout' in fig else {}),
                         cls=utils.PlotlyJSONEncoder)
 
     # TODO: It'd be cool to expose the platform for RaspPi and others
@@ -1419,9 +1418,9 @@ def _send_to_plotly(figure, **plot_options):
                    key=api_key,
                    origin='plot',
                    kwargs=kwargs)
-
-    if 'frames' in kwargs:
-        r = create_animations(fig, kwargs, payload)
+    #print fig
+    #if 'frames' in fig:
+    #    r = create_animations(fig, kwargs, payload)
 
     url = get_config()['plotly_domain'] + "/clientresp"
 
@@ -1449,10 +1448,6 @@ def _send_to_plotly(figure, **plot_options):
 
 
 def create_animations(fig, kwargs, payload):
-    """
-    The plan is to create a grid then a plot by deconstructing the fig if
-    frames is found in the kwargs.
-    """
     url_v2_plot = "https://api.plot.ly/v2/plots"
     url_v2_grid = "https://api.plot.ly/v2/grids"
     auth = HTTPBasicAuth(str(payload['un']), str(payload['key']))
@@ -1462,8 +1457,12 @@ def create_animations(fig, kwargs, payload):
     if 'layout' not in fig:
         fig['layout'] = {}
 
+    # make a copy of fig
+    fig_copy = copy.deepcopy(fig)
+
     # make grid
     cols_dict = {}
+    frames_cols_dict = {}
     counter = 0
     trace_num = 0
     for trace in fig['data']:
@@ -1481,6 +1480,22 @@ def create_animations(fig, kwargs, payload):
             counter += 1
         trace_num += 1
 
+    # add frames data to grid
+    for j in range(len(fig['frames'])):
+        for var in ['x', 'y']:
+            if 'name' in fig['frames'][j]['data']:
+                cols_dict["{name}, {x_or_y}".format(name=fig['frames'][j]['data'][0]['name'],
+                                                    x_or_y=var)] = {
+                    "data": list(fig['frames'][j]['data'][0][var]), "order": counter
+                }
+            else:
+                cols_dict["Trace {num}, {x_or_y}".format(num=trace_num,
+                                                         x_or_y=var)] = {
+                    "data": list(fig['frames'][j]['data'][0][var]), "order": counter
+                }
+            counter += 1
+        trace_num += 1
+
     grid_info = {
         "data": {"cols": cols_dict},
         "world_readable": True
@@ -1489,8 +1504,6 @@ def create_animations(fig, kwargs, payload):
     r = requests.post('https://api.plot.ly/v2/grids', auth=auth,
                       headers=headers, json=grid_info)
     r_dict = json.loads(r.text)
-    # make a copy of the fig so that we not altering the original fig
-    fig_copy = copy.deepcopy(fig)
 
     # make plot
     fid = r_dict['file']['fid']
@@ -1509,13 +1522,29 @@ def create_animations(fig, kwargs, payload):
         )
         cols_index += 2
 
+    # replace data in frames by grid ids
+    for j in range(len(fig['frames'])):
+        if 'x' in fig_copy['frames'][j]['data'][0]:
+            del fig_copy['frames'][j]['data'][0]['x']
+        if 'y' in fig_copy['frames'][j]['data'][0]:
+            del fig_copy['frames'][j]['data'][0]['y']
+
+        fig_copy['frames'][j]['data'][0]["xsrc"] = "{fid}:{idlocal}".format(
+            fid=fid, idlocal=r_dict['file']['cols'][cols_index]['uid']
+        )
+        fig_copy['frames'][j]['data'][0]["ysrc"] = "{fid}:{idlocal}".format(
+            fid=fid, idlocal=r_dict['file']['cols'][cols_index + 1]['uid']
+        )
+        cols_index += 2
+
     plots_info = {
         "figure": fig_copy,
         "world_readable": True
     }
 
-    return requests.post('https://api.plot.ly/v2/plots', auth=auth,
-                         headers=headers, json=plots_info)
+    requests.post('https://api.plot.ly/v2/plots', auth=auth,
+                  headers=headers, json=plots_info)
+
 
 
 def _open_url(url):