diff --git a/check_results.py b/check_results.py index 4d07475..1fa5c47 100644 --- a/check_results.py +++ b/check_results.py @@ -108,14 +108,8 @@ def check_results(): firestore_collection = db_manager.get_all_successful_items() reserve_list = params.mongo_store_manager.get_all_successful_items_for_day() print("size is " + str(len(reserve_list))) - for result in reserve_list: - print(result) - # result_list = [] - # for appointment in collection.stream(): - # reserve_pojo = ReserveResultPojo.from_firestore_dict(appointment.to_dict()) - # result_list.append(reserve_pojo) - start_check(reserve_list, firestore_collection, False) + reserve_list = params.mongo_store_manager.get_all_successful_items_for_day() start_check(reserve_list, firestore_collection, True) @@ -125,6 +119,7 @@ def start_check(reserve_list, firestore_collection, need_send_email: bool): for reserve in reserve_list: count = count + 1 if reserve.accepted is None or ResultEnum.ACCEPTED.value == reserve.accepted: + print("will check result") if reserve.url != BLANK_URL: if reserve.url != WELCOME_URL: executor.submit(ResultChecker().run, reserve, firestore_collection, diff --git a/db/mongo_manager.py b/db/mongo_manager.py index 2cfb13f..9222a2c 100644 --- a/db/mongo_manager.py +++ b/db/mongo_manager.py @@ -49,6 +49,15 @@ class MongoDbManager: result_list.append(ReserveResultPojo.from_firestore_dict(document)) return result_list + def get_all_successful_items_for_one_day(self, day_in_str: str) -> list: + params.oracle_log_sender.send_read_db_event("get_all_successful_items_for_one_day") + collection_name = day_in_str + result_list = [] + cursor = self.db[collection_name] + for document in cursor.find(): + result_list.append(ReserveResultPojo.from_firestore_dict(document)) + return result_list + def update_reserve_result(self, reserve_id: str, result: ResultEnum): collection_name = str(datetime.date.today()) collection = self.db[collection_name]