From 23f55fd654ebad024d212d3e5b7760b718d3ec08 Mon Sep 17 00:00:00 2001 From: narugo1992 Date: Sun, 11 Aug 2024 03:45:35 +0800 Subject: [PATCH] dev(narugo): add cancel --- felinewhisker/entry/init.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/felinewhisker/entry/init.py b/felinewhisker/entry/init.py index 377db48..c79552f 100644 --- 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