Add krakenex.API.set_connection() method to set default connection.

Otherwise, setting a connection within a function won't work once the
function has returned (lost scope).

krakenex/api.py

@@ -38,18 +38,21 @@ class API:
     query_private
     
     """
-    def __init__(self, key = '', secret = ''):
+    def __init__(self, key = '', secret = '', conn = None):
         """Create an object with authentication information.
         
         Arguments:
         key    -- key required to make queries to the API (default: '')
         secret -- private key used to sign API messages (default: '')
+        conn   -- krakenex.Connection object for connection reuse
+                  (default: None)
         
         """
         self.key = key
         self.secret = secret
         self.uri = 'https://api.kraken.com'
         self.apiversion = '0'
+        self.conn = conn
 
 
     def load_key(self, path):
@@ -64,6 +67,16 @@ def load_key(self, path):
         self.secret = f.readline().strip()
 
 
+    def set_connection(self, conn):
+        """Set an existing connection to be used as a default in queries.
+
+        Argument:
+        conn -- connection (krakenex.Connection object, no default)
+
+        """
+        self.conn = conn
+
+
     def _query(self, urlpath, req = {}, conn = None, headers = {}):
         """Low-level query handling.
         
@@ -77,7 +90,10 @@ def _query(self, urlpath, req = {}, conn = None, headers = {}):
         url = self.uri + urlpath
 
         if conn is None:
-            conn = connection.Connection()
+            if self.conn is None:
+                conn = connection.Connection()
+            else:
+                conn = self.conn
 
         ret = conn._request(url, req, headers)
         return json.loads(ret)

krakenex/connection.py

@@ -34,7 +34,7 @@ def __init__(self, uri = 'api.kraken.com', timeout = 30):
         timeout -- blocking operations' timeout in seconds (default: 30)
         """
         self.headers = {
-            'User-Agent': 'krakenex/0.0.4.1 (+https://github.com/veox/python3-krakenex)'
+            'User-Agent': 'krakenex/0.0.5 (+https://github.com/veox/python3-krakenex)'
         }
 
         self.conn = http.client.HTTPSConnection(uri, timeout = timeout)

setup.py

@@ -3,7 +3,7 @@
 from distutils.core import setup
 
 setup(name='krakenex',
-      version='0.0.4.1',
+      version='0.0.5',
       description='kraken.com cryptocurrency exchange API',
       author='Noel Maersk',
       author_email='veox@wemakethings.net',