update how to add recipient

latchset · Apr 4, 2024 · fed8c42 · fed8c42
1 parent 7c4c70f
commit fed8c42
Showing 1 changed file with 5 additions and 8 deletions.
diff --git a/jwcrypto/jwe.py b/jwcrypto/jwe.py
@@ -253,10 +253,10 @@ def add_recipient(self, key, header=None):
         if 'ciphertext' not in self.objects:
             self._encrypt(alg, enc, jh)
 
-        if self.flattened:
-            if 'recipients' in self.objects:
-                self.objects['recipients'].append(rec)
-            elif 'encrypted_key' in self.objects or 'header' in self.objects:
+        if 'recipients' in self.objects:
+            self.objects['recipients'].append(rec)
+        elif self.flattened:
+            if 'encrypted_key' in self.objects or 'header' in self.objects:
                 self.objects['recipients'] = []
                 n = {}
                 if 'encrypted_key' in self.objects:
@@ -268,10 +268,7 @@ def add_recipient(self, key, header=None):
             else:
                 self.objects.update(rec)
         else:
-            if 'recipients' in self.objects:
-                self.objects['recipients'].append(rec)
-            else:
-                self.objects['recipients'] = [rec]
+            self.objects['recipients'] = [rec]
 
 
     def serialize(self, compact=False):