Made the UI not hang wile grabbing cores via giving the core grabber its own thread.

RetroUFO_GUI.py

@@ -10,10 +10,12 @@ __license__ = "MIT"
 import os
 import platform
 import sys
+import time
 import zipfile
 from shutil import rmtree
 from urllib.request import urlretrieve
 
+from PySide2.QtCore import QThread, Signal
 from PySide2.QtWidgets import (QApplication, QCheckBox, QComboBox, QDialog,
                                QFileDialog, QLineEdit, QPushButton, QTextEdit,
                                QVBoxLayout, QMessageBox)
@@ -28,6 +30,71 @@ CORE_LOCATION = {
 }
 
 
+class GrabThread(QThread):
+    add_to_log = Signal(str)
+
+    def __init__(self, _platform, _architecture, _location):
+        QThread.__init__(self)
+        self.platform = _platform
+        self.architecture = _architecture
+        self.location = _location
+
+    def __del__(self):
+        self.wait()
+
+    def run(self):
+        self.add_to_log.emit('~Starting UFO Grabber~\n')
+        self.download_cores(self.platform, self.architecture)
+        self.extract_cores(self.location)
+        pass
+
+    def download_cores(self, _platform, _architecture):
+        """ Downloads every core to the working directory """
+
+        cores = []
+
+        # Makes core directory to store archives if needed
+        if not os.path.isdir('cores'):
+            os.makedirs("cores")
+        time.sleep(1)  # TODO: Do not leave this here! Find a way to wait for the core dir to be made!
+
+        # Downloads a list of all the cores available
+        urlretrieve(
+            '{}/{}/{}/latest/.index-extended'.format(
+                URL, _platform, _architecture), 'cores/index')
+        time.sleep(1)  # TODO: Do not leave this here! Find a way to wait for the index to download!
+        self.add_to_log.emit('Obtained core index!\n')
+
+        # Adds all the core's file names to a list
+        core_index = open('cores/index')
+
+        for line in core_index:
+            file_name = line.split(' ', 2)[2:]
+            cores.append(file_name[0].rstrip())
+        core_index.close()
+        cores.sort()
+
+        # Downloads each core from the list
+        self.add_to_log.emit('Downloading Cores\n')
+        for core in cores:
+            urlretrieve(
+                '{}/{}/{}/latest/{}'.format(URL, _platform, _architecture,
+                                            core), 'cores/{}'.format(core))
+            self.add_to_log.emit('Downloaded {} ...'.format(core))
+
+        # Removes index file for easier extraction
+        os.remove('cores/index')
+
+    def extract_cores(self, _location):
+        """ Extracts each downloaded core to the RA core directory """
+        self.add_to_log.emit('\nExtracting all cores to: {}\n'.format(_location))
+
+        for file in os.listdir('cores'):
+            archive = zipfile.ZipFile('cores/{}'.format(file))
+            archive.extractall(_location)
+            self.add_to_log.emit('Extracted {} ...'.format(file))
+
+
 class Form(QDialog):
     def __init__(self, parent=None):
         super(Form, self).__init__(parent)
@@ -108,18 +175,21 @@ class Form(QDialog):
 
         self.leditCoreLocation.insert(directory)
 
+    def update_log(self, _info):
+        self.teditLog.insertPlainText('{}\n'.format(_info))
+
     def grab_cores(self):
-        self.teditLog.insertPlainText('~Starting UFO Grabber~\n')
         """ Where the magic happens """
 
-        # If a platform and/or architecture is not supplied it is grabbed automatically
+        # TODO: Lock (disable) the UI elements while grabbing cores
-        platform = self.get_platform()  # TODO: rename this var to prevent conflict
-        architecture = self.get_architecture()
-        location = self.leditCoreLocation.text() if not self.chkboxLocationDetect.isChecked() \
-            else CORE_LOCATION[platform]
 
-        self.download_cores(platform, architecture)
+        platform = self.get_platform()
-        self.extract_cores(location)
+        architecture = self.get_architecture()
+        location = self.get_location()
+
+        self.grab = GrabThread(platform, architecture, location)
+        self.grab.add_to_log.connect(self.update_log)
+        self.grab.start()
 
         if not self.chkboxKeepDownload.isChecked():
             self.clean_up()
@@ -142,7 +212,6 @@ class Form(QDialog):
             else:
                 msgBox = QMessageBox.warning(self, 'Error', 'Platform not found or supported!', QMessageBox.Ok)
                 msgBox.exec_()
-                sys.exit(0)
 
     def get_architecture(self):
         """ Gets the Platform and Architecture if not supplied """
@@ -155,50 +224,12 @@ class Form(QDialog):
         else:
             msgBox = QMessageBox.warning(self, 'Error', 'Architecture not found or supported', QMessageBox.Ok)
             msgBox.exec_()
-            sys.exit(0)
 
-    def download_cores(self, _platform, _architecture):
+    def get_location(self):
-        """ Downloads every core to the working directory """
+        if not self.chkboxLocationDetect.isChecked():
-
+            return self.leditCoreLocation.text()
-        cores = []
+        else:
-
+            return CORE_LOCATION[self.get_platform()]
-        # Makes core directory to store archives if needed
-        if not os.path.isdir('cores'):
-            os.makedirs("cores")
-
-        # Downloads a list of all the cores available
-        urlretrieve(
-            '{}/{}/{}/latest/.index-extended'.format(
-                URL, _platform, _architecture), 'cores/index')
-        self.teditLog.insertPlainText('\nObtained core index!')
-
-        # Adds all the core's file names to a list
-        core_index = open('cores/index')
-
-        for line in core_index:
-            file_name = line.split(' ', 2)[2:]
-            cores.append(file_name[0].rstrip())
-        core_index.close()
-        cores.sort()
-
-        # Downloads each core from the list
-        for core in cores:
-            urlretrieve(
-                '{}/{}/{}/latest/{}'.format(URL, _platform, _architecture,
-                                            core), 'cores/{}'.format(core))
-            self.teditLog.insertPlainText('\nDownloaded {} ...'.format(core))
-
-        # Removes index file for easier extraction
-        os.remove('cores/index')
-
-    def extract_cores(self, _location):
-        """ Extracts each downloaded core to the RA core directory """
-        self.teditLog.insertPlainText('\nExtracting all cores to: {}'.format(_location))
-
-        for file in os.listdir('cores'):
-            archive = zipfile.ZipFile('cores/{}'.format(file))
-            archive.extractall(_location)
-            self.teditLog.insertPlainText('\nExtracted {} ...'.format(file))
 
     def clean_up(self):
         """ Removes all the downloaded files """