From 02959292803415fc5d66838735764aa84de8c0f8 Mon Sep 17 00:00:00 2001 From: Fred Drake Date: Tue, 22 May 2001 14:58:10 +0000 Subject: Added .get() and .setdefault() support to rfc822.Message. --- Lib/rfc822.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) diff --git a/Lib/rfc822.py b/Lib/rfc822.py index 101013d..9093beb 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()) -- cgit v0.12