Merge pull request influxdata#12 from bbinet/db-list

jvshahid · jvshahid · commit 0462ad846b7f · 2014-03-03T13:11:56.000-05:00
add support for get_database_list method
diff --git a/influxdb/client.py b/influxdb/client.py
@@ -15,7 +15,8 @@ class InfluxDBClient(object):
     InfluxDB Client
     """
 
-    def __init__(self, host, port, username, password, database):
+    def __init__(self, host='localhost', port=8086, username='root',
+                 password='root', database=None):
         """
         Initialize client
         """
@@ -243,6 +244,24 @@ def delete_database(self, database):
             raise Exception(
                 "{0}: {1}".format(response.status_code, response.content))
 
+    # ### get list of databases
+    # curl -X GET http://localhost:8086/db
+
+    def get_database_list(self):
+        """
+        Get the list of databases
+        """
+        response = session.get("{0}/db?u={1}&p={2}".format(
+            self._baseurl,
+            self._username,
+            self._password))
+
+        if response.status_code == 200:
+            return json.loads(response.content)
+        else:
+            raise Exception(
+                "{0}: {1}".format(response.status_code, response.content))
+
     # Security
     # get list of cluster admins
     # curl http://localhost:8086/cluster_admins?u=root&p=root
diff --git a/tests/influxdb/client_test.py b/tests/influxdb/client_test.py
@@ -143,6 +143,21 @@ def test_delete_database_fails(self):
             cli = InfluxDBClient('host', 8086, 'username', 'password', 'db')
             cli.delete_database('old_db')
 
+    def test_get_database_list(self):
+        with patch.object(session, 'get') as mocked_get:
+            mocked_get.return_value = _build_response_object(
+                status_code=200, content='[{"name": "a_db"}]')
+            cli = InfluxDBClient('host', 8086, 'username', 'password')
+            assert len(cli.get_database_list()) == 1
+            assert cli.get_database_list()[0]['name'] == 'a_db'
+
+    @raises(Exception)
+    def test_get_database_list_fails(self):
+        with patch.object(session, 'get') as mocked_get:
+            mocked_get.return_value = _build_response_object(status_code=401)
+            cli = InfluxDBClient('host', 8086, 'username', 'password')
+            cli.get_database_list()
+
     def test_get_list_cluster_admins(self):
         pass
 
