can pass segment_number to generate contact list

This commit is contained in:
2024-04-16 19:53:07 +02:00
parent a9da11fe33
commit 118f29d4ed
+17 -13
View File
@@ -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:
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")