Merge pull request #21 from PhonePe/bash_completion

Added support for generating shell completion file

Author: santanusinha

drove.py

@@ -18,14 +18,15 @@ def build_parser() -> argparse.ArgumentParser:
     parser.add_argument("--username", "-u", help="Drove cluster username")
     parser.add_argument("--password", "-p", help="Drove cluster password")
     parser.add_argument("--debug", "-d", help="Print details of errors", default=False, action="store_true")
+    parser.add_argument("--print-completion", choices=["bash", "zsh", "tcsh"], help="Print shell completion script for the given shell")
     return parser
 
 def get_parser():
     parser = build_parser()
     client = None
     client = drovecli.DroveCli(parser)
     return client.parser
- 
+
 
 def run():
     parser = build_parser()

drovecli.py

@@ -2,7 +2,7 @@
 import droveclient
 from plugins import DrovePlugin
 from types import SimpleNamespace
-
+import shtab
 
 
 class DroveCli:
@@ -24,6 +24,10 @@ def run(self):
         args = self.parser.parse_args()
         self.debug = args.debug
 
+        if args.print_completion:
+            print(shtab.complete(self.parser, shell=args.print_completion))
+            exit(0)
+
         # Load plugins
         if args.debug:
             print("Selected plugin: " + args.plugin)

requirements.txt

@@ -2,3 +2,4 @@ Requests==2.32.5
 tabulate==0.9.0
 tenacity==9.1.2
 urllib3==2.5.0
+shtab==1.8.0