introduce RSpecView, inherited from QTreeView

[sface.git] / sface / screens / rspecscreen.py

diff --git a/sface/screens/rspecscreen.py b/sface/screens/rspecscreen.py
index 320e065..e6a75d1 100644 (file)
--- a/sface/screens/rspecscreen.py
+++ b/sface/screens/rspecscreen.py
@@ -9,6 +9,18 @@ from sface.config import config
 from sface.screens.sfascreen import SfaScreen
 
 
+class RSpecView(QTreeView):
+    def __init__(self, parent):
+        QTreeView.__init__(self, parent)
+
+        self.setAnimated(True)
+        self.setItemsExpandable(True)
+        self.setRootIsDecorated(True)
+        self.setHeaderHidden(True)
+        self.setAttribute(Qt.WA_MacShowFocusRect, 0)
+        self.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
+
+
 class DomModel(QAbstractItemModel):
     def __init__(self, document, parent = 0):
         QAbstractItemModel.__init__(self, parent)
@@ -149,6 +161,7 @@ class DomItem:
         return self.rowNumber
     
 
+
 class RSpecWidget(QWidget):
     def __init__(self, parent=None):
         QWidget.__init__(self, parent)
@@ -163,9 +176,9 @@ class RSpecWidget(QWidget):
         document.setContent(open(rspec_file,'r').read())
         model = DomModel(document, self)
 
-        view = QTreeView(self)
-        view.setHeaderHidden(True)
+        view = RSpecView(self)
         view.setModel(model)
+        view.expandAll()
 
         layout = QVBoxLayout()
         layout.addWidget(view)