From 3fe735e596ccea02ad183f7464b3318085c6eb91 Mon Sep 17 00:00:00 2001
From: Marcel Kapfer <opensource@mmk2410.org>
Date: Wed, 9 Sep 2020 23:45:06 +0200
Subject: [PATCH] Offer clearing fields when closing code dialog

The offer is only made if the content has been copied at least once
---
 wikicookbook.py | 20 +++++++++++++++++++-
 1 file changed, 19 insertions(+), 1 deletion(-)

diff --git a/wikicookbook.py b/wikicookbook.py
index ff853f3..4e545cb 100644
--- a/wikicookbook.py
+++ b/wikicookbook.py
@@ -53,10 +53,11 @@ class WikiCodeDialog(QDialog, Ui_WikiCode):
     def __init__(self, code, *args, **kwargs):
         super(WikiCodeDialog, self).__init__(*args, **kwargs)
         self.setupUi(self)
+        self.copied = False
 
         self.e_code.setPlainText(code)
 
-        self.b_close.clicked.connect(self.close)
+        self.b_close.clicked.connect(self.clear_fields)
         self.b_copy.clicked.connect(self.copy)
 
         self.show()
@@ -64,6 +65,20 @@ class WikiCodeDialog(QDialog, Ui_WikiCode):
     def copy(self):
         self.e_code.selectAll()
         self.e_code.copy()
+        self.copied = True
+
+    def clear_fields(self):
+        """
+        Show the clear fields dialog if the code was at least copied once.
+        """
+        if self.copied:
+            dialog = ClearDialog()
+            dialog.rejected.connect(self.reject)
+            dialog.accepted.connect(self.accept)
+            dialog.exec_()
+        else:
+            self.close()
+
 
 class MainWindow(QMainWindow, Ui_MainWindow):
     def __init__(self, *args, **kwargs):
@@ -289,6 +304,9 @@ class MainWindow(QMainWindow, Ui_MainWindow):
         code = current_recipe.wikicode(writer)
 
         dialog = WikiCodeDialog(code)
+        dialog.accepted.connect(self.clear)
+        dialog.exec_()
+
     def clear(self):
         """
         Clear all fields