allow keys arg for upsert to be a string

Browse Source 
I want to be able to use `upsert` without formatting the keys arg as a list, ie:
```
table.upsert(new_data, "id")
```
This commit is contained in:
Brian Abelson 2013-08-10 21:34:00 -04:00
parent a228f48e23
commit 904e464dfa
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
dataset/persistence/table.py
View File

@ -147,6 +147,10 @@ class Table(object):
data = dict(id=10, title='I am a banana!') data = dict(id=10, title='I am a banana!')
table.upsert(data, ['id']) table.upsert(data, ['id'])
""" """
# check whether keys arg is a string and format as a list
if isinstance(keys, basestring):
keys = [keys]
self._check_dropped() self._check_dropped()
if ensure: if ensure:
self.create_index(keys) self.create_index(keys)