Bump version to 0.8.0 and add support for all image formats

kapitanbooru_uploader/ProcessingDialog.py

@@ -0,0 +1,119 @@
+from .I18N import _
+
+
+import inspect
+import queue
+import threading
+import tkinter as tk
+from tkinter import ttk
+
+
+class ProcessingDialog:
+    def __init__(self, root, target_function, *args):
+        self.root = root
+        self.top = tk.Toplevel(root)
+        self.top.title(_("Processing..."))
+        self.top.geometry("300x150")
+        self.top.protocol("WM_DELETE_WINDOW", self.on_close)
+
+        self.label = tk.Label(self.top, text=_("Processing, please wait..."))
+        self.label.pack(pady=10)
+
+        # Start with indeterminate progress bar
+        self.progress = ttk.Progressbar(self.top, mode="indeterminate")
+        self.progress.pack(pady=10, fill="x")
+        self.progress.start(10)
+
+        sig = inspect.signature(target_function)
+        if "secondary_progress_queue" in sig.parameters:
+            self.sub_progress = ttk.Progressbar(self.top, mode="indeterminate")
+            self.sub_progress.pack(pady=10, fill="x")
+            self.sub_progress.start(10)
+
+        # Setup communication queue and periodic checker
+        self.queue = queue.Queue()
+        self.sub_queue = queue.Queue()
+        self.running = True
+        self.cancel_event = threading.Event()  # Cancellation flag
+        self.thread = threading.Thread(
+            target=self.run_task, args=(target_function, *args)
+        )
+        self.thread.start()
+        self.top.after(100, self.process_queue)
+
+    def process_queue(self):
+        """Process messages from the background thread"""
+        while self.running:
+            try:
+                msg = self.queue.get_nowait()
+
+                if msg[0] == "mode":
+                    self.progress.config(mode=msg[1])
+                    if msg[1] == "determinate":
+                        self.progress["value"] = 0
+                        self.progress.stop()
+                    elif msg[1] == "indeterminate":
+                        self.progress["value"] = 0
+                        self.progress.start()
+                elif msg[0] == "max":
+                    self.progress["maximum"] = msg[1]
+                elif msg[0] == "progress":
+                    self.progress["value"] = msg[1]
+                elif msg[0] == "label":
+                    self.label.config(text=msg[1])
+
+                self.top.update_idletasks()
+            except queue.Empty:
+                break
+
+            try:
+                msg = self.sub_queue.get_nowait()
+
+                if msg[0] == "mode":
+                    self.sub_progress.config(mode=msg[1])
+                    if msg[1] == "determinate":
+                        self.sub_progress["value"] = 0
+                        self.sub_progress.stop()
+                    elif msg[1] == "indeterminate":
+                        self.sub_progress["value"] = 0
+                        self.sub_progress.start()
+                elif msg[0] == "max":
+                    self.sub_progress["maximum"] = msg[1]
+                elif msg[0] == "progress":
+                    self.sub_progress["value"] = msg[1]
+
+                self.top.update_idletasks()
+            except queue.Empty:
+                break
+
+        if self.running:
+            self.top.after(100, self.process_queue)
+
+    def run_task(self, target_function, *args):
+        """Execute target function with progress queue if supported"""
+        try:
+            sig = inspect.signature(target_function)
+            kwargs = {}
+            if "progress_queue" in sig.parameters:
+                kwargs["progress_queue"] = self.queue
+            if "secondary_progress_queue" in sig.parameters:
+                kwargs["secondary_progress_queue"] = self.sub_queue
+            if "cancel_event" in sig.parameters:
+                kwargs["cancel_event"] = self.cancel_event
+            target_function(*args, **kwargs)
+        finally:
+            self.close_dialog()
+
+    def close_dialog(self):
+        """Safely close the dialog"""
+        if self.running:
+            self.running = False
+            self.progress.stop()
+            self.top.after(0, self.top.destroy)
+        self.root.after(100, self.thread.join)
+
+    def on_close(self):
+        """Handle manual window closure"""
+        self.running = False
+        self.cancel_event.set()  # Notify target function that cancellation is requested
+        self.top.destroy()