import argparse def parse_args(): parser = argparse.ArgumentParser(description='search domain') parser.add_argument('command', help='command: generate, search', choices=['generate','search'] , default='generate') parser.add_argument('--export_all', action='store_true', help='export all domain') return parser.parse_args() if __name__=='__main__': args = parse_args() print(args)