SearchDomain-Python/searchdomain/generate_en_domain.py

#!/usr/bin/env python
# -*- encoding: utf-8 -*-
'''
@Contact :   liuyuqi.gov@msn.cn
@Time    :   2023/03/08 20:18:13
@License :   Copyright © 2017-2022 liuyuqi. All Rights Reserved.
@Desc    :   批量生成域名（双拼，双单词，双单词拼接）

参考项目： https://git.yoqi.me/lyq/SearchDomain

'''

import csv,os

class GenerateEnDomain(object):
    
    def __init__(self, params: dict):
        '''初始化列表'''
        self.initList=[]
        self.params =params
        # self.keyword=["chat","ai"] # chat+xx
        
        self.keyword= self.params['keyword'].split(',')
        # self.yuming=["com","cn","me","net","co","run","wiki","tech","org","info","vip","cc","app","io","one","tk","xyz"]
        self.yuming=self.params["domain"].split(',')
        self.composeDomain=[]
        self.composePinYin=[]

    def run(self):
        """ 批量生成域名, 保存到 domain.txt 或指定的输出文件 """
        output_file = self.params.get("output", "domain.txt")
        with open(os.path.join(self.params["app_path"], r"data/b.csv"), "r", encoding="utf-8") as f:
            csv_data = f.readlines()
            for row in csv_data:
                self.initList.append(row.strip())
            
            for i in self.initList:
                for j in self.keyword:
                    if self.params['position']=="prefix":
                        self.composePinYin.append(i+j)
                    else:
                        self.composePinYin.append(j+i)

            for i in self.composePinYin:
                for j in self.yuming:
                    self.composeDomain.append(i+"."+j)

            output_path = output_file if os.path.isabs(output_file) else os.path.join(self.params["app_path"], output_file)
            with open(output_path, "w", encoding="utf-8") as file:
                for i in self.composeDomain:
                    file.write(i+"\n")