Add --yes option and confirm prompt.

This commit is contained in:
Christopher Laprise 2017-12-29 14:17:30 -05:00
parent c82d5b0a73
commit ed11346977
No known key found for this signature in database
GPG Key ID: 448568C8B281C952

View File

@ -28,14 +28,26 @@ from qubesadmin.tools import QubesArgumentParser
parser = QubesArgumentParser(description=__doc__, parser = QubesArgumentParser(description=__doc__,
want_app=True, want_app=True,
vmname_nargs='+') vmname_nargs='+')
parser.add_argument("--yes", action="store_true", dest="no_confirm",
default=False, help="Do not prompt for confirmation")
def main(args=None, app=None): # pylint: disable=missing-docstring def main(args=None, app=None): # pylint: disable=missing-docstring
args = parser.parse_args(args, app=app) args = parser.parse_args(args, app=app)
for vm in args.domains: go_ahead = ""
del args.app.domains[vm.name] if not args.no_confirm:
print("This will completely remove the selected VM(s)...")
go_ahead = input("Are you sure? [y/N] ").upper()
if args.no_confirm or go_ahead == "Y":
for vm in args.domains:
del args.app.domains[vm.name]
return 0
else:
print("Remove cancelled.")
return 1
return 0
if __name__ == '__main__': if __name__ == '__main__':