diff options
Diffstat (limited to 'examples/table_display.py')
-rwxr-xr-x | examples/table_display.py | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/examples/table_display.py b/examples/table_display.py index a5a5c830..a8fd2cb0 100755 --- a/examples/table_display.py +++ b/examples/table_display.py @@ -14,9 +14,11 @@ and either the colored or colorama module """ from typing import Tuple -import cmd2 import tableformatter as tf +import cmd2 +from cmd2 import Cmd2ArgumentParser + # Configure colors for when users chooses the "-c" flag to enable color in the table output try: from colored import bg @@ -142,14 +144,14 @@ def high_density_objs(row_obj: CityInfo) -> dict: return opts -def make_table_parser() -> cmd2.ArgParser: +def make_table_parser() -> Cmd2ArgumentParser: """Create a unique instance of an argparse Argument parser for processing table arguments. NOTE: The two cmd2 argparse decorators require that each parser be unique, even if they are essentially a deep copy of each other. For cases like that, you can create a function to return a unique instance of a parser, which is what is being done here. """ - table_parser = cmd2.ArgParser() + table_parser = Cmd2ArgumentParser() table_item_group = table_parser.add_mutually_exclusive_group() table_item_group.add_argument('-c', '--color', action='store_true', help='Enable color') table_item_group.add_argument('-f', '--fancy', action='store_true', help='Fancy Grid') |