From 872da20feba57f42dde204da01dc48c4c87e1b54 Mon Sep 17 00:00:00 2001 From: Eric Lin Date: Thu, 10 Sep 2020 09:15:05 -0400 Subject: Changes default category to be heritable by default - meaning that subclasses will inherit the parent class's default category. Adds optional flag to disable heritability. --- cmd2/utils.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) (limited to 'cmd2/utils.py') diff --git a/cmd2/utils.py b/cmd2/utils.py index a2b1c854..d396fb6a 100644 --- a/cmd2/utils.py +++ b/cmd2/utils.py @@ -1038,7 +1038,10 @@ def categorize(func: Union[Callable, Iterable[Callable]], category: str) -> None for item in func: setattr(item, constants.CMD_ATTR_HELP_CATEGORY, category) else: - setattr(func, constants.CMD_ATTR_HELP_CATEGORY, category) + if inspect.ismethod(func): + setattr(func.__func__, constants.CMD_ATTR_HELP_CATEGORY, category) + else: + setattr(func, constants.CMD_ATTR_HELP_CATEGORY, category) def get_defining_class(meth): -- cgit v1.2.1