Added error check to _upsert_pre_check(), res = None when error

Browse Source
This commit is contained in:
Saun Shewanown 2016-04-27 16:39:50 -04:00
parent 7fd1a71506
commit e7f45b2a68
1 changed files with 14 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
dataset/persistence/table.py
View File

@ -183,6 +183,7 @@ class Table(object):
def _upsert_pre_check(self, row, keys, ensure):
# check whether keys arg is a string and format as a list
try:
if not isinstance(keys, (list, tuple)):
keys = [keys]
self._check_dropped()
@ -196,6 +197,8 @@ class Table(object):
filters[key] = row.get(key)
res = self.find_one(**filters)
except:
res = None
return res