main.py

from config import name_source, last_name, dislike_words, \
    min_stroke_count, max_stroke_count, allow_general, name_validate, gender, \
    check_name, check_name_resource
from name_set import check_resource, get_source
from wuge import check_wuge_config, get_stroke_list


def contain_bad_word(first_name):
    for word in first_name:
        if word in dislike_words:
            return True
    return False


if len(check_name) == 3:
    # 查看姓名配置
    check_wuge_config(check_name)
    if check_name_resource:
        check_resource(check_name)
    print(">>输出完毕")
else:
    # 起名
    names = list()
    with open("names.txt", "w+", encoding='utf-8') as f:
        for i in get_source(name_source, name_validate, get_stroke_list(last_name, allow_general)):
            if i.stroke_number1 < min_stroke_count or i.stroke_number1 > max_stroke_count or \
                    i.stroke_number2 < min_stroke_count or i.stroke_number2 > max_stroke_count:
                # 笔画数过滤
                continue
            if name_validate and gender != "" and i.gender != gender and i.gender != "双" and i.gender != "未知":
                # 性别过滤
                continue
            if contain_bad_word(i.first_name):
                # 不喜欢字过滤
                continue
            names.append(i)
        print(">>输出结果...")
        names.sort()
        for i in names:
            f.write(last_name + str(i) + "\n")
        print(">>输出完毕，请查看「names.txt」文件")