add_user supports custom attributes (#301)

* add user supports custom attributes

* small fix

* fail early

Author: deshmukhrajvardhan

duo_client/admin.py

@@ -843,7 +843,7 @@ def get_users_by_ids(self, user_ids):
     def add_user(self, username, realname=None, status=None,
                  notes=None, email=None, firstname=None, lastname=None,
                  alias1=None, alias2=None, alias3=None, alias4=None,
-                 aliases=None):
+                 aliases=None, custom_attribute_map=None):
         """
         Adds a user.
 
@@ -856,6 +856,9 @@ def add_user(self, username, realname=None, status=None,
         lastname - User's surname for ID Proofing (optional)
         alias1..alias4 - Aliases for the user's primary username (optional)
         aliases - Aliases for the user's primary username (optional)
+        custom_attribute_map - Map of custom attributes (optional). When provided this will be of type Dict[str|str]. e.g.
+        {"attribute_name":"attribute_value"}
+        Note: the custom attribute names have to be created prior to adding users
 
         Returns newly created user object.
 
@@ -886,6 +889,10 @@ def add_user(self, username, realname=None, status=None,
             params['alias4'] = alias4
         if aliases is not None:
             params['aliases'] = aliases
+        if custom_attribute_map is not None:
+            for key in custom_attribute_map:
+                params[f'custom_attributes.{key}'] = custom_attribute_map[key]
+        
         response = self.json_api_call('POST',
                                       '/admin/v1/users',
                                       params)

examples/Admin/create_user_and_phone.py

@@ -1,6 +1,7 @@
 #!/usr/bin/python
 import pprint
 import sys
+import json
 
 import duo_client
 
@@ -20,6 +21,7 @@ def get_next_arg(prompt):
 
 USERNAME = get_next_arg('user login name: ')
 REALNAME = get_next_arg('user full name: ')
+CUSTOM_ATTRIBUTE_MAP_STR = get_next_arg('custom attribute map (optional): ')
 
 # Refer to http://www.duosecurity.com/docs/adminapi for more
 # information about phone types and platforms.
@@ -31,6 +33,7 @@ def get_next_arg(prompt):
 user = admin_api.add_user(
     username=USERNAME,
     realname=REALNAME,
+    custom_attribute_map=json.loads(CUSTOM_ATTRIBUTE_MAP_STR) if CUSTOM_ATTRIBUTE_MAP_STR else None ,
 )
 print('Created user:')
 pprint.pprint(user)