Allow bypassing confirm message for benchmarks

This commit is contained in:
Joshua Boniface 2021-12-23 21:00:42 -05:00
parent 3ce4d90693
commit 9f122e916f
1 changed files with 18 additions and 9 deletions

View File

@ -3178,20 +3178,29 @@ def ceph_benchmark():
# pvc storage benchmark run # pvc storage benchmark run
############################################################################### ###############################################################################
@click.command(name="run", short_help="Run a storage benchmark.") @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") @click.argument("pool")
@cluster_req @cluster_req
def ceph_benchmark_run(pool): def ceph_benchmark_run(confirm_flag, pool):
""" """
Run a storage benchmark on POOL in the background. Run a storage benchmark on POOL in the background.
""" """
try: if not confirm_flag and not config["unsafe"]:
click.confirm( try:
"NOTE: Storage benchmarks take approximately 10 minutes to run and generate significant load on the cluster; they should be run sparingly. Continue", click.confirm(
prompt_suffix="? ", "NOTE: Storage benchmarks take approximately 10 minutes to run and generate significant load on the cluster; they should be run sparingly. Continue",
abort=True, prompt_suffix="? ",
) abort=True,
except Exception: )
exit(0) except Exception:
exit(0)
retcode, retmsg = pvc_ceph.ceph_benchmark_run(config, pool) retcode, retmsg = pvc_ceph.ceph_benchmark_run(config, pool)
cleanup(retcode, retmsg) cleanup(retcode, retmsg)