correction on check_results.py

This commit is contained in:
Lei PAN
2022-07-06 00:29:45 +02:00
parent f7168d92cc
commit e9150c885d
2 changed files with 11 additions and 7 deletions
+2 -7
View File
@@ -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,
+9
View File
@@ -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]