diff --git a/src/person_name/extract_name_with_pinyinlist.py b/src/person_name/extract_name_with_pinyinlist.py index 0b29794..feb3a7e 100644 --- a/src/person_name/extract_name_with_pinyinlist.py +++ b/src/person_name/extract_name_with_pinyinlist.py @@ -1,16 +1,13 @@ import itertools +import random import xlsxwriter -from pyhanlp import * -from Pinyin2Hanzi import DefaultHmmParams -from Pinyin2Hanzi import viterbi -from itertools import combinations from src.db.mongo_manager import MongoDbManager from src.pojo.contact_pojo import ContactPojo -from src.utils.excel_reader import get_random_phone_numbers, generate_email_from_name +from src.utils.excel_reader import get_random_phone_numbers from src.utils.generate_random_passport_id import get_random_passport_id_number -import random + def get_ordered_combins(stuff): list_to_return = [] @@ -93,8 +90,8 @@ def write_new_contacts_to_excel(valid_contacts: list): if __name__ == '__main__': db_manager = MongoDbManager() - mail_list = db_manager.get_destination_emails()[2501:2700] - # mail_list = db_manager.get_destination_emails()[6800:7000] + mail_list = db_manager.get_destination_emails()[0:100] + # mail_list = db_manager.get_destination_emails()[3001:3200] # mail_list = db_manager.get_destination_emails()[50:200] generate_contacts = [] pinyin_name_list = read_pinyin_list_from_file()