blob: 9b6e73b50f26428eb329482674d713b99a3bd5d6 [file] [log] [blame]
Guido van Rossumaad67612000-05-08 17:31:04 +00001"""A more or less complete user-defined wrapper around dictionary objects."""
Guido van Rossum5c971671996-07-22 15:23:25 +00002
3class UserDict:
Guido van Rossum9a744a91999-04-08 20:27:54 +00004 def __init__(self, dict=None):
5 self.data = {}
6 if dict is not None: self.update(dict)
Guido van Rossumff712aa1997-08-15 00:45:26 +00007 def __repr__(self): return repr(self.data)
8 def __cmp__(self, dict):
Guido van Rossum9a744a91999-04-08 20:27:54 +00009 if isinstance(dict, UserDict):
Guido van Rossum548703a1998-03-26 22:14:20 +000010 return cmp(self.data, dict.data)
Guido van Rossum9a744a91999-04-08 20:27:54 +000011 else:
12 return cmp(self.data, dict)
Guido van Rossumff712aa1997-08-15 00:45:26 +000013 def __len__(self): return len(self.data)
14 def __getitem__(self, key): return self.data[key]
15 def __setitem__(self, key, item): self.data[key] = item
16 def __delitem__(self, key): del self.data[key]
Guido van Rossum9a744a91999-04-08 20:27:54 +000017 def clear(self): self.data.clear()
Guido van Rossumff712aa1997-08-15 00:45:26 +000018 def copy(self):
Guido van Rossum9a744a91999-04-08 20:27:54 +000019 if self.__class__ is UserDict:
20 return UserDict(self.data)
Guido van Rossum548703a1998-03-26 22:14:20 +000021 import copy
22 return copy.copy(self)
Guido van Rossumff712aa1997-08-15 00:45:26 +000023 def keys(self): return self.data.keys()
24 def items(self): return self.data.items()
25 def values(self): return self.data.values()
26 def has_key(self, key): return self.data.has_key(key)
Guido van Rossum9a744a91999-04-08 20:27:54 +000027 def update(self, dict):
28 if isinstance(dict, UserDict):
29 self.data.update(dict.data)
30 elif isinstance(dict, type(self.data)):
31 self.data.update(dict)
Guido van Rossum548703a1998-03-26 22:14:20 +000032 else:
Guido van Rossum9a744a91999-04-08 20:27:54 +000033 for k, v in dict.items():
Guido van Rossum548703a1998-03-26 22:14:20 +000034 self.data[k] = v
Guido van Rossum0b233481997-11-26 15:44:34 +000035 def get(self, key, failobj=None):
Guido van Rossum9a744a91999-04-08 20:27:54 +000036 return self.data.get(key, failobj)
Guido van Rossum8d691c82000-09-01 19:25:51 +000037 def setdefault(self, key, failobj=None):
38 if not self.data.has_key(key):
39 self.data[key] = failobj
40 return self.data[key]