def load(self):
items = [TodoItem(*item) for item \
in self.SqlHlp.getAllTodoItems()]
items.sort()
for todoItem in items:
#print todoItem
self.addTodoItemToTree(todoItem)
def __init__(self):
super(TodoItemModel, self).__init__()
self.root = TodoItem(0, 0, 'root', 0, '', '', 0, '', 0, '')
self.root.parent = None
self.headers = ['Task', 'Time']
self.context = set()
self.whereTodo = set()
self.SqlHlp = SqliteHelper()
self.SqlHlp.initDatabase()
def insertRows(self,
position=APPEND_NEW_ITEM,
count=1,
parent=QModelIndex()):
''' Used to insert new rows into the tree and the database.
Insert count rows in the parent's positionth row.'''
parentItem = self.nodeFromIndex(parent)
if parentItem == None:
return False
fatherID = parentItem.ID
ancesters = parentItem.ancesters + str(fatherID)
if position == APPEND_NEW_ITEM:
startRow = parentItem.lenOfChildren()
self.beginInsertRows(parent, startRow, startRow + count - 1)
for row in range(count):
ID = self.SqlHlp.newTodoItem(t_father_id = fatherID, \
t_ancesters = ancesters)
item = TodoItem(ID)
item.parent = parentItem
parentItem.appendNewChild(item)
index = self.createIndex(startRow + row, 0, parentItem)
self.endInsertRows()
return True
