CCT-1200: updated virtwho_host function to provide an argument reserv…

…e_duration
VirtwhoQE · Feb 24, 2025 · 43f6cb2 · 43f6cb2
1 parent 42ec915
commit 43f6cb2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/utils/beaker.py b/utils/beaker.py
@@ -93,7 +93,9 @@ def beaker_job_submit(
         "--task /distribution/reservesys"
     )
     whiteboard = f'--whiteboard="reserve host for {job_name}"'
-    reserve = f"--reserve --reserve-duration {reserve_duration} --priority Urgent"
+    reserve = "--reserve  --priority Urgent" + \
+        ((reserve_duration and f" --reserve-duration {reserve_duration}") or \
+        (config.beaker.get("reserve_duration") and f" --reserve-duration {config.beaker.get('reserve_duration')}") or "")
     cmd = (
         f"bkr workflow-simple --prettyxml "
         f"{task} {whiteboard} {reserve} "

diff --git a/virtwho/provision/virtwho_host.py b/virtwho/provision/virtwho_host.py
@@ -159,7 +159,7 @@ def beaker_args_define(args):
     args.host = args.beaker_host
     args.host_type = "physical"
     args.host_require = None
-
+    args.reserve_duration = None
 
 def virtwho_install(ssh, url=None):
     """