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
searchdomain generate --lang zh --keyword gpt,go --position prefix --domain io,com,cn
searchdomain search --
'''

from searchdomain import SearchDomain, GenerateDomain, GenerateEnDomain
from searchdomain.options import parse_args
import sys

if __name__== "__main__":
    args = parse_args()
    if args.get("command","")=="generate":
        if args.get("lang","en")=="en":
            generateDomain = GenerateEnDomain(params=args) 
        elif args.get("lang", "en")=="zh":
            generateDomain = GenerateDomain(params=args)
        generateDomain.run()
    elif args.get("command","")=="search":
        serachdomain = SearchDomain(params=args, debug=True, export_all=True) 
        serachdomain.run()
    elif args.get("command","")=="version":
        # print(__verison__)
        pass
    else:
        print("please input command: generate, search")
        print(args.get('command',""))
        sys.exit(1)