connect wizard cancel button with shutdown

1 files changed, 7 insertions, 1 deletions

diff --git a/src/leap/gui/firstrun/wizard.py b/src/leap/gui/firstrun/wizard.py
index 7a3d6973..bbb48149 100755
--- a/src/leap/gui/firstrun/wizard.py
+++ b/src/leap/gui/firstrun/wizard.py
@@ -61,7 +61,8 @@ class FirstRunWizard(QtGui.QWizard):
             eipconfigchecker=eipchecks.EIPConfigChecker,
             start_eipconnection_signal=None,
             eip_statuschange_signal=None,
-            debug_server=None):
+            debug_server=None,
+            quitcallback=None):
         super(FirstRunWizard, self).__init__(
             parent,
             QtCore.Qt.WindowStaysOnTopHint)
@@ -97,6 +98,11 @@ class FirstRunWizard(QtGui.QWizard):
         self.start_eipconnection_signal = start_eipconnection_signal
         self.eip_statuschange_signal = eip_statuschange_signal
 
+        if quitcallback is not None:
+            self.button(
+                QtGui.QWizard.CancelButton).clicked.connect(
+                    quitcallback)
+
         self.providerconfig = None
         # previously registered
         # if True, jumps to LogIn page.