fix: default return for get_key

italia · Mar 12, 2024 · 215e0a6 · 215e0a6
1 parent a06b7b8
commit 215e0a6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/spid_cie_oidc/entity/jwks.py b/spid_cie_oidc/entity/jwks.py
@@ -59,9 +59,9 @@ def public_pem_from_jwk(jwk_dict: dict):
 def serialize_rsa_key(rsa_key, kind="public", hash_func="SHA-256"):
     """
     rsa_key can be
-        cryptography.hazmat.backends.openssl.rsa._RSAPublicKey
+         cryptography.hazmat.primitives.asymmetric.rsa.RSAPublicKey
         or
-        cryptography.hazmat.backends.openssl.rsa._RSAPrivateKey
+         cryptography.hazmat.primitives.asymmetric.rsa.RSAPrivateKey
     """
     data = {}
     if isinstance(rsa_key, rsa.RSAPublicKey):

diff --git a/spid_cie_oidc/entity/utils.py b/spid_cie_oidc/entity/utils.py
@@ -15,11 +15,10 @@
 
 
 def get_key(jwks, use=KeyUsage.signature):
-    selected_jwk = jwks[0]
     for jwk in jwks:
         if jwk['use'] == use:
             return jwk
-
+    return jwks[0]
 def iat_now() -> int:
     return int(datetime.datetime.now().timestamp())