From 27ad63341205103b817af0c60a1abd26d7e19d06 Mon Sep 17 00:00:00 2001 From: crupest Date: Mon, 11 Nov 2024 01:12:29 +0800 Subject: HALF WORK: 2024.1.8 --- tools/cru-py/cru/service/_template.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) (limited to 'tools/cru-py/cru/service/_template.py') diff --git a/tools/cru-py/cru/service/_template.py b/tools/cru-py/cru/service/_template.py index 5f0252a..23bff4d 100644 --- a/tools/cru-py/cru/service/_template.py +++ b/tools/cru-py/cru/service/_template.py @@ -1,4 +1,4 @@ -from argparse import ArgumentParser, Namespace +from argparse import Namespace from cru import CruIterator from cru.template import TemplateTree @@ -11,6 +11,8 @@ class TemplateManager(AppCommandFeatureProvider): def __init__(self, prefix: str = OWNER_NAME.upper()): super().__init__("template-manager") self._prefix = prefix + + def setup(self) -> None: self._templates_dir = self.app.add_path("templates", True) self._generated_dir = self.app.add_path("generated", True) self._template_tree: TemplateTree | None = None @@ -56,7 +58,10 @@ class TemplateManager(AppCommandFeatureProvider): self.generated_dir.full_path_str, config_manager.config_map ) - def add_arg_parser(self, arg_parser: ArgumentParser) -> None: + def get_command_info(self): + return ("template", "Template Management") + + def setup_arg_parser(self, arg_parser): subparsers = arg_parser.add_subparsers(dest="template_command") _list_parser = subparsers.add_parser("list", help="List templates.") _generate_parser = subparsers.add_parser("generate", help="Generate template.") -- cgit v1.2.3