From a8cad482bbce77a1449d8778de5e030766fbe8b7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Berkay=20=C3=9Cr=C3=BCn?=
 <berkay.ueruen@aisec.fraunhofer.de>
Date: Wed, 5 Jul 2023 14:56:27 +0200
Subject: [PATCH] Add overwrite flag

---
 controller.py | 17 ++++++++++++++---
 1 file changed, 14 insertions(+), 3 deletions(-)

diff --git a/controller.py b/controller.py
index 82648a0..d77ebfa 100755
--- a/controller.py
+++ b/controller.py
@@ -505,7 +505,7 @@ def controller(
     goldenrun_data = {}
 
     hdf5_file = Path(hdf5path)
-    if hdf5_file.is_file():
+    if hdf5_file.is_file() and not args.overwrite:
         clogger.info("HDF5 file already exits")
         try:
             [
@@ -514,7 +514,9 @@ def controller(
                 backup_goldenrun_data,
             ] = read_backup(hdf5_file)
         except NameError:
-            clogger.warning("Backup could not be found in the HDF5 file")
+            clogger.warning(
+                "Backup could not be found in the HDF5 file, run with the overwrite flag to overwrite!"
+            )
             return config_qemu
         clogger.info("Checking the backup")
         if check_backup(args, config_qemu, backup_config):
@@ -532,7 +534,9 @@ def controller(
             if not args.append:
                 overwrite_faults = True
         else:
-            clogger.warning("Backup does not match, ending the execution!")
+            clogger.warning(
+                "Backup does not match, run with the overwrite flag to overwrite!"
+            )
             return config_qemu
 
     if goldenrun:
@@ -768,6 +772,13 @@ def get_argument_parser():
         action="store_true",
         required=False,
     )
+    parser.add_argument(
+        "--overwrite",
+        "-o",
+        help="Overwrite the existing file",
+        action="store_true",
+        required=False,
+    )
     return parser