diff --git a/src/person_name/contact_manager.py b/src/person_name/contact_manager.py index 302d2fc..0980d4e 100755 --- a/src/person_name/contact_manager.py +++ b/src/person_name/contact_manager.py @@ -8,7 +8,7 @@ from src.utils.excel_reader import read_contacts, fr_phone_number_prefix, get_ra def upload_contacts_list(): - _contacts_to_book = read_contacts("/Users/panlei/Desktop/contact_list_2024-03-30.xlsx") + _contacts_to_book = read_contacts("/Users/lpan/Desktop/contact_list_2024-04-15.xlsx") return _contacts_to_book @@ -19,7 +19,7 @@ def fix_phone_number_format(file_path): if _contact.phone[0:2] not in fr_phone_number_prefix: print(_contact) _contact.phone = get_random_fr_phone_numbers() - write_new_contacts_to_excel(_contact_list, file_name="25_03_to_test") + write_new_contacts_to_excel(_contact_list, file_name="09_04_to_test") def write_new_contacts_to_excel(valid_contacts: list, file_name=str(datetime.date.today())): @@ -46,7 +46,7 @@ def write_new_contacts_to_excel(valid_contacts: list, file_name=str(datetime.dat workbook.close() -def generate_valid_contact_list_for_day(): +def generate_valid_contact_list_for_day(sement_number = 1): _valid_contact_list = MONGO_STORE_MANAGER.get_all_successful_items_for_day() _all_contacts = MONGO_STORE_MANAGER.get_all_contacts_to_book() _contact_to_save = [] @@ -62,8 +62,15 @@ def generate_valid_contact_list_for_day(): _contact.first_name = _true_contact.first_name if _contact.url_validated: - _contact_to_save.append(_contact) + if _contact.last_name is not None and len(_contact.last_name) > 0: + _contact_to_save.append(_contact) write_new_contacts_to_excel(_contact_to_save) + for i in range(0, sement_number): + _step = int(len(_contact_to_save) / sement_number) + _sublist = _contact_to_save[i * _step:_step * (i + 1)] + _file_name = str(datetime.date.today()) +"_"+ str(i+1) + write_new_contacts_to_excel(_sublist,file_name=_file_name) + def merge_contact_list_files(file_list: list, final_file_name="merged_contact_list"): @@ -90,14 +97,11 @@ def generate_all_contact_list(): if __name__ == '__main__': # contacts_to_book = upload_contacts_list() # MONGO_STORE_MANAGER.upload_contact_list(contacts_to_book) - # generate_valid_contact_list_for_day() + generate_valid_contact_list_for_day(sement_number=3) # generate_all_contact_list() - merge_contact_list_files( - ["/Users/panlei/Desktop/contact_list_2024-03-29.xlsx", - "/Users/panlei/Desktop/contact_list_2024-03-27.xlsx", - "/Users/panlei/Desktop/contact_list_2024-03-28.xlsx", - "/Users/panlei/Desktop/contact_list_2024-03-26.xlsx", - "/Users/panlei/Desktop/contact_list_2024-03-25.xlsx", - "/Users/panlei/Desktop/contact_list_2024-03-23.xlsx", - "/Users/panlei/Desktop/contact_list_2024-03-22.xlsx"]) - # fix_phone_number_format("/Users/panlei/Desktop/25_03_to_test.xlsx") + # merge_contact_list_files( + # ["/Users/lpan/Desktop/contact_list_2024-04-10.xlsx", + # "/Users/lpan/Desktop/contact_list_2024-04-09.xlsx", + # "/Users/lpan/Desktop/contact_list_2024-04-06.xlsx", + # "/Users/lpan/Desktop/contact_list_2024-04-04.xlsx"]) + # fix_phone_number_format("/Users/lpan/Desktop/09_04_to_test.xlsx")