# 批量生成域名（双拼，双单词，双单词拼接）

import os,sys,re,json
from urllib.parse import unquote, urlencode

def generate(domain, kw_first, kw_second, kw_third, word_length):
    # 获取域名和目录
    domain_dir = f'{os.getcwd()}/domains/'
    if not os.path.exists(domain_dir):
        os.makedirs(domain_dir)

    # 域名拼接
    domain = f'{domain}.'

    # 域名头部
    domain_head = f'{domain}'
    # 域名尾部
    domain_tail = f'{domain}'
    # 域名双拼
    domain_double_p = f'{domain}'
    # 域名双单词
    domain_double_s = f'{domain}'

    with open(("searchdomain/searchdomain.json"), "r", encoding="utf-8") as f:
        domain_dict = json.load(f)
        for key in domain_dict:
            if domain_dict[key]['length'] == word_length:
                domain_double_p = domain_dict[key]['first']
                domain_double_s = domain_dict[key]['second']
                domain_double_p = domain_double_p[:word_length]
                domain_double_s = domain_double_s[:word_length]
                break
        f.write(buffer)

if __name__ == "__main__":
    generate(domain, kw_first, kw_second, kw_third, word_length)