
import random

import zhconv
from tqdm import tqdm
from zhconv import convert


def random_name():
    # 删减部分小众姓氏
    # firstName = "赵钱孙李周吴郑王冯陈褚卫蒋沈韩杨朱秦尤许何吕施张孔曹严华金魏陶姜戚谢邹喻水云苏潘葛奚范彭郎鲁韦昌马苗凤花方俞任袁柳鲍史唐费岑薛雷贺倪汤滕殷罗毕郝邬安常乐于时傅卞齐康伍余元卜顾孟平" \
    #             "黄和穆萧尹姚邵湛汪祁毛禹狄米贝明臧计成戴宋茅庞熊纪舒屈项祝董粱杜阮席季麻强贾路娄危江童颜郭梅盛林刁钟徐邱骆高夏蔡田胡凌霍万柯卢莫房缪干解应宗丁宣邓郁单杭洪包诸左石崔吉" \
    #             "龚程邢滑裴陆荣翁荀羊甄家封芮储靳邴松井富乌焦巴弓牧隗山谷车侯伊宁仇祖武符刘景詹束龙叶幸司韶黎乔苍双闻莘劳逄姬冉宰桂牛寿通边燕冀尚农温庄晏瞿茹习鱼容向古戈终居衡步都耿满弘国文东殴沃曾关红游盖益桓公晋楚闫"
    # 百家姓姓氏
    firstName = "赵钱孙李周吴郑王冯陈褚卫蒋沈韩杨朱秦尤许何吕施张孔曹严华金魏陶姜戚谢邹喻柏水窦章云苏潘葛奚范彭郎鲁韦昌马苗凤花方俞任袁柳酆鲍史唐费廉岑薛雷贺倪汤滕殷罗毕郝邬安常乐于时傅皮卞齐康伍余元卜顾孟平" \
                "黄和穆萧尹姚邵湛汪祁毛禹狄米贝明臧计伏成戴谈宋茅庞熊纪舒屈项祝董粱杜阮蓝闵席季麻强贾路娄危江童颜郭梅盛林刁钟徐邱骆高夏蔡田樊胡凌霍虞万支柯昝管卢莫经房裘缪干解应宗丁宣贲邓郁单杭洪包诸左石崔吉钮" \
                "龚程嵇邢滑裴陆荣翁荀羊於惠甄麴家封芮羿储靳汲邴糜松井段富巫乌焦巴弓牧隗山谷车侯宓蓬全郗班仰秋仲伊宫宁仇栾暴甘钭厉戎祖武符刘景詹束龙叶幸司韶郜黎蓟薄印宿白怀蒲邰从鄂索咸籍赖卓蔺屠蒙池乔阴欎胥能苍" \
                "双闻莘党翟谭贡劳逄姬申扶堵冉宰郦雍舄璩桑桂濮牛寿通边扈燕冀郏浦尚农温别庄晏柴瞿阎充慕连茹习宦艾鱼容向古易慎戈廖庾终暨居衡步都耿满弘匡国文寇广禄阙东殴殳沃利蔚越夔隆师巩厍聂晁勾敖融冷訾辛阚那简饶空" \
                "曾毋沙乜养鞠须丰巢关蒯相查後荆红游竺权逯盖益桓公晋楚闫法汝鄢涂钦归海帅缑亢况后有琴梁丘左丘商牟佘佴伯赏南宫墨哈谯笪年爱阳佟言福百家姓终"
    # 百家姓中双姓氏
    firstName2 = "万俟司马上官欧阳夏侯诸葛闻人东方赫连皇甫尉迟公羊澹台公冶宗政濮阳淳于单于太叔申屠公孙仲孙轩辕令狐钟离宇文长孙慕容鲜于闾丘司徒司空亓官司寇仉督子颛孙端木巫马公西漆雕乐正壤驷公良拓跋夹谷宰父谷梁段干百里东郭南门呼延羊舌微生梁丘左丘东门西门南宫南宫"
    # # 女孩名字
    # girl = '秀娟英华慧巧美娜静淑惠珠翠雅芝玉萍红娥玲芬芳燕彩春菊兰凤洁梅琳素云莲真环雪荣爱妹霞香月莺媛艳瑞凡佳嘉琼勤珍贞莉桂娣叶璧璐娅琦晶妍茜秋珊莎锦黛青倩婷姣婉娴瑾颖露瑶怡婵雁蓓纨仪荷丹蓉眉君琴蕊薇菁梦岚苑婕馨瑗琰韵融园艺咏卿聪澜纯毓悦昭冰爽琬茗羽希宁欣飘育滢馥筠柔竹霭凝晓欢霄枫芸菲寒伊亚宜可姬舒影荔枝思丽'
    # # 男孩名字
    # boy = '伟刚勇毅俊峰强军平保东文辉力明永健世广志义兴良海山仁波宁贵福生龙元全国胜学祥才发武新利清飞彬富顺信子杰涛昌成康星光天达安岩中茂进林有坚和彪博诚先敬震振壮会思群豪心邦承乐绍功松善厚庆磊民友裕河哲江超浩亮政谦亨奇固之轮翰朗伯宏言若鸣朋斌梁栋维启克伦翔旭鹏泽晨辰士以建家致树炎德行时泰盛雄琛钧冠策腾楠榕风航弘'
    # # 名
    # name = '中笑贝凯歌易仁器义礼智信友上都卡被好无九加电金马钰玉忠孝'
    fp = open('materials/chinese_cht_dict.txt', 'r', encoding='utf-8')
    name = []
    for line in fp.readlines():
        name.append(line.rstrip('\n'))
    fp.close()
    # 地點
    addr = ['北市', '新北市', '中市', '桃市', '南市', '彰縣', '屛縣', '竹縣', '苗縣', '南縣', '雲縣', '嘉縣', '宜縣', '花縣', '臺縣', '澎縣', '金縣', '連縣']
    # 狀態
    state = ['初發', '補發', '換發']

    addr_id = random.randint(0, 17)
    a = addr[addr_id]
    state_id = random.randint(0, 2)
    s = state[state_id]
    # 10%的机遇生成双数姓氏
    if random.choice(range(100)) > 10:
        firstName_name = firstName[random.choice(range(len(firstName)))]
    else:
        i = random.choice(range(len(firstName2)))
        firstName_name = firstName2[i:i + 2]

    sex = random.choice(range(2))
    name_1 = ""
    # 生成并返回一个名字
    if sex > 0:
        name_1 = name[random.randint(0, len(name) - 1)]
        if random.choice(range(2)) > 0:
            name_1 += name[random.randint(0, len(name) - 1)]
        return firstName_name + name_1 + "\t女" +\
               '\t' + a + '\t' + s + '\t' + './person_imgs/lisa.png'
    else:
        name_1 = name[random.randint(0, len(name) - 1)]
        if random.choice(range(2)) > 0:
            name_1 += name[random.randint(0, len(name) - 1)]
        return firstName_name + name_1 + "\t男" + '\t' + a + '\t' + s + '\t' + './person_imgs/02.webp'


fp = open('materials/name_cht.txt', 'w+', encoding='utf-8')


for i in tqdm(range(2000)):
    s1 = random_name()
    name = s1.split('\t')[0]
    temp_str = ''
    if len(name) <= 6:
        num_of_space = (2 * (6 - len(name))) / (len(name) - 1)
        for i in range(0, len(name) - 1):
            temp_str += name[i]
            for j in range(0, int(num_of_space)):
                temp_str += ' '
        temp_str += name[len(name) - 1]
        name = temp_str
    sex = s1.split('\t')[1]
    addr = s1.split('\t')[2]
    state = s1.split('\t')[3]
    img = s1.split('\t')[4]
    birth_year = random.randint(0, 100)
    birth_month = random.randint(1, 12)
    birth_day = random.randint(1, 30)
    time_year = random.randint(0, 100)
    time_month = random.randint(1, 12)
    time_day = random.randint(1, 30)
    word_index = random.randint(0, 25)
    id = random.randint(0, 1000000000)
    # 字母
    word = ['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U',
            'V', 'W', 'X', 'Y', 'Z']

    name = zhconv.convert(name, 'zh-tw')
    # print(name)
    # line = name + ',' + str(birth_year) + ',' + str(birth_month) + ',' + str(birth_day) + ',民國 ' + str(
    #     time_year) + ' 年 ' + str(time_month) + ' 月 ' + str(time_day) + ' 日(' + addr + ')' + state + ',' + sex + ',' + word[word_index] + str(id).zfill(9) + ',' + img + '\n'
    # br = "民國 " + str(birth_year) + " 年 " + str(birth_month) + ' 月 ' + str(birth_day) + ' 日'
    # generate_date = '民國 ' + str(time_year) + ' 年 ' + str(time_month) + ' 月 ' + str(time_day) + ' 日(' + addr + ')' + state
    fp.write(name + '\n')
