added Task class and DataStore.getTaskByID()

This commit is contained in:
Josh Holtrop 2011-01-02 14:04:50 -05:00
parent af10a21be7
commit 6574f4d4fd

View File

@ -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