spacepaste

Paste details

reply | raw

  1.  
  2. #!/usr/bin/env python
  5. #############################################################################
  6. ##
  7. ## Copyright (C) 2010 Riverbank Computing Limited.
  8. ## Copyright (C) 2010 Nokia Corporation and/or its subsidiary(-ies).
  9. ## All rights reserved.
  10. ##
  11. ## This file is part of the examples of PyQt.
  12. ##
  13. ## $QT_BEGIN_LICENSE:BSD$
  14. ## You may use this file under the terms of the BSD license as follows:
  15. ##
  16. ## "Redistribution and use in source and binary forms, with or without
  17. ## modification, are permitted provided that the following conditions are
  18. ## met:
  19. ## * Redistributions of source code must retain the above copyright
  20. ## notice, this list of conditions and the following disclaimer.
  21. ## * Redistributions in binary form must reproduce the above copyright
  22. ## notice, this list of conditions and the following disclaimer in
  23. ## the documentation and/or other materials provided with the
  24. ## distribution.
  25. ## * Neither the name of Nokia Corporation and its Subsidiary(-ies) nor
  26. ## the names of its contributors may be used to endorse or promote
  27. ## products derived from this software without specific prior written
  28. ## permission.
  29. ##
  30. ## THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  31. ## "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  32. ## LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  33. ## A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  34. ## OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  35. ## SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  36. ## LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  37. ## DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  38. ## THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  39. ## (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  40. ## OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE."
  41. ## $QT_END_LICENSE$
  42. ##
  43. #############################################################################
  46. from PyQt4 import QtCore, QtGui
  48. class TreeItem(object):
  49. def __init__(self, data, parent=None):
  50. self.parentItem = parent
  51. self.checkable = data[0][0] == "x"
  52. self.itemData = data
  53. self.childItems = []
  54. self.checked = None
  55. if self.checkable:
  56. self.checked = QtCore.Qt.Checked
  58. def appendChild(self, item):
  59. self.childItems.append(item)
  61. def child(self, row):
  62. return self.childItems[row]
  64. def childCount(self):
  65. return len(self.childItems)
  67. def columnCount(self):
  68. return len(self.itemData)
  70. def data(self, column):
  71. try:
  72. return self.itemData[column]
  73. except IndexError:
  74. return None
  76. def parent(self):
  77. return self.parentItem
  79. def row(self):
  80. if self.parentItem:
  81. return self.parentItem.childItems.index(self)
  83. return 0
  86. class TreeModel(QtCore.QAbstractItemModel):
  87. def __init__(self, data, parent=None):
  88. super(TreeModel, self).__init__(parent)
  89. self.states = {}
  90. self.rootItem = TreeItem(("Title", "Summary"))
  91. self.setupModelData(data.split('\n'), self.rootItem)
  93. def columnCount(self, parent):
  94. if parent.isValid():
  95. return parent.internalPointer().columnCount()
  96. else:
  97. return self.rootItem.columnCount()
  99. def data(self, index, role):
  100. if not index.isValid():
  101. return None
  103. item = index.internalPointer()
  105. if role == QtCore.Qt.DisplayRole:
  106. return item.data(index.column())
  108. elif role == QtCore.Qt.CheckStateRole:
  109. if index.column() == 0:
  110. return item.checked
  112. return None
  115. def setData(self, index, value, role):
  116. status = False
  117. item = index.internalPointer()
  119. if role == QtCore.Qt.CheckStateRole:
  120. if index.column() == 0:
  121. if item.checked == QtCore.Qt.Checked:
  122. item.checked = QtCore.Qt.Unchecked
  124. else:
  125. item.checked = QtCore.Qt.Checked
  127. status = True
  128. self.emit(QtCore.SIGNAL('dataChanged(QModelIndex,QModelIndex)'), index, index)
  130. return status
  132. def flags(self, index):
  133. if not index.isValid():
  134. return QtCore.Qt.NoItemFlags
  136. flags = QtCore.Qt.ItemIsEnabled | QtCore.Qt.ItemIsSelectable
  138. item = index.internalPointer()
  140. if item.checkable and index.column() == 0:
  141. flags = flags | QtCore.Qt.ItemIsUserCheckable
  143. return flags
  145. def headerData(self, section, orientation, role):
  146. if orientation == QtCore.Qt.Horizontal and role == QtCore.Qt.DisplayRole:
  147. return self.rootItem.data(section)
  149. return None
  151. def index(self, row, column, parent):
  152. if not self.hasIndex(row, column, parent):
  153. return QtCore.QModelIndex()
  155. if not parent.isValid():
  156. parentItem = self.rootItem
  157. else:
  158. parentItem = parent.internalPointer()
  160. childItem = parentItem.child(row)
  161. if childItem:
  162. return self.createIndex(row, column, childItem)
  163. else:
  164. return QtCore.QModelIndex()
  166. def parent(self, index):
  167. if not index.isValid():
  168. return QtCore.QModelIndex()
  170. childItem = index.internalPointer()
  171. parentItem = childItem.parent()
  173. if parentItem == self.rootItem:
  174. return QtCore.QModelIndex()
  176. return self.createIndex(parentItem.row(), 0, parentItem)
  178. def rowCount(self, parent):
  179. if parent.column() > 0:
  180. return 0
  182. if not parent.isValid():
  183. parentItem = self.rootItem
  184. else:
  185. parentItem = parent.internalPointer()
  187. return parentItem.childCount()
  189. def setupModelData(self, lines, parent):
  190. parents = [parent]
  191. indentations = [0]
  193. number = 0
  195. while number < len(lines):
  196. position = 0
  197. while position < len(lines[number]):
  198. if lines[number][position] != ' ':
  199. break
  200. position += 1
  202. lineData = lines[number][position:].trimmed()
  204. if lineData:
  205. # Read the column data from the rest of the line.
  206. columnData = [s for s in lineData.split('\t') if s]
  208. if position > indentations[-1]:
  209. # The last child of the current parent is now the new
  210. # parent unless the current parent has no children.
  212. if parents[-1].childCount() > 0:
  213. parents.append(parents[-1].child(parents[-1].childCount() - 1))
  214. indentations.append(position)
  216. else:
  217. while position < indentations[-1] and len(parents) > 0:
  218. parents.pop()
  219. indentations.pop()
  221. # Append a new item to the current parent's list of children.
  222. parents[-1].appendChild(TreeItem(columnData, parents[-1]))
  224. number += 1
  227. if __name__ == '__main__':
  229. import sys
  231. app = QtGui.QApplication(sys.argv)
  233. f = QtCore.QFile('default.txt')
  234. f.open(QtCore.QIODevice.ReadOnly)
  235. model = TreeModel(f.readAll())
  236. f.close()
  238. view = QtGui.QTreeView()
  239. view.setModel(model)
  240. view.setWindowTitle("Simple Tree Model")
  241. view.show()
  242. sys.exit(app.exec_())
  243.  
Powered by spacepaste. Paste removal.