"""A `safe' dictionary for string interpolation."""
from types import StringType
from UserDict import UserDict
COMMASPACE = ', '
class SafeDict(UserDict):
"""Dictionary which returns a default value for unknown keys.
This is used in maketext so that editing templates is a bit more robust.
"""
def __getitem__(self, key):
try:
return self.data[key]
except KeyError:
if isinstance(key, StringType):
return '%('+key+')s'
else:
return '<Missing key: %s>' % `key`
def interpolate(self, template):
return template % self
class MsgSafeDict(SafeDict):
def __init__(self, msg, dict=None):
self.__msg = msg
SafeDict.__init__(self, dict)
def __getitem__(self, key):
if key.startswith('msg_'):
return self.__msg.get(key[4:], 'n/a')
elif key.startswith('allmsg_'):
missing = []
all = self.__msg.get_all(key[7:], missing)
if all is missing:
return 'n/a'
return COMMASPACE.join(all)
else:
return SafeDict.__getitem__(self, key)
def copy(self):
d = self.data.copy()
for k in self.__msg.keys():
vals = self.__msg.get_all(k)
if len(vals) == 1:
d['msg_'+k.lower()] = vals[0]
else:
d['allmsg_'+k.lower()] = COMMASPACE.join(vals)
return d