make folder if it doesnt exist

Author: chriddyp

plotly/plotly/__init__.py

@@ -9,5 +9,6 @@
 """
 from . plotly import (
     sign_in, update_plot_options, get_plot_options, get_credentials, iplot,
-    plot, iplot_mpl, plot_mpl, get_figure, Stream, image, grid_ops
+    plot, iplot_mpl, plot_mpl, get_figure, Stream, image, grid_ops,
+    meta_ops, file_ops, _api_v2
 )

plotly/plotly/plotly.py

@@ -628,38 +628,35 @@ def save_as(cls, figure_or_data, filename, format=None, width=None, height=None)
         f.write(img)
         f.close()
 
+
 class file_ops:
     """ Interface to Plotly's File System API
     """
 
     @classmethod
-    def mkdir(cls, folder_path):
+    def mkdirs(cls, folder_path):
         """ Make a folder in Plotly at folder_path
-            Mimics the shell's mkdir.
-            Exceptions:
-            - Parent folder doesn't exist
-            - A file already exists with that same name
+            Mimics the shell's mkdir -p.
+            Returns:
+            - 200 if folders already existed, nothing was created
+            - 201 if path was created
+            Raises:
+            -  requests.exceptions.RequestException: 400
+
             Usage examples:
-            >> mkdir('new folder')
-            >> mkdir('existing folder/new folder')
+            >> mkdirs('new folder')
+            >> mkdirs('existing folder/new folder')
+            >> mkdirs('new/folder/path')
         """
 
-        # trim trailing slash
+        # trim trailing slash TODO: necessesary?
         if folder_path[-1] == '/':
             folder_path = folder_path[0:-1]
 
-        folder_names = folder_path.split('/')
-        name = folder_names[-1]
         payload = {
-            'name': name
+            'path': folder_path
         }
 
-        parent_path = '/'.join(folder_names[0:-1])
-        if parent_path == '':
-            payload['parent'] = -1
-        else:
-            payload['parent_path'] = parent_path
-
         url = _api_v2.api_url('folders')
 
         res = requests.post(url, data=payload, headers=_api_v2.headers())
@@ -684,9 +681,19 @@ def _fill_in_response_column_ids(cls, request_columns,
     @classmethod
     def upload(cls, grid, filename, world_readable=True, auto_open=True, meta=None):
         """ Upload a grid to your Plotly account with the specified filename.
-
         """
 
+        # Made a folder path
+        if filename[-1] == '/':
+            filename = filename[0:-1]
+
+        paths = filename.split('/')
+        parent_path = '/'.join(paths[0:-1])
+        filename = paths[-1]
+
+        if parent_path != '':
+            file_ops.mkdirs(parent_path)
+
         # transmorgify grid object into plotly's format
         grid_json = {'cols': {}}
         if meta is not None:
@@ -704,8 +711,13 @@ def upload(cls, grid, filename, world_readable=True, auto_open=True, meta=None):
             'world_readable': world_readable
         }
 
+        if parent_path != '':
+            payload['parent_path'] = parent_path
+
         upload_url = _api_v2.api_url('grids')
-        req = requests.post(upload_url, data=payload, headers=_api_v2.headers())
+        req = requests.post(upload_url, data=payload,
+                            headers=_api_v2.headers())
+
         res = _api_v2.response_handler(req)
 
         response_columns = res['file']['cols']