finish changing modes for selections

diff --git a/sface/screens/mainscreen.py b/sface/screens/mainscreen.py
index 3ac5520..3d9aaff 100644 (file)
--- a/sface/screens/mainscreen.py
+++ b/sface/screens/mainscreen.py
@@ -9,6 +9,8 @@ from sface.config import config
 from sface.sfiprocess import SfiProcess
 from sface.screens.sfascreen import SfaScreen
 
+already_in_nodes = []
+
 class NodeView(QTreeView):
     def __init__(self, parent):
         QTreeView.__init__(self, parent)
@@ -21,18 +23,37 @@ class NodeView(QTreeView):
         self.setAttribute(Qt.WA_MacShowFocusRect, 0)
         self.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
 
-class SelectDelegate(QStyledItemDelegate):
-    pass
+    def mouseDoubleClickEvent(self, event):
+        index = self.currentIndex()
+        model = index.model()
+        select_index = model.index(index.row(), 2, index.parent())
+        select_data = select_index.data().toString()
+        hostname_index = model.index(index.row(), 1, index.parent())
+        hostname_data = hostname_index.data().toString()
+
+        if select_data == "true":
+            model.setData(select_index, QString("remove"))
+        elif select_data == "false":
+            model.setData(select_index, QString("add"))
+        elif select_data in ("add", "remove"):
+            if hostname_data in already_in_nodes: model.setData(select_index, QString("true"))
+            else: model.setData(select_index, QString("false"))
+
+        model.emit(SIGNAL("dataChanged(QModelIndex, QModelIndex)"), hostname_index, hostname_index)
+                
+        
 
 class NodeNameDelegate(QStyledItemDelegate):
     def __init__(self, parent):
-        QStyledItemDelegate.__init__(self)
+        QStyledItemDelegate.__init__(self, parent)
 
     def paint(self, painter, option, index):
         model = index.model()
         data = "%s" % index.data().toString()
         select_index = model.index(index.row(), 2, index.parent())
         select_data = select_index.data().toString()
+#        if select_data == "false":
+#            model.setData(select_index, QString("add"))
 
         if select_data not in ("true", "add", "remove"): # default view
             QStyledItemDelegate.paint(self, painter, option, index)
@@ -219,7 +240,7 @@ class NodeModel(QAbstractItemModel):
     def flags(self, index):
         if not index.isValid():
             return 0
-        return Qt.ItemIsEnabled | Qt.ItemIsSelectable | Qt.ItemIsEditable
+        return Qt.ItemIsEnabled | Qt.ItemIsSelectable# | Qt.ItemIsEditable
 
     def setData(self, index, value, role):
         if role != Qt.EditRole:
@@ -257,7 +278,6 @@ class SliceWidget(QWidget):
         self.filterModel = QSortFilterProxyModel(self) # enable filtering
 
         self.nodeNameDelegate = NodeNameDelegate(self)
-        self.selectDelegate = SelectDelegate(self)
 
         refresh = QPushButton("Update Slice Data", self)
         refresh.setSizePolicy(QSizePolicy.Maximum, QSizePolicy.Maximum)
@@ -318,8 +338,11 @@ class SliceWidget(QWidget):
         self.updateView()
 
     def updateView(self):
+        global already_in_nodes
+        already_in_nodes = []
         self.network_names = []
         self.nodeModel.clear()
+        
         rspec_string = self.readSliceRSpec()
         if not rspec_string:
             return None
@@ -333,6 +356,8 @@ class SliceWidget(QWidget):
             sliver_nodes = rspec_get_sliver_nodes_from_network(rspec_string, network)
             available_nodes = filter(lambda x:x not in sliver_nodes, all_nodes)
 
+            already_in_nodes += sliver_nodes
+
             for node in sliver_nodes:
                 nodeItem = TreeItem([QString(""), QString("%s" % node), QString("true")], networkItem)
                 networkItem.appendChild(nodeItem)
@@ -348,7 +373,6 @@ class SliceWidget(QWidget):
         self.filterModel.setDynamicSortFilter(True)
 
         self.nodeView.setItemDelegateForColumn(1, self.nodeNameDelegate)
-        self.nodeView.setItemDelegateForColumn(2, self.selectDelegate)
         self.nodeView.setModel(self.filterModel)
         self.nodeView.expandAll()
         self.nodeView.resizeColumnToContents(1)