Merge pull request #36 from derekmwright/master

Fixes #35 - Preserve content headers on enveloped data
abhishek-ram · Sep 6, 2021 · c8cc68f · c8cc68f
2 parents ead0a64 + 69c551b
commit c8cc68f
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/pyas2lib/as2.py b/pyas2lib/as2.py
@@ -655,7 +655,12 @@ def parse(self, raw_content, find_org_cb, find_partner_cb, find_message_cb=None)
 
         # Update the payload headers with the original headers
         for k, v in as2_headers.items():
-            if self.payload.get(k) and k.lower() != "content-disposition":
+            preserve = [
+                "content-type",
+                "content-disposition",
+                "content-transfer-encoding",
+            ]
+            if self.payload.get(k) and k.lower() not in preserve:
                 del self.payload[k]
             self.payload.add_header(k, v)