From bcac83a5a9c673afbc8c5cbba7c98c72bd940415 Mon Sep 17 00:00:00 2001
From: Josh Holtrop <jholtrop+git@gmail.com>
Date: Sun, 2 Jan 2011 15:01:48 -0500
Subject: [PATCH] added DataStore.addTime()

---
 DataStore.py | 27 +++++++++++++++++++++++++++
 1 file changed, 27 insertions(+)

diff --git a/DataStore.py b/DataStore.py
index 521abff..3c95a19 100644
--- a/DataStore.py
+++ b/DataStore.py
@@ -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()