From dba87f62df8fa4a6657f6cd38a3665ebbad1def1 Mon Sep 17 00:00:00 2001
From: nuwang <2070605+nuwang@users.noreply.github.com>
Date: Tue, 28 Feb 2023 11:25:42 +0530
Subject: [PATCH] refactor: Add optional openid connection constructor param to
 keycloak admin

---
 src/keycloak/keycloak_admin.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/keycloak/keycloak_admin.py b/src/keycloak/keycloak_admin.py
index a6ca09c..a955dc6 100644
--- a/src/keycloak/keycloak_admin.py
+++ b/src/keycloak/keycloak_admin.py
@@ -75,6 +75,8 @@ class KeycloakAdmin:
     :type auto_refresh_token: list
     :param timeout: connection timeout in seconds
     :type timeout: int
+    :param connection: A KeycloakOpenIDConnectionManager as an alternative to individual params.
+    :type connection: KeycloakOpenIDConnectionManager
     """
 
     PAGE_SIZE = 100
@@ -96,6 +98,7 @@ class KeycloakAdmin:
         user_realm_name=None,
         auto_refresh_token=None,
         timeout=60,
+        connection: KeycloakOpenIDConnectionManager = None,
     ):
         """Init method.
 
@@ -127,8 +130,10 @@ class KeycloakAdmin:
         :type auto_refresh_token: list
         :param timeout: connection timeout in seconds
         :type timeout: int
+        :param connection: An OpenID Connection as an alternative to individual params.
+        :type connection: KeycloakOpenIDConnectionManager
         """
-        self.connection = KeycloakOpenIDConnectionManager(
+        self.connection = connection or KeycloakOpenIDConnectionManager(
             server_url=server_url,
             username=username,
             password=password,