Ticket #2028: save-before-quit.patch

tawindow.py

@@ -284 +284 @@
                 self._run_stack(blk)
         return
 
+    def is_project_empty(self):
+        return len(self.just_blocks()) == 1
+
     def stop_button(self):
         """ Stop button """
         stop_logo(self)

turtleart.py

@@ -117 +117 @@
             win.move(self.x, self.y)
             win.maximize()
             win.set_title(_("Turtle Art"))
-            win.connect("delete_event", lambda w, e: gtk.main_quit())
+            win.connect("delete_event", self._quit_ta)
 
             menu = gtk.Menu()
 
@@ -304 +304 @@
             self.tw.run_button(0)
             self.tw.save_as_image(self.ta_file, canvas)
 
+
+    def _quit_ta(self, widget, e):
+        project_empty = self.tw.is_project_empty()
+
+        if project_empty == False:
+            dlg = gtk.MessageDialog(parent=None, type=gtk.MESSAGE_INFO,
+                                    buttons=gtk.BUTTONS_OK_CANCEL,
+                                    message_format = _("You have un-saved work. Would you like to save before quitting?"))
+            dlg.set_title("Save project?")
+            dlg.set_property("skip-taskbar-hint", False)
+
+            resp = dlg.run()
+            dlg.destroy()
+            if resp == gtk.RESPONSE_OK:
+                self._save_as()
+
+        gtk.main_quit()
+
     def _do_new_cb(self, widget):
         """ Callback for new project. """
         self.tw.new_project()
@@ -319 +337 @@
 
     def _do_save_as_cb(self, widget):
         """ Callback for save-as project. """
+        self._save_as()
+
+    def _save_as(self):
         self.tw.save_file_name = None
         self.tw.save_file()