From 6574f4d4fd51b0b5a233f2bea9a6f7bb5dfd6a00 Mon Sep 17 00:00:00 2001 From: Josh Holtrop Date: Sun, 2 Jan 2011 14:04:50 -0500 Subject: [PATCH] added Task class and DataStore.getTaskByID() --- DataStore.py | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) diff --git a/DataStore.py b/DataStore.py index 75b698d..0acb72e 100644 --- a/DataStore.py +++ b/DataStore.py @@ -5,6 +5,12 @@ from datetime import datetime HISTORY_DT_FMT = '%Y-%m-%d %H:%M:%S' +class Task: + def __init__(self, name, longname, parentid): + self.name = name + self.longname = longname + self.parentid = parentid + class TaskRef: def __init__(self, taskid, dt): self.taskid = taskid @@ -77,3 +83,16 @@ VALUES (0, ?, ?) ''', (ct.taskid, ct.dt.strftime(HISTORY_DT_FMT))) c.close() self.conn.commit() + + def getTaskByID(self, taskid): + t = None + c = self.conn.cursor() + c.execute(''' +SELECT name, longname, parentid +FROM tasks +WHERE id = ? +''', (taskid,)) + for row in c: + t = Task(row[0], row[1], row[2]) + c.close() + return t