adding graceful termination of keyboard interrupt

Signed-off-by: Michael Haigh <mhaigh@netapp.com>
NetApp · Apr 26, 2024 · 3391daf · 3391daf
1 parent 6a12daf
commit 3391daf
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/tkSrc/create.py b/tkSrc/create.py
@@ -440,7 +440,7 @@ def main(args, parser, ard):
                 args.snapshot,
                 args.reclaimPolicy,
             )
-            if not args.dry_run and not args.background:
+            if backup and not args.dry_run and not args.background:
                 monitorV3ProtectionTask(
                     backup, args.pollTimer, parser, args.v3, args.skip_tls_verify
                 )
@@ -709,7 +709,7 @@ def main(args, parser, ard):
                 createdTimeout=args.createdTimeout,
                 readyToUseTimeout=args.readyToUseTimeout,
             )
-            if not args.dry_run and not args.background:
+            if snapshot and not args.dry_run and not args.background:
                 monitorV3ProtectionTask(
                     snapshot, args.pollTimer, parser, args.v3, args.skip_tls_verify
                 )

diff --git a/toolkit.py b/toolkit.py
@@ -249,4 +249,7 @@ def main(argv=sys.argv):
 
 
 if __name__ == "__main__":
-    main()
+    try:
+        main()
+    except KeyboardInterrupt:
+        pass