Merge pull request influxdata#330 from babilen/show-privileges

aviau · web-flow · commit 045e76f81e28 · 2016-06-13T13:43:02.000-04:00
Add get_list_privileges() to InfluxDBClient
diff --git a/influxdb/client.py b/influxdb/client.py
@@ -740,6 +740,28 @@ def revoke_privilege(self, privilege, database, username):
                                                    username)
         self.query(text)
 
+    def get_list_privileges(self, username):
+        """Get the list of all privileges granted to given user.
+
+        :param username: the username to get privileges of
+        :type username: str
+
+        :returns: all privileges granted to given user
+        :rtype: list of dictionaries
+
+        :Example:
+
+        ::
+
+            >> privileges = client.get_list_privileges('user1')
+            >> privileges
+            [{u'privilege': u'WRITE', u'database': u'db1'},
+             {u'privilege': u'ALL PRIVILEGES', u'database': u'db2'},
+             {u'privilege': u'NO PRIVILEGES', u'database': u'db3'}]
+        """
+        text = "SHOW GRANTS FOR {0}".format(username)
+        return list(self.query(text).get_points())
+
     def send_packet(self, packet):
         """Send an UDP packet.
 
diff --git a/influxdb/tests/client_test.py b/influxdb/tests/client_test.py
@@ -798,6 +798,31 @@ def test_revoke_privilege_invalid(self):
         with _mocked_session(cli, 'get', 400):
             self.cli.revoke_privilege('', 'testdb', 'test')
 
+    def test_get_list_privileges(self):
+        data = {'results': [
+            {'series': [
+                {'columns': ['database', 'privilege'],
+                 'values': [
+                     ['db1', 'READ'],
+                     ['db2', 'ALL PRIVILEGES'],
+                     ['db3', 'NO PRIVILEGES']]}
+            ]}
+        ]}
+
+        with _mocked_session(self.cli, 'get', 200, json.dumps(data)):
+            self.assertListEqual(
+                self.cli.get_list_privileges('test'),
+                [{'database': 'db1', 'privilege': 'READ'},
+                 {'database': 'db2', 'privilege': 'ALL PRIVILEGES'},
+                 {'database': 'db3', 'privilege': 'NO PRIVILEGES'}]
+            )
+
+    @raises(Exception)
+    def test_get_list_privileges_fails(self):
+        cli = InfluxDBClient('host', 8086, 'username', 'password')
+        with _mocked_session(cli, 'get', 401):
+            cli.get_list_privileges('test')
+
     def test_invalid_port_fails(self):
         with self.assertRaises(ValueError):
             InfluxDBClient('host', '80/redir', 'username', 'password')
