replace_header(): New method given by Skip Montanaro in SF patch
#601959. Modified slightly by Barry (who added the KeyError in case
the header is missing.
diff --git a/Lib/email/Message.py b/Lib/email/Message.py
index 65e3c6d..0a8d90b 100644
--- a/Lib/email/Message.py
+++ b/Lib/email/Message.py
@@ -350,7 +350,6 @@
Example:
msg.add_header('content-disposition', 'attachment', filename='bud.gif')
-
"""
parts = []
for k, v in _params.items():
@@ -362,6 +361,21 @@
parts.insert(0, _value)
self._headers.append((_name, SEMISPACE.join(parts)))
+ def replace_header(self, _name, _value):
+ """Replace a header.
+
+ Replace the first matching header found in the message, retaining
+ header order and case. If no matching header was found, a KeyError is
+ raised.
+ """
+ _name = _name.lower()
+ for i, (k, v) in zip(range(len(self._headers)), self._headers):
+ if k.lower() == _name:
+ self._headers[i] = (k, _value)
+ break
+ else:
+ raise KeyError, _name
+
#
# These methods are silently deprecated in favor of get_content_type() and
# friends (see below). They will be noisily deprecated in email 3.0.