From bf6d88f9dd5b19d12b2c59c5db1cdfea48e54716 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Bar=C4=B1=C5=9F=20Metin?= Date: Mon, 13 Sep 2010 10:38:36 +0200 Subject: [PATCH] make model editable --- sface/screens/mainscreen.py | 175 ++++++++++++++++++++++++------------ 1 file changed, 120 insertions(+), 55 deletions(-) diff --git a/sface/screens/mainscreen.py b/sface/screens/mainscreen.py index e68ca21..ac0dedb 100644 --- a/sface/screens/mainscreen.py +++ b/sface/screens/mainscreen.py @@ -33,11 +33,10 @@ class NodeNameDelegate(QStyledItemDelegate): # model = index.model() # select_index = model.index(index.row(), 2, index.parent()) # select_data = select_index.data().toString() -# if select_data == "true": +# if select_data == "false": # print select_data -# model.setData(select_index, QString("TEST"), Qt.DisplayRole) -# model.emit(SIGNAL('dataChanged(QModelIndex, QModelIndex)'), -# select_index, select_index) +# model.setData(index, QString("*%s" % data), Qt.EditRole) +# model.setData(select_index, QString("true"), Qt.EditRole) if data.startswith("*"): # already in the sliver @@ -85,12 +84,47 @@ class TreeItem: def childCount(self): return len(self.childItems) + def childNumber(self): + if self.parentItem: + return self.parentItem.childItems.index(self) + return 0 + def columnCount(self): return len(self.itemData) def data(self, column): return self.itemData[column] + def insertChildren(self, position, count, columns): + if position < 0 or position > len(self.childItems): + return False + + for row in range(count): + data = self.data(columns) + item = TreeItem(data, self) + self.childItems.insert(position, item) + + return True + + def insertColumns(self, position, columns): + if position < 0 or position > len(self.itemData): + return False + + for column in range(columns): + self.itemData.insert(position, QVariant()) + + for child in self.childItems: + child.insertColumns(position, columns) + + return True + + def setData(self, column, value): + if column < 0 or column >= len(self.itemData): + return False + + self.itemData[column] = value + return True + def row(self): if (self.parentItem): try: @@ -116,41 +150,23 @@ class NodeModel(QAbstractItemModel): def __initRoot(self): self.rootItem = TreeItem([QString("Testbed"), QString("Hostname"), QString("Selected")]) - def columnCount(self, parent): - if parent.isValid(): - return parent.internalPointer().columnCount() - else: - return self.rootItem.columnCount() - - def data(self, index, role): - if not index.isValid(): - return QVariant() - if role != Qt.DisplayRole: - return QVariant() - - item = index.internalPointer() - return item.data(index.column()) + def getItem(self, index): + if index.isValid(): + item = index.internalPointer() + if item: return item + return self.rootItem def headerData(self, section, orientation, role): if orientation == Qt.Horizontal and role == Qt.DisplayRole: return self.rootItem.data(section) return QVariant() - def flags(self, index): - if not index.isValid(): - return 0 - return Qt.ItemIsEnabled | Qt.ItemIsSelectable # | Qt.ItemIsEditable - def index(self, row, column, parent): if not self.hasIndex(row, column, parent): return QModelIndex() - parentItem = None - if not parent.isValid(): - parentItem = self.rootItem - else: - parentItem = parent.internalPointer() + parentItem = self.getItem(parent) childItem = parentItem.child(row) if childItem: @@ -158,32 +174,63 @@ class NodeModel(QAbstractItemModel): else: return QModelIndex() + def insertColumns(self, position, columns, parent): + self.beginInsertColumns(parent, position, position + columns -1) + ret = self.rootItem.insertColumns(position, columns) + self.endInsertColumns() + return ret + + def insertRows(self, position, rows, parent): + parentItem = self.getItem(parent) + self.beginInsertRows(parent, position, position + rows -1) + ret = parentItem.insertChildren(position, rows, self.rootItem.columnCount()) + self.endInsertRows() + return ret + def parent(self, index): if not index.isValid(): return QModelIndex() - childItem = index.internalPointer() + childItem = self.getItem(index) parentItem = childItem.parent() - if not parentItem: - return QModelIndex() - if parentItem is self.rootItem: return QModelIndex() return self.createIndex(parentItem.row(), 0, parentItem) - def rowCount(self, parent): - if parent.column() > 0: + def rowCount(self, parent=QModelIndex()): + parentItem = self.getItem(parent) + return parentItem.childCount() + + def columnCount(self, parent=None): + return self.rootItem.columnCount() + + def data(self, index, role): + if not index.isValid(): + return QVariant() + + if role != Qt.DisplayRole and role != Qt.EditRole: + return QVariant() + + item = self.getItem(index) + return item.data(index.column()) + + def flags(self, index): + if not index.isValid(): return 0 + return Qt.ItemIsEnabled | Qt.ItemIsSelectable | Qt.ItemIsEditable - parentItem = None - if not parent.isValid(): - parentItem = self.rootItem - else: - parentItem = parent.internalPointer() + def setData(self, index, value, role): + if role != Qt.EditRole: + return False + + item = self.getItem(index) + ret = item.setData(index.column(), value) + if ret: + self.emit(SIGNAL("dataChanged(QModelIndex, QModelIndex)"), index, index) + return ret - return parentItem.childCount() class SliceWidget(QWidget): @@ -192,46 +239,63 @@ class SliceWidget(QWidget): slicename = QLabel ("Slice : %s"%(config.getSlice() or "None"),self) slicename.setScaledContents(False) + searchlabel = QLabel ("Search: ", self) + searchlabel.setScaledContents(False) + searchbox = QLineEdit(self) + searchbox.setAttribute(Qt.WA_MacShowFocusRect, 0) + + toplayout = QHBoxLayout() + toplayout.addWidget(slicename, 0, Qt.AlignLeft) + toplayout.addStretch() + toplayout.addWidget(searchlabel, 0, Qt.AlignRight) + toplayout.addWidget(searchbox, 0, Qt.AlignRight) self.nodeView = NodeView(self) self.nodeModel = NodeModel(self) + self.filterModel = QSortFilterProxyModel(self) # enable filtering + self.filterModel.setSourceModel(self.nodeModel) + self.nodeView.setModel(self.filterModel) + self.filterModel.setDynamicSortFilter(True) self.nodeView.setModel(self.nodeModel) + self.nodeNameDelegate = NodeNameDelegate(self) self.selectDelegate = SelectDelegate(self) - refresh = QPushButton("Update Slice Data", self) refresh.setSizePolicy(QSizePolicy.Maximum, QSizePolicy.Maximum) submit = QPushButton("Submit", self) submit.setSizePolicy(QSizePolicy.Maximum, QSizePolicy.Maximum) - hlayout = QHBoxLayout() - hlayout.addWidget(refresh, 0, Qt.AlignLeft) - hlayout.addStretch() - hlayout.addWidget(submit, 0, Qt.AlignRight) + bottomlayout = QHBoxLayout() + bottomlayout.addWidget(refresh, 0, Qt.AlignLeft) + bottomlayout.addStretch() + bottomlayout.addWidget(submit, 0, Qt.AlignRight) layout = QVBoxLayout() - layout.addWidget(slicename) + layout.addLayout(toplayout) layout.addWidget(self.nodeView) - layout.addLayout(hlayout) - + layout.addLayout(bottomlayout) self.setLayout(layout) self.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding) self.connect(refresh, SIGNAL('clicked()'), self.refresh) self.connect(submit, SIGNAL('clicked()'), self.submit) + self.connect(searchbox, SIGNAL('textChanged(QString)'), self.filter) - rspec_file = os.path.expanduser("~/.sfi/%s.rspec" % config.getSlice()) - if os.path.exists(rspec_file): - self.updateView() + self.updateView() + + def filter(self, filter): + self.filterModel.setFilterRegExp(QRegExp(filter)) def submit(self): self.parent().setStatus("TODO: Submit not implemented yet!", 3000) def readSliceRSpec(self): rspec_file = config.getSliceRSpecFile() - xml = open(rspec_file).read() - return xml + if os.path.exists(rspec_file): + xml = open(rspec_file).read() + return xml + return None def refresh(self): if not config.getSlice(): @@ -252,8 +316,10 @@ class SliceWidget(QWidget): def updateView(self): self.nodeModel.clear() rspec_string = self.readSliceRSpec() - networks = rspec_get_networks(rspec_string) + if not rspec_string: + return None + networks = rspec_get_networks(rspec_string) for network in networks: networkItem = TreeItem([QString(network), QString(""), QString("")], self.nodeModel.rootItem) @@ -273,7 +339,6 @@ class SliceWidget(QWidget): self.nodeView.expandAll() self.nodeView.resizeColumnToContents(1) - self.nodeView.setItemDelegateForColumn(1, self.nodeNameDelegate) self.nodeView.setItemDelegateForColumn(2, self.selectDelegate) -- 2.43.0