From 50d53beef6ec5a2d2d61e379d4da9a4d716fc562 Mon Sep 17 00:00:00 2001
From: Alan Mason <1923621+2Shirt@users.noreply.github.com>
Date: Fri, 24 Nov 2017 21:12:54 -0800
Subject: [PATCH] 2017-04: Retroactive Updates

* Adjusted data loss waning
---
 Scripts/menu.py | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/Scripts/menu.py b/Scripts/menu.py
index d7d010d..eb84d18 100644
--- a/Scripts/menu.py
+++ b/Scripts/menu.py
@@ -12,6 +12,14 @@ bin = os.path.abspath('..\\')
 sys.path.append(os.getcwd())
 from functions import *
 
+## Colors
+COLORS = {
+    'CLEAR': '\033[0m',
+    'RED': '\033[31m',
+    'GREEN': '\033[32m',
+    'YELLOW': '\033[33m',
+    'BLUE': '\033[34m'}
+
 def menu_backup_imaging():
     """Take backup images of partition(s) in the WIM format and save them to a backup share"""
     errors = False
@@ -121,8 +129,7 @@ def menu_windows_setup():
     # Safety check    
     print('\nSAFETY CHECK')
     print_warning('All data will be DELETED from the drive and partition(s) listed above.')
-    print_error('This is irreversible and will lead to ', end='', flush=True)
-    print('DATA LOSS.')
+    print_warning('This is irreversible and will lead to {CLEAR}{RED}DATA LOSS.'.format(**COLORS))
     if (not ask('Asking again to confirm, is this correct?')):
         abort_to_main_menu('Aborting Windows setup')