From 7f6f75b58b73e342618296d150cab57db072ef46 Mon Sep 17 00:00:00 2001
From: Jean-Michel Crepel <jean-michel.crepel@camptocamp.com>
Date: Tue, 21 Jan 2025 12:23:58 +0100
Subject: [PATCH] manage creation and update role's user with the console

---
 console/console.py | 38 +++++++++++++++++++++++++++++++++++++-
 1 file changed, 37 insertions(+), 1 deletion(-)

diff --git a/console/console.py b/console/console.py
index d24ce59..d1c5155 100644
--- a/console/console.py
+++ b/console/console.py
@@ -114,7 +114,43 @@ def delorgs(self, id):
     # def updateorgs(self):
 
     # ROLES
-    # def createnewroles(self):
+    def createnewroles(self, cn="NEWROLE", description="Default description"):
+        newrole = {
+            "cn": cn,
+            "description": description,
+        }
+        url = self.server + "/console/private/roles"
+
+        response = self.session.post(url,
+                                     data=json.dumps(newrole),
+                                     auth=(self.username, self.password),
+                                     headers=self.headers)
+
+        if (response.status_code == 200):
+            return response.text
+        else:
+            print(response.status_code)
+            return "Error while creating the role"
+
+    def updaterolesuser(self, uuid="testadmin", cn="ROLE"):
+        updateroles = {
+            "users": [uuid],
+            "PUT": [cn],
+            "DELETE": []
+        }
+        url = self.server + "/console/private/roles_users"
+
+        response = self.session.post(url,
+                                     data=json.dumps(updateroles),
+                                     auth=(self.username, self.password),
+                                     headers=self.headers)
+
+        if (response.status_code == 200):
+            return response.text
+        else:
+            print(response.status_code)
+            return "Error while updating the role's users"
+
     def getroles(self, uid=""):
         headers = {'Accept': 'application/json'}
         url = self.server + "/console/private/roles/" + uid