Example added demonstrating bash completion bridging argcomplete with AutoCompleter

1 files changed, 81 insertions, 0 deletions

diff --git a/examples/bash_completion.py b/examples/bash_completion.py
new file mode 100755
index 00000000..666d3a0c
--- /dev/null
+++ b/examples/bash_completion.py
@@ -0,0 +1,81 @@
+#!/usr/bin/env python3
+# coding=utf-8
+# PYTHON_ARGCOMPLETE_OK - This is required at the beginning of the file to enable argcomplete support
+"""A simple example demonstrating integration with argcomplete"""
+import argparse
+
+optional_strs = ['Apple', 'Banana', 'Cranberry', 'Durian', 'Elderberry']
+
+bash_parser = argparse.ArgumentParser(prog='base')
+
+bash_parser.add_argument('option', choices=['load', 'export', 'reload'])
+
+bash_parser.add_argument('-u', '--user', help='User name')
+bash_parser.add_argument('-p', '--passwd', help='Password')
+
+input_file = bash_parser.add_argument('-f', '--file', type=str, help='Input File')
+
+if __name__ == '__main__':
+    from cmd2.argcomplete_bridge import bash_complete
+    # bash_complete flags this argument telling AutoCompleter to yield to bash to perform
+    # tab completion of a file path
+    bash_complete(input_file)
+
+flag_opt = bash_parser.add_argument('-o', '--optional', help='Optional flag with choices')
+setattr(flag_opt, 'arg_choices', optional_strs)
+
+# Handle bash completion if it's installed
+# This early check allows the script to bail out early to provide tab-completion results
+# to the argcomplete library. Putting this at the end of the file would cause the full application
+# to load fulfill every tab-completion request coming from bash.  This can cause a notable delay
+# on the bash prompt.
+try:
+    # only move forward if we can import CompletionFinder and AutoCompleter
+    from cmd2.argcomplete_bridge import CompletionFinder
+    from cmd2.argparse_completer import AutoCompleter
+    import sys
+    if __name__ == '__main__':
+        completer = CompletionFinder()
+
+        # completer will return results to argcomplete and exit the script
+        completer(bash_parser, AutoCompleter(bash_parser))
+except ImportError:
+    pass
+
+# Intentionally below the bash completion code to reduce tab completion lag
+import cmd2
+
+
+class DummyApp(cmd2.Cmd):
+    """
+    Dummy cmd2 app
+    """
+
+    def __init__(self):
+        super().__init__()
+
+
+if __name__ == '__main__':
+    args = bash_parser.parse_args()
+
+    # demonstrates some handling of the command line parameters
+
+    if args.user is None:
+        user = input('Username: ')
+    else:
+        user = args.user
+
+    if args.passwd is None:
+        import getpass
+        password = getpass.getpass()
+    else:
+        password = args.passwd
+
+    if args.file is not None:
+        print('Loading file: {}'.format(args.file))
+
+    # Clear the argumentns so cmd2 doesn't try to parse them
+    sys.argv = sys.argv[:1]
+
+    app = DummyApp()
+    app.cmdloop()