variable api domain

Author: chriddyp

plotly/plotly/plotly.py

@@ -900,8 +900,8 @@ def response_handler(cls, response):
 
     @classmethod
     def api_url(cls, resource):
-        # TODO: Variable URL
-        return 'https://api-local.plot.ly/v2/{}'.format(resource)
+        return '{}/v2/{}'.format(tools.get_config_file()['plotly_api_domain'],
+            resource)
 
     @classmethod
     def headers(cls):
@@ -924,6 +924,7 @@ def _get_session_username_and_key():
         raise exceptions.PlotlyLocalCredentialsError()
     return username, api_key
 
+
 def _send_to_plotly(figure, **plot_options):
     """
     """

plotly/tools.py

@@ -49,7 +49,8 @@ def warning_on_one_line(message, category, filename, lineno, file=None, line=Non
                                     'api_key': '',
                                     'stream_ids': []},
                  CONFIG_FILE: {'plotly_domain': 'https://plot.ly',
-                               'plotly_streaming_domain': 'stream.plot.ly'}}
+                               'plotly_streaming_domain': 'stream.plot.ly',
+                               'plotly_api_domain': 'https://api.plot.ly'}}
 
 try:
     os.mkdir(TEST_DIR)
@@ -139,7 +140,9 @@ def reset_credentials_file():
 
 ### config tools ###
 
-def set_config_file(plotly_domain=None, plotly_streaming_domain=None):
+def set_config_file(plotly_domain=None,
+                    plotly_streaming_domain=None,
+                    plotly_api_domain=None):
     """Set the keyword-value pairs in `~/.plotly/.config`.
 
     """
@@ -152,6 +155,8 @@ def set_config_file(plotly_domain=None, plotly_streaming_domain=None):
         settings['plotly_domain'] = plotly_domain
     if isinstance(plotly_streaming_domain, six.string_types):
         settings['plotly_streaming_domain'] = plotly_streaming_domain
+    if isinstance(plotly_api_domain, six.string_types):
+        settings['plotly_api_domain'] = plotly_api_domain
     utils.save_json_dict(CONFIG_FILE, settings)
     ensure_local_plotly_files()  # make sure what we just put there is OK
 
@@ -183,7 +188,7 @@ def reset_config_file():
 
 def get_embed(file_owner_or_url, file_id=None, width="100%", height=525):
     """Returns HTML code to embed figure on a webpage as an <iframe>
-    
+
     Plotly uniquely identifies figures with a 'file_owner'/'file_id' pair.
     Since each file is given a corresponding unique url, you may also simply
     pass a valid plotly url as the first argument.
@@ -262,7 +267,7 @@ def get_embed(file_owner_or_url, file_id=None, width="100%", height=525):
 
 def embed(file_owner_or_url, file_id=None, width="100%", height=525):
     """Embeds existing Plotly figure in IPython Notebook
-    
+
     Plotly uniquely identifies figures with a 'file_owner'/'file_id' pair.
     Since each file is given a corresponding unique url, you may also simply
     pass a valid plotly url as the first argument.