diff options
Diffstat (limited to 'cmd2/__init__.py')
| -rw-r--r-- | cmd2/__init__.py | 12 |
1 files changed, 11 insertions, 1 deletions
diff --git a/cmd2/__init__.py b/cmd2/__init__.py index 8e8a8845..8fc5e9f2 100644 --- a/cmd2/__init__.py +++ b/cmd2/__init__.py @@ -11,7 +11,17 @@ except DistributionNotFound: pass from .ansi import style -from .argparse_custom import Cmd2ArgumentParser, CompletionError, CompletionItem +from .argparse_custom import Cmd2ArgumentParser, CompletionError, CompletionItem, set_default_argument_parser + +# Check if user has defined a module that sets a custom value for argparse_custom.DEFAULT_ARGUMENT_PARSER +import argparse +cmd2_parser_module = getattr(argparse, 'cmd2_parser_module', None) +if cmd2_parser_module is not None: + import importlib + importlib.import_module(cmd2_parser_module) + +# Get the current value for argparse_custom.DEFAULT_ARGUMENT_PARSER +from .argparse_custom import DEFAULT_ARGUMENT_PARSER from .cmd2 import Cmd, EmptyStatement from .constants import COMMAND_NAME, DEFAULT_SHORTCUTS from .decorators import categorize, with_argument_list, with_argparser, with_argparser_and_unknown_args, with_category |