Merge pull request #17 from scraperdragon/upsert_all_key

Browse Source 
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
dataset/persistence/table.py
View File

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

9
test/test_persistence.py
View File

@ -66,6 +66,15 @@ class TableTestCase(unittest.TestCase):
) )
assert len(self.tbl) == len(TEST_DATA)+1, len(self.tbl) 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): def test_delete(self):
self.tbl.insert({ self.tbl.insert({
'date': datetime(2011, 01, 02), 'date': datetime(2011, 01, 02),