Merge pull request #2550 from jestabro/non-interactive-add-delete-image

image-tools: T5751: allow non-interactive add/delete image

1 files changed, 10 insertions, 7 deletions

diff --git a/src/op_mode/image_manager.py b/src/op_mode/image_manager.py
index e4b2f4833..e75485f9f 100755
--- a/src/op_mode/image_manager.py
+++ b/src/op_mode/image_manager.py
@@ -36,7 +36,7 @@ MSG_DELETE_IMAGE_DEFAULT: str = 'Default image cannot be deleted; set another im
 
 @compat.grub_cfg_update
 def delete_image(image_name: Optional[str] = None,
-                 prompt: bool = True) -> None:
+                 no_prompt: bool = False) -> None:
     """Remove installed image files and boot entry
 
     Args:
@@ -44,7 +44,7 @@ def delete_image(image_name: Optional[str] = None,
     """
     available_images: list[str] = grub.version_list()
     if image_name is None:
-        if not prompt:
+        if no_prompt:
             exit('An image name is required for delete action')
         else:
             image_name = select_entry(available_images,
@@ -60,8 +60,9 @@ def delete_image(image_name: Optional[str] = None,
     if not persistence_storage:
         exit('Persistence storage cannot be found')
 
-    if not ask_yes_no(f'Do you really want to delete the image {image_name}?',
-                      default=False):
+    if (not no_prompt and
+        not ask_yes_no(f'Do you really want to delete the image {image_name}?',
+                       default=False)):
         exit()
 
     # remove files and menu entry
@@ -171,11 +172,13 @@ def parse_arguments() -> Namespace:
                         choices=['delete', 'set', 'rename', 'list'],
                         required=True,
                         help='action to perform with an image')
+    parser.add_argument('--no-prompt', action='store_true',
+                        help='perform action non-interactively')
     parser.add_argument(
-        '--image_name',
+        '--image-name',
         help=
         'a name of an image to add, delete, install, rename, or set as default')
-    parser.add_argument('--image_new_name', help='a new name for image')
+    parser.add_argument('--image-new-name', help='a new name for image')
     args: Namespace = parser.parse_args()
     # Validate arguments
     if args.action == 'rename' and (not args.image_name or
@@ -189,7 +192,7 @@ if __name__ == '__main__':
     try:
         args: Namespace = parse_arguments()
         if args.action == 'delete':
-            delete_image(args.image_name)
+            delete_image(args.image_name, args.no_prompt)
         if args.action == 'set':
             set_image(args.image_name)
         if args.action == 'rename':