moparisthebest/SickRage
1
0
Fork 0
mirror of https://github.com/moparisthebest/SickRage synced 2024-11-10 19:35:08 -05:00
Code Issues Releases Wiki Activity

Fix for sql rowcount error

Browse Source
This commit is contained in:
echel0n 2014-06-05 02:19:48 -07:00
parent 8fbd9155d3
commit f052848e52
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
sickbeard/db.py
View File

@ -204,6 +204,8 @@ class DBConnection:
def upsert(self, tableName, valueDict, keyDict):
changesBefore = self.connection.total_changes
genParams = lambda myDict: [x + " = ?" for x in myDict.keys()]
query = "UPDATE " + tableName + " SET " + ", ".join(genParams(valueDict)) + " WHERE " + " AND ".join(
@ -211,7 +213,7 @@ class DBConnection:
result = self.action(query, valueDict.values() + keyDict.values())
if result.rowcount > 0:
if self.connection.total_changes == changesBefore:
query = "INSERT INTO " + tableName + " (" + ", ".join(valueDict.keys() + keyDict.keys()) + ")" + \
" VALUES (" + ", ".join(["?"] * len(valueDict.keys() + keyDict.keys())) + ")"
self.action(query, valueDict.values() + keyDict.values())