shtab

import argparse import shtab import os def get_main_parser(): parser = argparse.ArgumentParser(prog='mycli', description='A simple CLI tool') parser.add_argument('--auth-key', type=str, default=os.environ.get('MYCLI_AUTH_KEY', ''), help='Authentication key') parser.add_argument('--output', '-o', choices=['json', 'yaml', 'text'], default='text', help='Output format') parser.add_argument('command', choices=['status', 'run', 'config'], help='Command to execute') return parser if __name__ == '__main__': parser = get_main_parser() # Integrate shtab for completion. This creates a hidden argument like --print-completion shtab.add_argument(parser, shells=['bash', 'zsh', 'tcsh']) args = parser.parse_args() # Example of how a user would generate and install completion: # mycli --print-completion bash > ~/.bashrc.d/mycli.bash-completion # source ~/.bashrc.d/mycli.bash-completion print(f"Running command: {args.command}") if args.auth_key: print("Auth key provided.") else: print("No auth key provided.") print(f"Output format: {args.output}")