From be1b67b8f067a35f1d5a99ce703d00b51b17ec75 Mon Sep 17 00:00:00 2001
From: "Joshua M. Boniface" <joshua@boniface.me>
Date: Thu, 23 Dec 2021 21:00:42 -0500
Subject: [PATCH] Allow bypassing confirm message for benchmarks

---
 client-cli/pvc/pvc.py | 27 ++++++++++++++++++---------
 1 file changed, 18 insertions(+), 9 deletions(-)

diff --git a/client-cli/pvc/pvc.py b/client-cli/pvc/pvc.py
index fb3f917f..e542377a 100755
--- a/client-cli/pvc/pvc.py
+++ b/client-cli/pvc/pvc.py
@@ -3178,20 +3178,29 @@ def ceph_benchmark():
 # pvc storage benchmark run
 ###############################################################################
 @click.command(name="run", short_help="Run a storage benchmark.")
+@click.option(
+    "-y",
+    "--yes",
+    "confirm_flag",
+    is_flag=True,
+    default=False,
+    help="Confirm the run",
+)
 @click.argument("pool")
 @cluster_req
-def ceph_benchmark_run(pool):
+def ceph_benchmark_run(confirm_flag, pool):
     """
     Run a storage benchmark on POOL in the background.
     """
-    try:
-        click.confirm(
-            "NOTE: Storage benchmarks take approximately 10 minutes to run and generate significant load on the cluster; they should be run sparingly. Continue",
-            prompt_suffix="? ",
-            abort=True,
-        )
-    except Exception:
-        exit(0)
+    if not confirm_flag and not config["unsafe"]:
+        try:
+            click.confirm(
+                "NOTE: Storage benchmarks take approximately 10 minutes to run and generate significant load on the cluster; they should be run sparingly. Continue",
+                prompt_suffix="? ",
+                abort=True,
+            )
+        except Exception:
+            exit(0)
 
     retcode, retmsg = pvc_ceph.ceph_benchmark_run(config, pool)
     cleanup(retcode, retmsg)