fix CI that fails due to pypi + override BaseDict.get as it was missing

This commit is contained in:
Bastien Gérard
2018-08-30 14:33:57 +02:00
parent d72daf5f39
commit 7ddbea697e
2 changed files with 39 additions and 16 deletions

View File

@@ -43,7 +43,14 @@ class BaseDict(dict):
self._name = name
super(BaseDict, self).__init__(dict_items)
def __getitem__(self, key, *args, **kwargs):
def get(self, key, default=None):
# get does not use __getitem__ by default so we must override it as well
try:
return self.__getitem__(key)
except KeyError:
return default
def __getitem__(self, key):
value = super(BaseDict, self).__getitem__(key)
EmbeddedDocument = _import_class('EmbeddedDocument')