janware/jw-python
1
mirror of ssh://git.janware.com/srv/git/janware/proj/jw-python synced 2026-01-15 09:53:32 +01:00
Code Releases Activity

Cmd.add_subcommand(): Add support for passing cmd argument as string

Browse source 
Signed-off-by: Jan Lindemann <jan@janware.com>
This commit is contained in:
Jan Lindemann 2024-12-03 13:34:25 +01:00
commit 4b8d67b8ee
1 changed files with 12 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

14
tools/python/jwutils/Cmd.py
View file

@ -1,9 +1,10 @@
from __future__ import annotations from __future__ import annotations
from typing import List, Type, Union, TypeVar
import abc import abc
import argparse import argparse
from abc import ABC from abc import ABC
from argparse import ArgumentParser, _SubParsersAction from argparse import ArgumentParser, _SubParsersAction
from typing import List, Type, Union, TypeVar import inspect, sys, re
from jwutils import log from jwutils import log
@ -32,7 +33,16 @@ class Cmd(ABC): # export
r.set_defaults(func=self.run) r.set_defaults(func=self.run)
return r return r
def add_subcommands(self, cmd: Union[Type[Cmd], List[Type[Cmd]]]) -> None: def add_subcommands(self, cmd: Union[str, Type[Cmd], List[Type[Cmd]]]) -> None:
if isinstance(cmd, str):
sc = []
for name, obj in inspect.getmembers(sys.modules[self.__class__.__module__]):
if inspect.isclass(obj):
if re.search(cmd, str(obj)):
sc.append(obj)
log.slog(log.DEBUG, f"Found subcommand {obj}")
self.add_subcommands(sc)
return
if isinstance(cmd, list): if isinstance(cmd, list):
for c in cmd: for c in cmd:
self.add_subcommands(c) self.add_subcommands(c)