Audit qt dialogs connections

pyface/tests/test_progress_dialog.py

@@ -56,6 +56,14 @@ def test_can_cancel(self):
         with self.event_loop():
             self.dialog.destroy()
 
+    def test_can_ok(self):
+        # test that creation works with can_ok
+        self.dialog.can_ok = True
+        with self.event_loop():
+            self.dialog._create()
+        with self.event_loop():
+            self.dialog.destroy()
+
     def test_show_time(self):
         # test that creation works with show_time
         self.dialog.show_time = True

pyface/ui/qt4/about_dialog.py

@@ -71,6 +71,17 @@ class AboutDialog(MAboutDialog, Dialog):
 
     image = Instance(ImageResource, ImageResource("about"))
 
+    # -------------------------------------------------------------------------
+    # 'IWidget' interface.
+    # -------------------------------------------------------------------------
+
+    def destroy(self):
+        if self.control is not None:
+            buttons = self.control.findChild(QtGui.QDialogButtonBox)
+            buttons.accepted.disconnect(self.control.accept)
+
+        super(AboutDialog, self).destroy()
+
     # ------------------------------------------------------------------------
     # Protected 'IDialog' interface.
     # ------------------------------------------------------------------------

pyface/ui/qt4/dialog.py

@@ -15,8 +15,9 @@
 from pyface.qt import QtCore, QtGui
 
 
-from traits.api import Bool, Enum, Int, provides, Str
-
+from traits.api import (
+    Any, Bool, Callable, Enum, Int, List, provides, Str, Tuple
+)
 
 from pyface.i_dialog import IDialog, MDialog
 from pyface.constant import OK, CANCEL, YES, NO
@@ -60,6 +61,13 @@ class Dialog(MDialog, Window):
 
     title = Str("Dialog")
 
+    # Private interface ---------------------------------------------------#
+
+    #: A list of connected Qt signals to be removed before destruction.
+    #: First item in the tuple is the Qt signal. The second item is the event
+    #: handler.
+    _connections_to_remove = List(Tuple(Any, Callable))
+
     # ------------------------------------------------------------------------
     # Protected 'IDialog' interface.
     # ------------------------------------------------------------------------
@@ -77,6 +85,7 @@ def _create_buttons(self, parent):
 
         btn.setDefault(True)
         btn.clicked.connect(self.control.accept)
+        self._connections_to_remove.append((btn.clicked, self.control.accept))
 
         # 'Cancel' button.
         if self.cancel_label:
@@ -87,6 +96,7 @@ def _create_buttons(self, parent):
             btn = buttons.addButton(QtGui.QDialogButtonBox.Cancel)
 
         btn.clicked.connect(self.control.reject)
+        self._connections_to_remove.append((btn.clicked, self.control.reject))
 
         # 'Help' button.
         # FIXME v3: In the original code the only possible hook into the help
@@ -138,6 +148,18 @@ def _show_modal(self):
         retval = dialog.exec_()
         return _RESULT_MAP[retval]
 
+    # -------------------------------------------------------------------------
+    # 'IWidget' interface.
+    # -------------------------------------------------------------------------
+
+    def destroy(self):
+        if self.control is not None:
+            while self._connections_to_remove:
+                signal, handler = self._connections_to_remove.pop()
+                signal.disconnect(handler)
+
+        super(Dialog, self).destroy()
+
     # ------------------------------------------------------------------------
     # Protected 'IWidget' interface.
     # ------------------------------------------------------------------------
@@ -149,6 +171,9 @@ def _create_control(self, parent):
         # MDialog's open method. For 'nonmodal', we do it here.
         if self.style == "nonmodal":
             dlg.finished.connect(self._finished_fired)
+            self._connections_to_remove.append(
+                (dlg.finished, self._finished_fired)
+            )
 
         if self.size != (-1, -1):
             dlg.resize(*self.size)

pyface/ui/qt4/progress_dialog.py

@@ -97,6 +97,20 @@ def close(self):
 
         super(ProgressDialog, self).close()
 
+    # -------------------------------------------------------------------------
+    # 'IWidget' interface.
+    # -------------------------------------------------------------------------
+
+    def destroy(self):
+        if self.control is not None:
+            buttons = self.control.findChild(QtGui.QDialogButtonBox)
+            if self.can_cancel:
+                buttons.rejected.disconnect(self.control.reject)
+            if self.can_ok:
+                buttons.accepted.disconnect(self.control.accept)
+
+        super(ProgressDialog, self).destroy()
+
     # -------------------------------------------------------------------------
     # IProgressDialog Interface
     # -------------------------------------------------------------------------