Move from genimage —> v2/images/

zhenfenghan · Mar 8, 2015 · 5139385 · 5139385
1 parent ce0adcc
commit 5139385
Showing 1 changed file with 9 additions and 11 deletions.
diff --git a/plotly/plotly/plotly.py b/plotly/plotly/plotly.py
@@ -24,6 +24,7 @@
 import six
 import base64
 import requests
+from requests.auth import HTTPBasicAuth
 
 if sys.version[:1] == '2':
     from urlparse import urlparse
@@ -582,25 +583,22 @@ def get(figure_or_data, format='png', width=None, height=None):
         credentials = get_credentials()
         validate_credentials(credentials)
         username, api_key = credentials['username'], credentials['api_key']
-        headers = {'plotly-username': username,
-                   'plotly-apikey': api_key,
-                   'plotly-version': version.__version__,
-                   'plotly-platform': 'python'}
-
-        payload = {
-            'figure': figure,
-            'format': format
-        }
+        headers = {'Plotly-Version': version.__version__,
+                   'Content-Type': 'application/json',
+                   'Plotly-Client-Platform': 'python'}
 
+        payload = {'figure': figure, 'format': format}
         if width is not None:
             payload['width'] = width
         if height is not None:
             payload['height'] = height
 
-        url = get_config()['plotly_domain'] + "/apigenimage/"
+        url = _api_v2.api_url('images/')
+
         res = requests.post(
             url, data=json.dumps(payload, cls=utils.PlotlyJSONEncoder),
-            headers=headers, verify=get_config()['plotly_ssl_verification']
+            headers=headers, verify=get_config()['plotly_ssl_verification'],
+            auth=HTTPBasicAuth(username, api_key)
         )
 
         headers = res.headers