1 files changed, 13 insertions, 14 deletions
diff --git a/src/utils/legacy.py b/src/utils/legacy.py
index 387f790..d3a0318 100644
--- a/src/utils/legacy.py
+++ b/src/utils/legacy.py
@@ -209,8 +209,7 @@ def restoreImageCustom(repo_ip, image_name, disk, partition, method):
     """
     """
     if not shutil.which('restoreImageCustom'):
-        logging.error('Invalid restoreImageCustom invocation')
-        raise ValueError('Error: restoreImageCustom not found')
+        raise OSError('restoreImageCustom not found')
 
     cmd = f'restoreImageCustom {repo_ip} {image_name} {disk} {partition} {method}'
     with open('/tmp/command.log', 'wb', 0) as logfile:
@@ -218,10 +217,10 @@ def restoreImageCustom(repo_ip, image_name, disk, partition, method):
             proc = subprocess.run(cmd,
                                   stdout=logfile,
                                   encoding='utf-8',
-                                  shell=True)
-        except:
-            logging.error('Exception when running restoreImageCustom subprocess')
-            raise ValueError('Error: Incorrect command value')
+                                  shell=True,
+                                  check=True)
+        except OSError as e:
+            raise OSError(f'Error processing restoreImageCustom: {e}') from e
     return proc.returncode
 
 
@@ -237,11 +236,11 @@ def configureOs(disk, partition):
         proc = subprocess.run(cmd_configure,
                               stdout=PIPE,
                               encoding='utf-8',
-                              shell=True)
+                              shell=True,
+                              check=True)
         out = proc.stdout
-    except:
-        logging.error('Exception when running configureOs subprocess')
-        raise ValueError('Error: Incorrect command value')
+    except OSError as e:
+        raise OSError(f'Error processing configureOsCustom: {e}') from e
 
     return out
 
@@ -250,7 +249,7 @@ def ogCopyEfiBootLoader(disk, partition):
     cmd = f'ogCopyEfiBootLoader {disk} {partition}'
     try:
         proc = subprocess.run(cmd,
-                              shell=True)
-    except:
-        logging.error('Exception when running ogCopyEfiBootLoader subprocess')
-        raise ValueError('Subprocess error: ogCopyEfiBootloader')
+                              shell=True,
+                              check=True)
+    except OSError as e:
+        raise OSError(f'Error processing ogCopyEfiBootLoader: {e}') from e