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

This commit is contained in:
Fred Drake 2001-05-22 14:58:10 +00:00
parent 706314909b
commit 0295929280
1 changed files with 20 additions and 0 deletions

View File

@ -421,6 +421,26 @@ class Message:
for i in list: for i in list:
del self.headers[i] 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): def has_key(self, name):
"""Determine whether a message contains the named header.""" """Determine whether a message contains the named header."""
return self.dict.has_key(name.lower()) return self.dict.has_key(name.lower())