diff options
Diffstat (limited to 'examples/basic_completion.py')
| -rwxr-xr-x | examples/basic_completion.py | 55 |
1 files changed, 25 insertions, 30 deletions
diff --git a/examples/basic_completion.py b/examples/basic_completion.py index f33029c9..aab7d3d1 100755 --- a/examples/basic_completion.py +++ b/examples/basic_completion.py @@ -22,14 +22,13 @@ food_item_strs = ['Pizza', 'Ham', 'Ham Sandwich', 'Potato'] sport_item_strs = ['Bat', 'Basket', 'Basketball', 'Football', 'Space Ball'] # This data is used to demonstrate delimiter_complete -file_strs = \ - [ - '/home/user/file.db', - '/home/user/file space.db', - '/home/user/another.db', - '/home/other user/maps.db', - '/home/other user/tests.db' - ] +file_strs = [ + '/home/user/file.db', + '/home/user/file space.db', + '/home/user/another.db', + '/home/other user/maps.db', + '/home/other user/tests.db', +] class BasicCompletion(cmd2.Cmd): @@ -46,20 +45,17 @@ class BasicCompletion(cmd2.Cmd): def complete_flag_based(self, text, line, begidx, endidx) -> List[str]: """Completion function for do_flag_based""" - flag_dict = \ - { - # Tab complete food items after -f and --food flags in command line - '-f': food_item_strs, - '--food': food_item_strs, - - # Tab complete sport items after -s and --sport flags in command line - '-s': sport_item_strs, - '--sport': sport_item_strs, - - # Tab complete using path_complete function after -p and --path flags in command line - '-p': self.path_complete, - '--path': self.path_complete, - } + flag_dict = { + # Tab complete food items after -f and --food flags in command line + '-f': food_item_strs, + '--food': food_item_strs, + # Tab complete sport items after -s and --sport flags in command line + '-s': sport_item_strs, + '--sport': sport_item_strs, + # Tab complete using path_complete function after -p and --path flags in command line + '-p': self.path_complete, + '--path': self.path_complete, + } return self.flag_based_complete(text, line, begidx, endidx, flag_dict=flag_dict) @@ -69,12 +65,11 @@ class BasicCompletion(cmd2.Cmd): def complete_index_based(self, text, line, begidx, endidx) -> List[str]: """Completion function for do_index_based""" - index_dict = \ - { - 1: food_item_strs, # Tab complete food items at index 1 in command line - 2: sport_item_strs, # Tab complete sport items at index 2 in command line - 3: self.path_complete, # Tab complete using path_complete function at index 3 in command line - } + index_dict = { + 1: food_item_strs, # Tab complete food items at index 1 in command line + 2: sport_item_strs, # Tab complete sport items at index 2 in command line + 3: self.path_complete, # Tab complete using path_complete function at index 3 in command line + } return self.index_based_complete(text, line, begidx, endidx, index_dict=index_dict) @@ -83,8 +78,7 @@ class BasicCompletion(cmd2.Cmd): self.poutput("Args: {}".format(statement.args)) # Use a partialmethod to set arguments to delimiter_complete - complete_delimiter_complete = functools.partialmethod(cmd2.Cmd.delimiter_complete, - match_against=file_strs, delimiter='/') + complete_delimiter_complete = functools.partialmethod(cmd2.Cmd.delimiter_complete, match_against=file_strs, delimiter='/') def do_raise_error(self, statement: cmd2.Statement): """Demonstrates effect of raising CompletionError""" @@ -103,5 +97,6 @@ class BasicCompletion(cmd2.Cmd): if __name__ == '__main__': import sys + app = BasicCompletion() sys.exit(app.cmdloop()) |