Refactor

bakdata · Feb 26, 2024 · ae05cfb · ae05cfb
1 parent 4050978
commit ae05cfb
Showing 1 changed file with 18 additions and 17 deletions.
diff --git a/keycloak/oauth.py b/keycloak/oauth.py
@@ -25,7 +25,7 @@ class KeycloakOAuth2:
     def __init__(
         self,
         client_id: str,
-        client_secret: str | bytes,
+        client_secret: str | bytes | None,
         server_metadata_url: str,
         client_kwargs: dict[str, Any],
         base_url: str = "/",
@@ -39,24 +39,25 @@ def __init__(
 
         # TODO pass properly
         # Generated via `openssl genrsa - out keypair.pem 2048`
-        client_secret = Path("keypair.pem").read_bytes()
+        if not client_secret:
+            client_secret = Path("keypair.pem").read_bytes()
 
-        # Generated via `openssl rsa -in keypair.pem -pubout -out publickey.crt`
-        self.pub = JsonWebKey.import_key(
-            Path("publickey.crt").read_text(), {"kty": "RSA", "use": "sig"}
-        ).as_dict()
+            # Generated via `openssl rsa -in keypair.pem -pubout -out publickey.crt`
+            self.pub = JsonWebKey.import_key(
+                Path("publickey.crt").read_text(), {"kty": "RSA", "use": "sig"}
+            ).as_dict()
 
-        # TODO call self.keycloak.load_server_metadata() and get token_endpoint
-        token_endpoint = (
-            "http://localhost:8180/realms/daewy/protocol/openid-connect/token"
-        )
-        auth_method = PrivateKeyJWT(token_endpoint)
-        client_kwargs.update(
-            {
-                "client_auth_methods": [auth_method],
-                "token_endpoint_auth_method": auth_method.name,
-            }
-        )
+            # TODO call self.keycloak.load_server_metadata() and get token_endpoint
+            token_endpoint = (
+                "http://localhost:8180/realms/daewy/protocol/openid-connect/token"
+            )
+            auth_method = PrivateKeyJWT(token_endpoint)
+            client_kwargs.update(
+                {
+                    "client_auth_methods": [auth_method],
+                    "token_endpoint_auth_method": auth_method.name,
+                }
+            )
 
         oauth.register(
             name="keycloak",