X-Git-Url: http://git.onelab.eu/?p=sface.git;a=blobdiff_plain;f=sface%2Fmainwindow.py;h=b361b22a0078bcc9b49b42116c96846aa49b5b2b;hp=cefec17f11d9b4e277bc022b2284baed840101b8;hb=194e800bc6c16d787fff48410a4ef71928efe8fb;hpb=ca5d17301939ddd993f658a3acae09d1aa56e5c9

diff --git a/sface/mainwindow.py b/sface/mainwindow.py
index cefec17..b361b22 100644
--- a/sface/mainwindow.py
+++ b/sface/mainwindow.py
@@ -121,8 +121,10 @@ class MainWindow(QWidget):
     def __init__(self, parent=None):
         QWidget.__init__(self, parent)
 
-        self.logWindow = LogWindow(self)
-        self.rspecWindow = RSpecWindow(self)
+        # These are top-level windows and should be initialized with parent set
+        # to our parent. Otherwise, getting a segfault on exit in Ubuntu.
+        self.logWindow = LogWindow(parent)
+        self.rspecWindow = RSpecWindow(parent)
 
         self.pix = QLabel(self)
 
@@ -131,15 +133,23 @@ class MainWindow(QWidget):
 
         self.screens = QStackedWidget(self)
         for screen in screenClasses:
-            screenWidget = screen(self)
-            self.screenWidgets.append(screenWidget)
-            self.screens.addWidget(screenWidget)
+            # use a try/catch block to isolate the screen. Third-party plugins
+            # could be buggy.
+            try:
+                screenWidget = screen(self)
+                self.screenWidgets.append(screenWidget)
+                self.screens.addWidget(screenWidget)
+            except:
+                print "Exception while creating screen", screen.__name__
+                traceback.print_exc()
 
         self.screens.addWidget(self.pix)
         self.next_screen = None
 
         self.nav = Nav(self.screenWidgets, self)
-        self.nav.setTitle(self.screenWidgets[0].getTitleText())
+
+        if self.screenWidgets:
+            self.nav.setTitle(self.screenWidgets[0].getTitleText())
 
         self.status = Status(self)
         self.log = QLabel("<a href='showlog'>Show Log</a>", self)
@@ -222,3 +232,17 @@ class MainWindow(QWidget):
     def nodeSelectionChanged(self, hostname):
         if self.rspecWindow.isVisible():
             self.rspecWindow.showNode(hostname)
+
+    def closeEvent(self, event):
+        # give the screens an opportunity to veto the close
+        for screen in self.screenWidgets:
+            if not screen.canClose():
+                event.ignore()
+                return
+
+        # give the screens an opportunity to close gracefully
+        for screen in self.screenWidgets:
+            screen.mainWindowClose()
+
+        QWidget.closeEvent(self, event)
+