fixed function names, as well as added a better check for the caller of get_image_download_script

Author: yankev

plotly/offline/offline.py

@@ -11,6 +11,7 @@
 import warnings
 from pkg_resources import resource_string
 import time
+import sys
 import webbrowser
 
 import plotly
@@ -49,17 +50,13 @@ def get_plotlyjs():
     plotlyjs = resource_string('plotly', path).decode('utf-8')
     return plotlyjs
 
-def get_image_download_script(caller):
+def get_image_download_script():
     '''
-    This function will return a script that will download an image of a Plotly
-    plot.
-
-    Keyword Arguments:
-    caller ('plot', 'iplot') -- specifies which function made the call for the
-        download script. If `iplot`, then an extra condition is added into the
-        download script to ensure that download prompts aren't iniitated on
-        page reloads.
+    This function will return a script that will download an image of the
+    most recently rendered Plotly plot
     '''
+    # get the name of the function that called `get_image_download_script`
+    caller = sys._getframe(1).f_code.co_name
 
     if caller == 'iplot':
         check_start = 'if(document.readyState == \'complete\') {{'
@@ -303,11 +300,11 @@ def iplot(figure_or_data, show_link=True, link_text='Export to plot.ly',
                                                  )
                              )
         # if image is given, and is a valid format, we will download the image
-        script = image_download_script('iplot').format(format=image,
-                                                       width=image_width,
-                                                       height=image_height,
-                                                       filename=filename,
-                                                       plot_id=plotdivid)
+        script = get_image_download_script().format(format=image,
+                                                    width=image_width,
+                                                    height=image_height,
+                                                    filename=filename,
+                                                    plot_id=plotdivid)
         # allow time for the plot to draw
         time.sleep(1)
         # inject code to download an image of the plot
@@ -426,7 +423,7 @@ def plot(figure_or_data,
                                      )
                 # if the check passes then download script is injected.
                 # write the download script:
-                script = image_download_script('plot')
+                script = get_image_download_script()
                 script = script.format(format=image,
                                        width=image_width,
                                        height=image_height,