From bd8ec22dc0154eeb1d8852d538a6c4239ced1995 Mon Sep 17 00:00:00 2001
From: luxick <git@luxick.com>
Date: Sun, 2 Jul 2017 15:57:47 +0200
Subject: [PATCH] Add cancel option when exiting with unsaved changes.

---
 cardvault/application.py |  6 +++++-
 cardvault/handlers.py    | 10 +++++-----
 2 files changed, 10 insertions(+), 6 deletions(-)

diff --git a/cardvault/application.py b/cardvault/application.py
index 60d8a99..f4b0a10 100644
--- a/cardvault/application.py
+++ b/cardvault/application.py
@@ -178,7 +178,10 @@ class Application:
 
     def show_question_dialog(self, title, message):
         dialog = Gtk.MessageDialog(self.ui.get_object("mainWindow"), 0, Gtk.MessageType.WARNING,
-                                   Gtk.ButtonsType.YES_NO, title)
+                                   Gtk.ButtonsType.NONE, title)
+        dialog.add_buttons(Gtk.STOCK_YES, Gtk.ResponseType.YES,
+                           Gtk.STOCK_NO, Gtk.ResponseType.NO,
+                           Gtk.STOCK_CANCEL, Gtk.ResponseType.CANCEL)
         dialog.format_secondary_text(message)
         response = dialog.run()
         dialog.destroy()
@@ -346,6 +349,7 @@ class Application:
     def remove_card_from_want_list(self, card: mtgsdk.Card, list: str):
         l = self.wants[list]
         l.remove(card)
+        self.unsaved_changes = True
         util.log("Removed '{}' from wants list '{}'".format(card.name, list), util.LogLevel.Info)
 
     def get_mana_icons(self, mana_string):
diff --git a/cardvault/handlers.py b/cardvault/handlers.py
index 3677dea..df17152 100644
--- a/cardvault/handlers.py
+++ b/cardvault/handlers.py
@@ -1,4 +1,5 @@
 import gi
+
 gi.require_version('Gtk', '3.0')
 import datetime
 import os
@@ -78,9 +79,12 @@ class Handlers:
     def do_delete_event(self, arg1, arg2):
         if self.app.unsaved_changes:
             response = self.app.show_question_dialog("Unsaved Changes", "You have unsaved changes in your library. "
-                                                                    "Save before exiting?")
+                                                                        "Save before exiting?")
             if response == Gtk.ResponseType.YES:
                 self.app.save_library()
+                return False
+            elif response == Gtk.ResponseType.CANCEL:
+                return True
 
     # ---------------------------------Search----------------------------------------------
 
@@ -213,7 +217,6 @@ class Handlers:
         self.app.ui.get_object("typeCombo").set_active(0)
         self.app.ui.get_object("setEntry").set_text("")
 
-
     def do_show_card_details(self, menu_item):
         tree = self.app.ui.get_object("searchResults").get_child()
         cards = tree.get_selected_cards()
@@ -287,7 +290,6 @@ class Handlers:
             self.app.rename_tag(tag, new_name)
             self.app.current_page.emit('show')
 
-
     def do_tag_list_delete(self, tree):
         (model, pathlist) = tree.get_selection().get_selected_rows()
         for path in pathlist:
@@ -521,5 +523,3 @@ class Handlers:
             self.app.ui.get_object("wants_cardListPopup").emit('show')
             self.app.ui.get_object("wants_cardListPopup").popup(None, None, None, None, 0, event.time)
             return True
-
-