diff options
Diffstat (limited to 'tools/cru-py/cru/service/_template.py')
| -rw-r--r-- | tools/cru-py/cru/service/_template.py | 9 |
1 files changed, 7 insertions, 2 deletions
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.") |