mirror of https://github.com/python/cpython
Added .get() and .setdefault() support to rfc822.Message.
This commit is contained in:
parent
706314909b
commit
0295929280
|
@ -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())
|
||||||
|
|
Loading…
Reference in New Issue