Create creategroups.py

creates custom groups and establishes group membership from a csv file.

creategroups.py

@@ -0,0 +1,127 @@
+#!/usr/bin/python
+# -*- coding: utf-8 -*-
+#
+# creategroups.py
+# Februar 2020
+#
+# create custom groups and add members using a csv file as input
+# you can also add members to existing groups
+#
+# if the group already exists it will not be added and the http response is printed
+# if the user does not exist it will not be added to the group and the http response is printed
+# if the user is already a member of the group it will not be added to the group and the http response is printed
+#
+# none of the above conditions will prevent the processing of additional items in the csv
+#
+#
+# Change History
+#
+# 03feb2020 Initial development
+#
+# Format of csv file is two columns
+#  Column 1 group id 
+#  Column 2 group name 
+#  Column 3 is a description
+#  Column 4 member id
+#
+# For example:
+#group2,"Group 2","My Group 2"
+#group3,"Group 3","My Group3",geladm
+#group1,"Group 1","group 1"
+#
+# Copyright 2020, SAS Institute Inc., Cary, NC, USA.  All Rights Reserved.
+#
+#  Licensed under the Apache License, Version 2.0 (the License);
+#  you may not use this file except in compliance with the License.
+#  You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+#  Unless required by applicable law or agreed to in writing, software
+#  distributed under the License is distributed on an "AS IS" BASIS,
+#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#  See the License for the specific language governing permissions and
+#  limitations under the License.
+#
+import argparse
+import csv
+import os
+from sharedfunctions import callrestapi, getfolderid, file_accessible, getidsanduris
+
+# setup command-line arguements    
+parser = argparse.ArgumentParser(description="Create custom groups and establish membership")
+parser.add_argument("-f","--file", help="Full path to csv file containing groups ",required='True')
+args = parser.parse_args()
+file=args.file
+
+reqtype="post"
+
+check=file_accessible(file,'r')
+
+allgroups=callrestapi("/identities/groups","get")
+
+# create a list of all groups
+
+groupslist=[]
+
+if 'items' in allgroups:
+    
+        total_items=allgroups['count']
+        returned_items=len(allgroups['items'])
+             
+        for i in range(0,returned_items): 
+                                
+           groupslist.append(allgroups['items'][i]['id'])
+           
+# file can be read
+if check:
+
+    with open(file, 'rt') as f:
+        
+        filecontents = csv.reader(f)
+
+        for row in filecontents:
+            
+            #print(row)
+
+            cols=len(row)
+
+            id=row[0]
+            newgroup=row[1]
+            description=row[2]
+
+            data = {}
+            data["id"]=id
+            data['name'] = newgroup
+            data['description'] = description
+            data["state"]="active"
+
+            if id in groupslist:
+                 print("Note: group with name "+newgroup+"  and id "+id+" already exists." )
+
+            else:
+               
+                print ("Note: Trying to creating Group: "+newgroup )
+
+                reqtype='post'
+                reqval="/identities/groups/"
+
+                myresult=callrestapi(reqval,reqtype,data=data,stoponerror=0)
+
+                if myresult != None: print("Note: Group: "+newgroup+" created" )
+                      
+            # 4th column is group membership
+            if cols>=4:
+          
+                member=row[3]
+                print("Note: Trying to add user "+ member+ " to group "+newgroup )
+
+                reqval="/identities/groups/"+id+"/userMembers/"+member
+                reqtype='put'
+                myresult=callrestapi(reqval,reqtype,data=data,stoponerror=0)
+                if myresult != None: print("Note: user: "+member+" added to group "+newgroup )
+
+
+else:
+    print("ERROR: cannot read "+file)
+