From 01bf95e8630190ba8b9976e8b4d5254bd7989c26 Mon Sep 17 00:00:00 2001
From: Artem30801 <artem30801@gmail.com>
Date: Wed, 5 Feb 2020 21:36:42 +0300
Subject: [PATCH] Preset dialog close confirmation

---
 Server/copter_table.py | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/Server/copter_table.py b/Server/copter_table.py
index 35283ac..3192f3a 100644
--- a/Server/copter_table.py
+++ b/Server/copter_table.py
@@ -415,6 +415,20 @@ class HeaderEditDialog(QtWidgets.QDialog):
         self.setWindowTitle(f"Column preset editor - {self.widget.preset_widget.currentText()}"
                             + "*"*unsaved)
 
+    def closeEvent(self, event):
+        if not self.unsaved:
+            event.accept()
+            return
+
+        reply = QMessageBox.question(self, "Confirm exit", "There are unsaved changes in current preset. "
+                                                           "Are you sure you want to exit?",
+                                     QMessageBox.No | QMessageBox.Yes, QMessageBox.No)
+
+        if reply != QMessageBox.Yes:
+            event.ignore()
+        else:
+            event.accept()
+
 
 if __name__ == '__main__':
     import sys