support file list

request_sender_test.py

@@ -61,30 +61,39 @@ def send_appointment_request(message_queue_name, _contact_list):
     _cookiesPublisher = CookiesPublisher(queue_name=message_queue_name)
     _cookiesPublisher.set_up_connection()
     receiver = AppointmentRequestSender(sub_contact_list=_contact_list,
-                                        queue_name=message_queue_name, just_send=True,
+                                        queue_name=message_queue_name,
                                         cookiesPublisher=_cookiesPublisher, logger=logger)
     print("count is " + str(count))
     receiver.run()
 
 
-def start_send_requests(thread_number, file_path, data_queue_name=MORNING_DATA_CACHE):
+def start_send_requests(thread_number, contact_list, data_queue_name=MORNING_DATA_CACHE):
     print("start send requests")
-    contacts_file_path = file_path
-    _contact_list = read_contacts(contacts_file_path)
-    _contact_list_to_book = filter_contacts(_contact_list)
+    _contact_list_to_book = filter_contacts(contact_list)
     _segment_number = thread_number
     logger.info("{} contacts to book".format(len(_contact_list_to_book)))
-    last_thread = None
+    # last_thread = None
+    thread_list = []
     for i in range(0, _segment_number):
         logger.info("segment is {}".format(i))
         _step = int(len(_contact_list_to_book) / _segment_number)
         _sublist = _contact_list_to_book[i * _step:_step * (i + 1)]
         _thread1 = Thread(target=send_appointment_request, args=(data_queue_name, _sublist))
-        last_thread = _thread1
+        thread_list.append(_thread1)
         _thread1.start()
-    last_thread.join()
+    for _thread in thread_list:
+        _thread.join()
+
+
+def send_request_for_file_list(_file_list: list[str], thread_number: int = 20, data_queue_name=MORNING_DATA_CACHE):
+    for _file_path in file_list:
+        logger.info("send request for file: " + _file_path)
+        _contact_list = read_contacts(_file_path)
+        start_send_requests(thread_number=thread_number, contact_list=_contact_list,
+                            data_queue_name=data_queue_name)
 
 
 if __name__ == '__main__':
-    start_send_requests(thread_number=30, file_path='~/Desktop/contact_list_2024-05-14.xlsx',
-                        data_queue_name=MORNING_DATA_CACHE_2)
+    file_list = ['~/Desktop/16_05_to_test.xlsx']
+    send_request_for_file_list(_file_list=file_list, thread_number=20,
+                               data_queue_name=MORNING_DATA_CACHE_2)