dev(narugo): add cancel

deepghs · Aug 10, 2024 · 23f55fd · 23f55fd
1 parent 95bfd17
commit 23f55fd
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/felinewhisker/entry/init.py b/felinewhisker/entry/init.py
@@ -12,12 +12,16 @@
 from prompt_toolkit.document import Document
 from prompt_toolkit.validation import Validator, ValidationError
 
-from .base import CONTEXT_SETTINGS
+from .base import CONTEXT_SETTINGS, ClickWarningException
 from ..repository import LocalRepository, HfOnlineRepository, RepoAlreadyExistsError
 from ..tasks import init_cli, list_task_types
 from ..utils import HuggingFaceRepoValidator, StringNonEmptyValidator, hf_licence
 
 
+class InitCanceledException(ClickWarningException):
+    exit_code = 0x20
+
+
 class NewDirectoryValidator(Validator):
     def __init__(self, message: str = "Input is not a valid directory"):
         self._message = message
@@ -115,5 +119,7 @@ def init():
             ).execute()
             if confirm_clear:
                 fn_init(**init_params, force=True)
+            else:
+                raise InitCanceledException('Initialization cancelled.')
 
     return cli