support newer version of requests

kolanos · Mar 6, 2013 · 625159c · 625159c
1 parent 55835cb
commit 625159c
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/vaporize/core.py b/vaporize/core.py
@@ -83,7 +83,14 @@ def connect(user, apikey, region='DFW'):
                         url = endpoint['publicURL']
             _settings[name + '_url'] = url
         auth = Auth(_settings['token'])
-        _session = requests.session(auth=auth, headers=headers)
+        try:
+            _session = requests.session(auth=auth, headers=headers)
+        except TypeError:
+            _session = requests.Session()
+            _session.auth = auth
+            _session.headers.update(headers)
+        except Exception, e:
+            raise
     else:
         raise ConnectionError("HTTP %d: %s" % (response.status_code,
                                                response.content))