Merge pull request #17 from scraperdragon/upsert_all_key

Permit upsert with all columns as keys
This commit is contained in:
Friedrich Lindenberg 2013-06-27 09:39:13 -07:00
commit f520193a9e
2 changed files with 10 additions and 1 deletions

View File

@ -118,7 +118,7 @@ class Table(object):
``types``, matching the behavior of :py:meth:`insert() <dataset.Table.insert>`.
"""
self._check_dropped()
if not len(keys):
if not len(keys) or len(keys)==len(row):
return False
clause = [(u, row.get(u)) for u in keys]
"""

View File

@ -66,6 +66,15 @@ class TableTestCase(unittest.TestCase):
)
assert len(self.tbl) == len(TEST_DATA)+1, len(self.tbl)
def test_upsert_all_key(self):
for i in range(0,2):
self.tbl.upsert({
'date': datetime(2011, 01, 02),
'temperature': -10,
'place': 'Berlin'},
['date', 'temperature', 'place']
)
def test_delete(self):
self.tbl.insert({
'date': datetime(2011, 01, 02),