extract nodeview and nodemodel

diff --git a/sface/screens/mainscreen.py b/sface/screens/mainscreen.py
index 15f0ed0..117d0ab 100644 (file)
--- a/sface/screens/mainscreen.py
+++ b/sface/screens/mainscreen.py
@@ -9,21 +9,28 @@ from sface.sficonfig import config
 from sface.sfiprocess import SfiProcess
 from sface.screens.sfascreen import SfaScreen
 
+class NodeView(QTreeView):
+    def __init__(self, parent):
+        QTreeView.__init__(self, parent)
 
-class SliceWidget(QWidget):
+        self.setRootIsDecorated(False)
+        self.setAlternatingRowColors(True)
+        self.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
+
+class NodeModel(QStandardItemModel):
     def __init__(self, parent):
-        QWidget.__init__(self, parent)
+        QStandardItemModel.__init__(self, 0, 3, parent)
+        self.setHeaderData(0, Qt.Horizontal, QString("Testbed"))
+        self.setHeaderData(1, Qt.Horizontal, QString("Hostname"))
+        self.setHeaderData(2, Qt.Horizontal, QString("IsIn"))
 
-        self.nodeView = QTreeView(self)
-        self.nodeView.setRootIsDecorated(False)
-        self.nodeView.setAlternatingRowColors(True)
-        self.nodeView.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
 
-        self.nodeModel = QStandardItemModel(0, 3, self)
-        self.nodeModel.setHeaderData(0, Qt.Horizontal, QString("Testbed"))
-        self.nodeModel.setHeaderData(1, Qt.Horizontal, QString("Hostname"))
-        self.nodeModel.setHeaderData(2, Qt.Horizontal, QString("IsIn"))
+class SliceWidget(QWidget):
+    def __init__(self, parent):
+        QWidget.__init__(self, parent)
 
+        self.nodeView = NodeView(self)
+        self.nodeModel = NodeModel(self)
         self.nodeView.setModel(self.nodeModel)
 
         refresh = QLabel("<a href='refresh'>Refresh</a>", self)