From e892cd34064aea127f345a86ee928a4e8096a79d Mon Sep 17 00:00:00 2001
From: Alan Mason <1923621+2Shirt@users.noreply.github.com>
Date: Sun, 3 Dec 2017 09:48:59 -0800
Subject: [PATCH] Attempt to silence more "No disk" errors

---
 .bin/Scripts/functions/windows_setup.py | 2 ++
 .bin/Scripts/functions/winpe_menus.py   | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/.bin/Scripts/functions/windows_setup.py b/.bin/Scripts/functions/windows_setup.py
index ab2b3fcd..2db6ceb6 100644
--- a/.bin/Scripts/functions/windows_setup.py
+++ b/.bin/Scripts/functions/windows_setup.py
@@ -44,6 +44,7 @@ def find_windows_image(windows_version):
     imagename = windows_version['Image Name']
 
     # Search local source
+    set_thread_error_mode(silent=True) # Prevents "No disk" popups
     for d in psutil.disk_partitions():
         for ext in ['esd', 'wim', 'swm']:
             path = '{}images\{}.{}'.format(d.mountpoint, imagefile, ext)
@@ -53,6 +54,7 @@ def find_windows_image(windows_version):
                 if ext == 'swm':
                     image['Glob'] = '--ref="{}*.swm"'.format(image['Path'][:-4])
                 break
+    set_thread_error_mode(silent=False) # Return to normal
 
     # Check for network source
     if not image:
diff --git a/.bin/Scripts/functions/winpe_menus.py b/.bin/Scripts/functions/winpe_menus.py
index 3b79df77..c2b7cb5c 100644
--- a/.bin/Scripts/functions/winpe_menus.py
+++ b/.bin/Scripts/functions/winpe_menus.py
@@ -415,6 +415,7 @@ def select_minidump_path():
     assign_volume_letters()
 
     # Search for minidumps
+    set_thread_error_mode(silent=True) # Prevents "No disk" popups
     tmp = run_program('mountvol')
     tmp = [d for d in re.findall(r'.*([A-Za-z]):\\', tmp.stdout.decode())]
     # Remove RAMDisk letter
@@ -423,6 +424,7 @@ def select_minidump_path():
     for disk in tmp:
         if os.path.exists('{}:\\Windows\\MiniDump'.format(disk)):
             dumps.append({'Name': '{}:\\Windows\\MiniDump'.format(disk)})
+    set_thread_error_mode(silent=False) # Return to normal
 
     # Check results before showing menu
     if len(dumps) == 0: