added DataStore.addTime()

This commit is contained in:
Josh Holtrop 2011-01-02 15:01:48 -05:00
parent 8ad5fc3c71
commit bcac83a5a9

View File

@ -127,3 +127,30 @@ VALUES (?, ?, ?, ?)
c.close()
self.conn.commit()
return nextid
def addTime(self, date, taskid, seconds):
c = self.conn.cursor()
exists = False
oldseconds = 0
c.execute('''
SELECT seconds
FROM entries
WHERE date = ? AND taskid = ?
''', (date, taskid))
for row in c:
if row[0] is not None:
exists = True
oldseconds = row[0]
if exists:
c.execute('''
UPDATE entries
SET seconds = ?
WHERE date = ? AND taskid = ?
''', (oldseconds + seconds, date, taskid))
else:
c.execute('''
INSERT INTO entries
VALUES(?, ?, ?)
''', (date, taskid, seconds))
c.close()
self.conn.commit()