oci/zeronet
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Support insertOrUpdate in database cursor

Browse source
This commit is contained in:
shortcutme 2016-11-07 23:08:57 +01:00
parent d595373f9f
commit 8955998d25
1 changed files with 15 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
src/Db/DbCursor.py
View file

@ -64,6 +64,21 @@ class DbCursor:
# if query == "BEGIN": self.logging = False # Turn logging off on transaction commit # if query == "BEGIN": self.logging = False # Turn logging off on transaction commit
return res return res
# Creates on updates a database row without incrementing the rowid
def insertOrUpdate(self, table, query_sets, query_wheres, oninsert={}):
sql_sets = ["%s = :%s" % (key, key) for key in query_sets.keys()]
sql_wheres = ["%s = :%s" % (key, key) for key in query_wheres.keys()]
params = query_sets
params.update(query_wheres)
self.cursor.execute(
"UPDATE %s SET %s WHERE %s" % (table, ", ".join(sql_sets), " AND ".join(sql_wheres)),
params
)
if self.cursor.rowcount == 0:
params.update(oninsert) # Add insert-only fields
self.execute("INSERT INTO %s ?" % table, params)
# Create new table # Create new table
# Return: True on success # Return: True on success
def createTable(self, table, cols): def createTable(self, table, cols):