From: Fred Drake <fdrake@acm.org>
Date: Tue, 22 May 2001 14:58:10 +0000 (+0000)
Subject: Added .get() and .setdefault() support to rfc822.Message.
X-Git-Tag: v2.2a3~1713
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=02959292803415fc5d66838735764aa84de8c0f8;p=thirdparty%2FPython%2Fcpython.git

Added .get() and .setdefault() support to rfc822.Message.
---

diff --git a/Lib/rfc822.py b/Lib/rfc822.py
index 101013dfb04c..9093beb211f9 100644
--- a/Lib/rfc822.py
+++ b/Lib/rfc822.py
@@ -421,6 +421,26 @@ class Message:
         for i in list:
             del self.headers[i]
 
+    def get(self, name, default=None):
+        name = name.lower()
+        if self.dict.has_key(name):
+            return self.dict[name]
+        else:
+            return default
+
+    def setdefault(self, name, default=''):
+        lowername = name.lower()
+        if self.dict.has_key(lowername):
+            return self.dict[lowername]
+        else:
+            default = default or ""
+            self.dict[lowername] = default
+            text = "%s: %s" % (name, default)
+            lines = text.split("\n")
+            for line in lines:
+                self.headers.append(line + "\n")
+            return default
+
     def has_key(self, name):
         """Determine whether a message contains the named header."""
         return self.dict.has_key(name.lower())