Merge pull request #99 from uright008/feature/confirm-before-exit

feat: confirm before exit if there is still any task

Author: WorldObservationLog

src/cmd.py

@@ -1,6 +1,7 @@
 import argparse
 import asyncio
 import sys
+import os
 
 import grpc.aio
 from creart import it
@@ -111,8 +112,7 @@ async def command_parser(self, cmd: str):
             case "status":
                 await self.show_status()
             case "exit":
-                self.loop.stop()
-                sys.exit()
+                self.handle_exit()
 
     async def do_download(self, raw_urls: list[str], codec: str, force_download: bool, language: str, include: bool = False):
         for raw_url in raw_urls:
@@ -171,6 +171,16 @@ def completer(self):
         }
         return NestedCompleter.from_nested_dict(mycompleter)
 
+    def handle_exit(self):
+        if it(Measurer).tasks_count() > 0:
+            it(GlobalLogger).logger.info("There is still {} tasks, do you really want to exit? (y/N)".format(it(Measurer).tasks_count()))
+            response = input().strip().lower()
+            if response != 'y':
+                return
+        it(GlobalLogger).logger.info("Exit.")
+        self.loop.stop()
+        os._exit(0)
+
     async def handle_command(self):
         session = PromptSession("> ", bottom_toolbar=self.bottom_toolbar, completer=self.completer(), refresh_interval=1)
 
@@ -186,7 +196,7 @@ async def handle_command(self):
                 else:
                     await self.command_parser(command)
             except (EOFError, KeyboardInterrupt):
-                return
+                self.handle_exit()
 
     async def on_2fa(self, username: str, password: str):
         session = PromptSession()
@@ -224,6 +234,5 @@ async def start(self):
             try:
                 await self.handle_command()
             finally:
-                it(GlobalLogger).logger.info("Exit.")
                 if it(Config).localInstance.enable:
-                    await self.localInstance.terminate()
+                    self.localInstance.terminate()