From 21799f9a992a4540d3ae062f0be51dc9aae052da Mon Sep 17 00:00:00 2001
From: Remco Kranenburg <remco@burgsoft.nl>
Date: Thu, 21 Jun 2018 13:02:17 +0200
Subject: [PATCH] Add get_user_groups function

---
 keycloak/keycloak_admin.py | 14 +++++++++++++-
 keycloak/urls_patterns.py  |  1 +
 2 files changed, 14 insertions(+), 1 deletion(-)

diff --git a/keycloak/keycloak_admin.py b/keycloak/keycloak_admin.py
index 17e2c12..74f56e8 100644
--- a/keycloak/keycloak_admin.py
+++ b/keycloak/keycloak_admin.py
@@ -23,7 +23,7 @@ from .urls_patterns import \
     URL_ADMIN_SEND_UPDATE_ACCOUNT, URL_ADMIN_RESET_PASSWORD, URL_ADMIN_SEND_VERIFY_EMAIL, URL_ADMIN_GET_SESSIONS, \
     URL_ADMIN_SERVER_INFO, URL_ADMIN_CLIENTS, URL_ADMIN_CLIENT, URL_ADMIN_CLIENT_ROLES, URL_ADMIN_REALM_ROLES, \
     URL_ADMIN_GROUP, URL_ADMIN_GROUPS, URL_ADMIN_GROUP_CHILD, URL_ADMIN_USER_GROUP,\
-    URL_ADMIN_GROUP_PERMISSIONS, URL_ADMIN_USER_CLIENT_ROLES, URL_ADMIN_USER_STORAGE
+    URL_ADMIN_USER_GROUPS, URL_ADMIN_GROUP_PERMISSIONS, URL_ADMIN_USER_CLIENT_ROLES, URL_ADMIN_USER_STORAGE
 
 from .keycloak_openid import KeycloakOpenID
 
@@ -186,6 +186,18 @@ class KeycloakAdmin:
         data_raw = self.connection.raw_get(URL_ADMIN_USER.format(**params_path))
         return raise_error_from_response(data_raw, KeycloakGetError)
 
+    def get_user_groups(self, user_id):
+        """
+        Get user groups Returns a list of groups of which the user is a member
+
+        :param user_id: User id
+
+        :return: user groups list
+        """
+        params_path = {"realm-name": self.realm_name, "id": user_id}
+        data_raw = self.connection.raw_get(URL_ADMIN_USER_GROUPS.format(**params_path))
+        return raise_error_from_response(data_raw, KeycloakGetError)
+
     def update_user(self, user_id, payload):
         """
         Update the user
diff --git a/keycloak/urls_patterns.py b/keycloak/urls_patterns.py
index ce593da..6c9a827 100644
--- a/keycloak/urls_patterns.py
+++ b/keycloak/urls_patterns.py
@@ -35,6 +35,7 @@ URL_ADMIN_RESET_PASSWORD = "admin/realms/{realm-name}/users/{id}/reset-password"
 URL_ADMIN_GET_SESSIONS = "admin/realms/{realm-name}/users/{id}/sessions"
 URL_ADMIN_USER_CLIENT_ROLES = "admin/realms/{realm-name}/users/{id}/role-mappings/clients/{client-id}"
 URL_ADMIN_USER_GROUP = "admin/realms/{realm-name}/users/{id}/groups/{group-id}"
+URL_ADMIN_USER_GROUPS = "admin/realms/{realm-name}/users/{id}/groups"
 
 URL_ADMIN_SERVER_INFO = "admin/serverinfo"