From 47ca889e22a4a8c211ceadb8101f46fa02ce0b65 Mon Sep 17 00:00:00 2001
From: Melon Bread <bread.melon@gmail.com>
Date: Tue, 5 Mar 2019 13:18:30 -0500
Subject: [PATCH] Locks UI while grabbing cores

---
 RetroUFO_GUI.py | 39 +++++++++++++++++++++++++++------------
 1 file changed, 27 insertions(+), 12 deletions(-)

diff --git a/RetroUFO_GUI.py b/RetroUFO_GUI.py
index 1ca606d..737b20b 100755
--- a/RetroUFO_GUI.py
+++ b/RetroUFO_GUI.py
@@ -32,6 +32,7 @@ CORE_LOCATION = {
 
 class GrabThread(QThread):
     add_to_log = Signal(str)
+    lock = Signal(bool)
 
     def __init__(self, _platform, _architecture, _location):
         QThread.__init__(self)
@@ -43,10 +44,11 @@ class GrabThread(QThread):
         self.wait()
 
     def run(self):
+        self.lock.emit(True)
         self.add_to_log.emit('~Starting UFO Grabber~\n')
         self.download_cores(self.platform, self.architecture)
         self.extract_cores(self.location)
-        pass
+        self.lock.emit(False)
 
     def create_dir(self, _name):
         if not os.path.isdir(_name):
@@ -145,19 +147,19 @@ class Form(QDialog):
         self.btnGrabCores.clicked.connect(self.grab_cores)
 
         # Create layout and add widgets
-        layout = QVBoxLayout()
-        layout.addWidget(self.chkboxPlatformDetect)
-        layout.addWidget(self.cmbboxPlatform)
-        layout.addWidget(self.cmbboxArchitecture)
-        layout.addWidget(self.chkboxLocationDetect)
-        layout.addWidget(self.leditCoreLocation)
-        layout.addWidget(self.btnCoreLocation)
-        layout.addWidget(self.teditLog)
-        layout.addWidget(self.chkboxKeepDownload)
-        layout.addWidget(self.btnGrabCores)
+        self.formLayout = QVBoxLayout()
+        self.formLayout.addWidget(self.chkboxPlatformDetect)
+        self.formLayout.addWidget(self.cmbboxPlatform)
+        self.formLayout.addWidget(self.cmbboxArchitecture)
+        self.formLayout.addWidget(self.chkboxLocationDetect)
+        self.formLayout.addWidget(self.leditCoreLocation)
+        self.formLayout.addWidget(self.btnCoreLocation)
+        self.formLayout.addWidget(self.teditLog)
+        self.formLayout.addWidget(self.chkboxKeepDownload)
+        self.formLayout.addWidget(self.btnGrabCores)
 
         # Set dialog layout
-        self.setLayout(layout)
+        self.setLayout(self.formLayout)
 
     def auto_detect(self):
         if self.chkboxPlatformDetect.isChecked():
@@ -187,6 +189,18 @@ class Form(QDialog):
         # Auto scrolling on log UI
         self.teditLog.moveCursor(QTextCursor.End)
 
+    def lock_ui(self, _lock):
+        # Cycle through each widget and disable it except for log UI
+        widgets = (self.formLayout.itemAt(i).widget() for i in range(self.formLayout.count()))
+        for widget in widgets:
+            if isinstance(widget, QTextEdit):
+                pass
+            else:
+                widget.setDisabled(_lock)
+                # Have to run these to make sure only the correct things unlock after grab thread
+                self.auto_detect()
+                self.auto_location()
+
 
     def grab_cores(self):
         """ Where the magic happens """
@@ -201,6 +215,7 @@ class Form(QDialog):
 
         self.grab = GrabThread(platform, architecture, location)
         self.grab.add_to_log.connect(self.update_log)
+        self.grab.lock.connect(self.lock_ui)
         self.grab.start()
 
     def get_platform(self):