Merge remote-tracking branch 'origin/pr/269'

* origin/pr/269: Added more reasonable behavior to "should I force shutdown or wait" message
QubesOS · Oct 29, 2020 · 1e3da9e · 1e3da9e
2 parents 685dfd7 + 1cfd5b8
commit 1e3da9e
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/qubesmanager/qube_manager.py b/qubesmanager/qube_manager.py
@@ -533,7 +533,14 @@ def check_if_vm_has_shutdown(self):
                     self.tr("Wait another {0} seconds...").format(
                         self.shutdown_time / 1000),
                     QMessageBox.NoRole)
+                ignore_button = msgbox.addButton(self.tr("Don't ask again"),
+                                                 QMessageBox.RejectRole)
                 msgbox.setDefaultButton(wait_button)
+                msgbox.setEscapeButton(ignore_button)
+                msgbox.setWindowFlags(
+                    msgbox.windowFlags() | Qt.CustomizeWindowHint)
+                msgbox.setWindowFlags(
+                    msgbox.windowFlags() & ~Qt.WindowCloseButtonHint)
                 msgbox.exec_()
                 msgbox.deleteLater()
 
@@ -545,6 +552,8 @@ def check_if_vm_has_shutdown(self):
                         # shutting it down
                         pass
                     self.restart_vm_if_needed()
+                elif msgbox.clickedButton() is ignore_button:
+                    return
                 else:
                     self.shutdown_started = datetime.now()
                     self.check_again_later()