fix(jwe): fix when there is no encrypted_key

authlib · Jul 9, 2023 · 1e77df0 · 1e77df0
1 parent 78f4c9d
commit 1e77df0
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/joserfc/rfc7516/json.py b/src/joserfc/rfc7516/json.py
@@ -60,11 +60,13 @@ def extract_json(data: JSONSerialization) -> JSONEncryption:
         obj.flatten = False
         for item in data["recipients"]:
             recipient = Recipient(obj, item.get("header"))
-            recipient.encrypted_key = urlsafe_b64decode(to_bytes(item["encrypted_key"]))
+            if "encrypted_key" in item:
+                recipient.encrypted_key = urlsafe_b64decode(to_bytes(item["encrypted_key"]))
             obj.recipients.append(recipient)
     else:
         obj.flatten = True
         recipient = Recipient(obj, data.get("header"))
-        recipient.encrypted_key = urlsafe_b64decode(to_bytes(data["encrypted_key"]))
+        if "encrypted_key" in data:
+            recipient.encrypted_key = urlsafe_b64decode(to_bytes(data["encrypted_key"]))
         obj.recipients.append(recipient)
     return obj