SearchDomain-Python/test/args_test.py

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)