SearchDomain-Python/main.py

#!/usr/bin/env python
# -*- encoding: utf-8 -*-
'''
@Contact :   liuyuqi.gov@msn.cn
@Time    :   2023/03/08 17:39:34
@License :   Copyright © 2017-2022 liuyuqi. All Rights Reserved.
@Desc    :   enter point
'''

from searchdomain import SearchDomain, GenerateDomain, GenerateEnDomain
import sys
import argparse

def parse_args():
    parser = argparse.ArgumentParser(description='search domain')
    parser.add_argument('--command', action='store_true', help='command: generate, search, generate_en', choices=['generate', 'search', 'generate_en'])
    parser.add_argument('--export_all', action='store_true', help='export all domain')
    return parser.parse_args()

if __name__== "__main__":
    args = parse_args()

    if args.command=="generate":
        generateDomain = GenerateDomain() 
        generateDomain.run()
    elif args.command=="generate_en":
        generateDomain = GenerateEnDomain() 
        generateDomain.run()
    elif args.command=="search":
        serachdomain = SearchDomain(debug=True, export_all=False) 
        serachdomain.run()
    else:
        print("please input command: generate, search, generate_en")
        sys.exit(1)