From 855068c43f8244fd8f7585686d2b5538797e623e Mon Sep 17 00:00:00 2001
From: Josh Holtrop <jholtrop+git@gmail.com>
Date: Sun, 2 Jan 2011 11:53:25 -0500
Subject: [PATCH] added DataStore.getCurrentTask() and TaskRef class

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

diff --git a/DataStore.py b/DataStore.py
index f3f05e2..84b8e16 100644
--- a/DataStore.py
+++ b/DataStore.py
@@ -1,6 +1,14 @@
 
 import sqlite3
 import os
+from datetime import datetime
+
+HISTORY_DT_FMT = '%Y-%m-%d %H:%M:%S'
+
+class TaskRef:
+    def __init__(self, taskid, dt):
+        self.taskid = taskid
+        self.dt = dt
 
 class DataStore:
     def __init__(self, dbfile):
@@ -41,3 +49,18 @@ CREATE TABLE history (
         c.close()
         conn.close()
 
+    def getCurrentTask(self):
+        c = self.conn.cursor()
+        c.execute('''
+SELECT taskid, datetime
+FROM history
+WHERE id = 0
+''')
+        ct = None
+        for row in c:
+            taskid = row[0]
+            dt = datetime.strptime(row[1], HISTORY_DT_FMT)
+            ct = TaskRef(taskid, dt)
+        c.close()
+        self.conn.commit()
+        return ct