Merge pull request sonata-nfv#71 from mpeuster/dev

Added result overwriting functionality.

pipeline_local.sh

@@ -5,7 +5,8 @@ set -e
 
 # check style early
 flake8 --exclude .eggs --exclude venv --exclude build
-
+# local test
+pytest -v
 # build container
 pipeline/build/build.sh
 # run tests

src/tngsdk/benchmark/__init__.py

@@ -36,6 +36,7 @@
 import logging
 import coloredlogs
 import time
+import shutil
 from tngsdk.benchmark.experiment import ServiceExperiment, FunctionExperiment
 from tngsdk.benchmark.generator.sonata \
                 import SonataServiceConfigurationGenerator
@@ -106,6 +107,7 @@ def run(self):
         Run son-profile
         :return:
         """
+        self.check_rd_existence()
         self.populate_experiments()
         # trigger experiment execution
         self.cgen = self.load_generator()
@@ -115,6 +117,22 @@ def run(self):
         self.execute_experiments()
         self.process_results()
 
+    def check_rd_existence(self):
+        if os.path.exists(self.args.result_dir):
+            self.logger.info("Found old results: {}"
+                             .format(self.args.result_dir))
+            # ask for overwrite (if not -y/--force-yes)
+            if not self.args.force_yes:
+                # ask user
+                ui = input("Do you want to overwrite '{}'? (y/n/default: y)"
+                           .format(self.args.result_dir))
+                if "n" in ui or "N" in ui:
+                    return
+            # delte old results
+            self.logger.info("Overwriting old results: {}"
+                             .format(self.args.result_dir))
+            shutil.rmtree(self.args.result_dir)
+
     def populate_experiments(self):
         if self.args.no_population:
             print("Skipping population: --no-population")
@@ -424,6 +442,15 @@ def parse_args(manual_args=None,
         default=None,
         dest="ibbd_dir")
 
+    parser.add_argument(
+        "-y",
+        "--force-yes",
+        help="Answer all user questions that might appear with: yes.",
+        required=False,
+        default=False,
+        dest="force_yes",
+        action="store_true")
+
     if manual_args is not None:
         return parser.parse_args(manual_args)
     return parser.parse_args()