Skip to content

Commit f05203c

Browse files
committed
implement default argument return in BasicDatabase.get()
1 parent 40520b7 commit f05203c

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

roundup/test/memorydb.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -163,9 +163,13 @@ def __contains__(self, key):
163163
return super(BasicDatabase, self).__contains__(s2b(key))
164164
def exists(self, infoid):
165165
return infoid in self
166-
def get(self, infoid, value, default=None):
166+
_marker = []
167+
def get(self, infoid, value, default=_marker):
167168
if infoid not in self:
168-
raise KeyError
169+
if default is self._marker:
170+
raise KeyError
171+
else:
172+
return default
169173
return self[infoid].get(value, default)
170174
def getall(self, infoid):
171175
if infoid not in self:

0 commit comments

Comments
 (0)