From 201dae54aae35e706af57d334312da2a60665064 Mon Sep 17 00:00:00 2001
From: Mykola Grymalyuk <khronokernel@icloud.com>
Date: Wed, 6 Apr 2022 15:25:10 -0600
Subject: [PATCH] gui_main.py: Disable return button while patching

---
 gui/gui_main.py | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/gui/gui_main.py b/gui/gui_main.py
index 98ed354a8..7550742d0 100644
--- a/gui/gui_main.py
+++ b/gui/gui_main.py
@@ -884,6 +884,7 @@ class wx_python_gui:
         )
         self.return_to_main_menu.Bind(wx.EVT_BUTTON, self.main_menu)
         self.return_to_main_menu.Centre(wx.HORIZONTAL)
+        self.return_to_main_menu.Disable()
 
         self.frame.SetSize(-1, self.return_to_main_menu.GetPosition().y + self.return_to_main_menu.GetSize().height + 40)
 
@@ -900,6 +901,7 @@ class wx_python_gui:
             pass
         sys.stdout = self.stock_stdout
         sys.stderr = self.stock_stderr
+        self.return_to_main_menu.Enable()
 
         wx.GetApp().Yield()
     
@@ -963,6 +965,7 @@ class wx_python_gui:
         )
         self.return_to_main_menu.Bind(wx.EVT_BUTTON, self.main_menu)
         self.return_to_main_menu.Centre(wx.HORIZONTAL)
+        self.return_to_main_menu.Disable()
 
         self.frame.SetSize(-1, self.return_to_main_menu.GetPosition().y + self.return_to_main_menu.GetSize().height + 40)
 
@@ -977,6 +980,7 @@ class wx_python_gui:
             pass
         sys.stdout = self.stock_stdout
         sys.stderr = self.stock_stderr
+        self.return_to_main_menu.Enable()
 
         wx.GetApp().Yield()
 
@@ -1043,18 +1047,16 @@ class wx_python_gui:
         self.download_label.SetPosition(
             wx.Point(
                 self.header.GetPosition().x,
-                self.header.GetPosition().y + self.header.GetSize().height + 10
+                self.header.GetPosition().y + self.header.GetSize().height + 15
             )
         )
         self.download_label.Centre(wx.HORIZONTAL)
-        # Redirect stdout to label
-        sys.stdout=menu_redirect.RedirectLabel(self.download_label)
 
         self.return_to_main_menu = wx.Button(self.frame, label="Return to Main Menu")
         self.return_to_main_menu.SetPosition(
             wx.Point(
                 self.download_label.GetPosition().x,
-                self.download_label.GetPosition().y + self.download_label.GetSize().height + 30
+                self.download_label.GetPosition().y + self.download_label.GetSize().height + 15
             )
         )
         self.return_to_main_menu.Bind(wx.EVT_BUTTON, self.main_menu)
@@ -1062,12 +1064,12 @@ class wx_python_gui:
 
         self.frame.SetSize(-1, self.return_to_main_menu.GetPosition().y + self.return_to_main_menu.GetSize().height + 40)
         self.frame.Show()
+        wx.GetApp().Yield()
 
         # Download installer catalog
         available_installers = installer.list_downloadable_macOS_installers(self.constants.payload_path, "PublicSeed")
 
         self.frame.DestroyChildren()
-        sys.stdout = self.stock_stdout
 
         # Header
         self.header = wx.StaticText(self.frame, label="Download macOS Installer")