Support readFrom/writeTo in 'add_database_user'

jonashaag · jonashaag · commit c587bb2efef8 · 2014-10-29T15:33:49.000+01:00
diff --git a/influxdb/client.py b/influxdb/client.py
@@ -613,9 +613,11 @@ def get_database_users(self):
 
         return response.json()
 
-    def add_database_user(self, new_username, new_password):
+    def add_database_user(self, new_username, new_password, permissions=None):
         """
         Add database user
+
+        :param permissions: A ``(readFrom, writeTo)`` tuple
         """
         url = "db/{0}/users".format(self._database)
 
@@ -624,6 +626,12 @@ def add_database_user(self, new_username, new_password):
             'password': new_password
         }
 
+        if permissions:
+            try:
+                data['readFrom'], data['writeTo'] = permissions
+            except (ValueError, TypeError):
+                raise TypeError("'permissions' must be (readFrom, writeTo) tuple")
+
         self.request(
             url=url,
             method='POST',
