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 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
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
class TodoItemModel(QAbstractItemModel):
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 flags(self, index):
'''
Returns the item flags for the given index.
All the items is editable and drag & drop enabled.
'''
if not index.isValid():
return (Qt.ItemIsEnabled | Qt.ItemIsDropEnabled)
if index.column() == WHAT:
return (QAbstractItemModel.flags(self, index)
|Qt.ItemIsEditable|Qt.ItemIsDragEnabled
|Qt.ItemIsDropEnabled|Qt.ItemIsUserCheckable)
return (QAbstractItemModel.flags(self, index)
| Qt.ItemIsEditable | Qt.ItemIsDragEnabled \
| Qt.ItemIsDropEnabled )
def load(self):
items = [TodoItem(*item) for item \
in self.SqlHlp.getAllTodoItems()]
items.sort()
for todoItem in items:
#print todoItem
self.addTodoItemToTree(todoItem)
def addTodoItemToTree(self, todoItem):
'''
Add a todo item into the datas in memor.
Build a tree from the root node.
'''
root = self.root
ancesters = todoItem.getAncestersList()
#print ancesters
if ancesters == [0]:
self.root.addChild(todoItem)
todoItem.parent = self.root
else:
for ancstID in ancesters:
ancst = root.getChildByID(ancstID)
if not ancst:
# The path of ancesters does not exist!
self.root.addChild(todoItem) # Add the item to root
todoItem.fatherID = self.root.ID
todoItem.ancesters = str(self.root.ID)
todoItem.parent = self.root
self.SqlHlp.updateTodoItem(todoItem.ID, \
t_father_id=todoItem.fatherID,\
t_ancesters=todoItem.ancesters)
else:
root = ancst
root.addChild(todoItem)
todoItem.parent = root
def nodeFromIndex(self, index):
return index.internalPointer() \
if index.isValid() else self.root
def index(self, row, column, parent):
assert self.root
father = self.nodeFromIndex(parent)
assert father is not None
return self.createIndex(row, column, \
father.childAtRow(row))
def parent(self, child):
''' If the child is the top level, i.e., its
parent is self.root, return None. '''
child = self.nodeFromIndex(child)
if child is None:
return QModelIndex()
parent = child.parent
if parent is None:
return QModelIndex()
grandparent = parent.parent
if grandparent is None:
return QModelIndex()
row = grandparent.rowOfChild(parent)
assert row != -1
return self.createIndex(row, 0, parent)
def data(self, index, role=Qt.DisplayRole):
item = self.nodeFromIndex(index)
if not item:
return QVariant()
column = index.column()
if role == Qt.DisplayRole:
if column == WHAT:
return QVariant(item.what)
if column == WHEN:
timeFomat = '%Y-%m-%d %H-%m'
sTime = time.strftime(timeFomat, time.gmtime(item.when))
return QVariant(sTime)
if role == Qt.CheckStateRole and column == WHAT:
return item.finish
return QVariant()
def headerData(self, section, orientation, \
role=Qt.DisplayRole):
if orientation == Qt.Horizontal and \
role == Qt.DisplayRole:
assert 0<= section < len(self.headers)
return QVariant(self.headers[section])
return QVariant()
def rowCount(self, index=QModelIndex()):
item = self.nodeFromIndex(index)
if item is None:
return 0
return item.lenOfChildren()
def columnCount(self, index=QModelIndex()):
''' The columns to show are: FINISH, WHAT, WHEN. '''
return len(self.headers)
def setData(self, index, value, role=Qt.EditRole):
''' setData '''
item = self.nodeFromIndex(index)
if not item:
return False
column = index.column()
print 'setData:', str(role), str(index.row()), \
str(index.column()), str(item.what), str(item.finish)
if role == Qt.CheckStateRole and column == WHAT:
item.finish = value.toInt()[0]
if role == Qt.EditRole:
if column == WHAT:
item.what = str(value.toString())
elif column == WHEN:
item.when = value.toInt()
#item.parent.updateOrderKeys()
item.parent.sortChildren()
self.emit(SIGNAL("dataChanged(QModelIndex,QModelIndex)"),
index, index)
datas = item.dataToUpdate()
self.SqlHlp.updateTodoItem(*datas)
#self.reset()
return True
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
def removeRows(self, startRow, count, parent = QModelIndex()):
''' Used to remove specified rows.'''
parentItem = self.nodeFromIndex(parent)
print parentItem, parentItem == None
print startRow, count
print self.root.children
if parentItem == None:
return False
self.beginRemoveRows(parent, startRow, startRow + count - 1)
for row in range(count):
IDs = parentItem.deleteChild(startRow + row)
self.SqlHlp.deleteTodoItems(IDs)
print IDs, 'are deleted.'
self.endRemoveRows()
return True
class TodoItemModel(QAbstractItemModel):
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 flags(self, index):
'''
Returns the item flags for the given index.
All the items is editable and drag & drop enabled.
'''
if not index.isValid():
return (Qt.ItemIsEnabled | Qt.ItemIsDropEnabled)
if index.column() == WHAT:
return (QAbstractItemModel.flags(self, index)
| Qt.ItemIsEditable | Qt.ItemIsDragEnabled
| Qt.ItemIsDropEnabled | Qt.ItemIsUserCheckable)
return (QAbstractItemModel.flags(self, index)
| Qt.ItemIsEditable | Qt.ItemIsDragEnabled \
| Qt.ItemIsDropEnabled )
def load(self):
items = [TodoItem(*item) for item \
in self.SqlHlp.getAllTodoItems()]
items.sort()
for todoItem in items:
#print todoItem
self.addTodoItemToTree(todoItem)
def addTodoItemToTree(self, todoItem):
'''
Add a todo item into the datas in memor.
Build a tree from the root node.
'''
root = self.root
ancesters = todoItem.getAncestersList()
#print ancesters
if ancesters == [0]:
self.root.addChild(todoItem)
todoItem.parent = self.root
else:
for ancstID in ancesters:
ancst = root.getChildByID(ancstID)
if not ancst:
# The path of ancesters does not exist!
self.root.addChild(todoItem) # Add the item to root
todoItem.fatherID = self.root.ID
todoItem.ancesters = str(self.root.ID)
todoItem.parent = self.root
self.SqlHlp.updateTodoItem(todoItem.ID, \
t_father_id=todoItem.fatherID,\
t_ancesters=todoItem.ancesters)
else:
root = ancst
root.addChild(todoItem)
todoItem.parent = root
def nodeFromIndex(self, index):
return index.internalPointer() \
if index.isValid() else self.root
def index(self, row, column, parent):
assert self.root
father = self.nodeFromIndex(parent)
assert father is not None
return self.createIndex(row, column, \
father.childAtRow(row))
def parent(self, child):
''' If the child is the top level, i.e., its
parent is self.root, return None. '''
child = self.nodeFromIndex(child)
if child is None:
return QModelIndex()
parent = child.parent
if parent is None:
return QModelIndex()
grandparent = parent.parent
if grandparent is None:
return QModelIndex()
row = grandparent.rowOfChild(parent)
assert row != -1
return self.createIndex(row, 0, parent)
def data(self, index, role=Qt.DisplayRole):
item = self.nodeFromIndex(index)
if not item:
return QVariant()
column = index.column()
if role == Qt.DisplayRole:
if column == WHAT:
return QVariant(item.what)
if column == WHEN:
timeFomat = '%Y-%m-%d %H-%m'
sTime = time.strftime(timeFomat, time.gmtime(item.when))
return QVariant(sTime)
if role == Qt.CheckStateRole and column == WHAT:
return item.finish
return QVariant()
def headerData(self, section, orientation, \
role=Qt.DisplayRole):
if orientation == Qt.Horizontal and \
role == Qt.DisplayRole:
assert 0 <= section < len(self.headers)
return QVariant(self.headers[section])
return QVariant()
def rowCount(self, index=QModelIndex()):
item = self.nodeFromIndex(index)
if item is None:
return 0
return item.lenOfChildren()
def columnCount(self, index=QModelIndex()):
''' The columns to show are: FINISH, WHAT, WHEN. '''
return len(self.headers)
def setData(self, index, value, role=Qt.EditRole):
''' setData '''
item = self.nodeFromIndex(index)
if not item:
return False
column = index.column()
print 'setData:', str(role), str(index.row()), \
str(index.column()), str(item.what), str(item.finish)
if role == Qt.CheckStateRole and column == WHAT:
item.finish = value.toInt()[0]
if role == Qt.EditRole:
if column == WHAT:
item.what = str(value.toString())
elif column == WHEN:
item.when = value.toInt()
#item.parent.updateOrderKeys()
item.parent.sortChildren()
self.emit(SIGNAL("dataChanged(QModelIndex,QModelIndex)"), index, index)
datas = item.dataToUpdate()
self.SqlHlp.updateTodoItem(*datas)
#self.reset()
return True
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
def removeRows(self, startRow, count, parent=QModelIndex()):
''' Used to remove specified rows.'''
parentItem = self.nodeFromIndex(parent)
print parentItem, parentItem == None
print startRow, count
print self.root.children
if parentItem == None:
return False
self.beginRemoveRows(parent, startRow, startRow + count - 1)
for row in range(count):
IDs = parentItem.deleteChild(startRow + row)
self.SqlHlp.deleteTodoItems(IDs)
print IDs, 'are deleted.'
self.endRemoveRows()
return True
