diff options
Diffstat (limited to 'cmd2')
| -rw-r--r-- | cmd2/argparse_completer.py | 2 | ||||
| -rw-r--r-- | cmd2/argparse_custom.py | 11 |
2 files changed, 12 insertions, 1 deletions
diff --git a/cmd2/argparse_completer.py b/cmd2/argparse_completer.py index f31584e7..c00f3e60 100644 --- a/cmd2/argparse_completer.py +++ b/cmd2/argparse_completer.py @@ -30,7 +30,7 @@ from .constants import ( INFINITY, ) -if TYPE_CHECKING: +if TYPE_CHECKING: # pragma: no cover from .cmd2 import ( Cmd, ) diff --git a/cmd2/argparse_custom.py b/cmd2/argparse_custom.py index dd4db570..b33dd95c 100644 --- a/cmd2/argparse_custom.py +++ b/cmd2/argparse_custom.py @@ -1252,7 +1252,16 @@ class Cmd2ArgumentParser(argparse.ArgumentParser): conflict_handler: str = 'error', add_help: bool = True, allow_abbrev: bool = True, + *, + ap_completer_type: Optional[Type['ArgparseCompleter']] = None, ) -> None: + """ + # Custom parameter added by cmd2 + + :param ap_completer_type: optional parameter which specifies a subclass of ArgparseCompleter for custom tab completion + behavior on this parser. If this is None or not present, then cmd2 will use + argparse_completer.DEFAULT_AP_COMPLETER when tab completing this parser's arguments + """ super(Cmd2ArgumentParser, self).__init__( prog=prog, usage=usage, @@ -1268,6 +1277,8 @@ class Cmd2ArgumentParser(argparse.ArgumentParser): allow_abbrev=allow_abbrev, ) + self.set_ap_completer_type(ap_completer_type) # type: ignore[attr-defined] + # noinspection PyProtectedMember def add_subparsers(self, **kwargs: Any) -> argparse._SubParsersAction: """ |