Properly handle live state with restart confirm

If "--live" is passed (the default), we shouldn't confirm to restart the
VM as this is not required. Instead only confirm if "--no-live" was
passed or if the flag doesn't exist.
This commit is contained in:
Joshua Boniface 2023-11-01 11:46:59 -04:00
parent 5b4dd61754
commit 988c777912
1 changed files with 3 additions and 2 deletions

View File

@ -169,9 +169,10 @@ def restart_opt(function):
@wraps(function)
def confirm_action(*args, **kwargs):
restart_state = kwargs.get("restart_flag", None)
live_state = kwargs.get("live_flag", False)
if restart_state is None:
# Neither "--restart" or "--no-restart" was passed: prompt for restart or restart if "--unsafe"
if restart_state is None and not live_state:
# Neither "--restart" or "--no-restart" was passed, and "--no-live" was passed: prompt for restart or restart if "--unsafe"
try:
click.confirm(
f"Restart VM {kwargs.get('domain')} to apply changes",