diff --git a/DataStore.py b/DataStore.py index cd6a0ac..521abff 100644 --- a/DataStore.py +++ b/DataStore.py @@ -6,7 +6,8 @@ from datetime import datetime HISTORY_DT_FMT = '%Y-%m-%d %H:%M:%S' class Task: - def __init__(self, name, longname, parentid): + def __init__(self, taskid, name, longname, parentid): + self.taskid = taskid self.name = name self.longname = longname self.parentid = parentid @@ -93,7 +94,7 @@ FROM tasks WHERE id = ? ''', (taskid,)) for row in c: - t = Task(row[0], row[1], row[2]) + t = Task(taskid, row[0], row[1], row[2]) c.close() self.conn.commit() return t