panleicim/appointment_tool
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

optimizations

Browse Source
This commit is contained in:
Lei PAN
2022-07-08 15:19:30 +02:00
parent cdabe04073
commit 904f15c151
2 changed files with 11 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/check_results.py
+6 -7
View File
@@ -94,12 +94,11 @@ class ResultChecker:
else: else:
print("status is ACCEPTED") print("status is ACCEPTED")
status = ResultEnum.ACCEPTED status = ResultEnum.ACCEPTED
if need_send_email: # send email
# send email try:
try: mailer.send_email(get_accepted_result_from(reserve_pojo), to_all=need_send_email)
mailer.send_email(get_accepted_result_from(reserve_pojo)) except Exception as err:
except Exception as err: print(err)
print(err)
reserve_pojo.accepted = status reserve_pojo.accepted = status
firestore_collection.document(reserve_pojo.id).update({u'accepted': status.name}) firestore_collection.document(reserve_pojo.id).update({u'accepted': status.name})
MONGO_STORE_MANAGER.update_reserve_result(reserve_pojo.id, status) MONGO_STORE_MANAGER.update_reserve_result(reserve_pojo.id, status)
@@ -114,7 +113,7 @@ def check_results(headless=False):
print("size is " + str(len(reserve_list))) print("size is " + str(len(reserve_list)))
start_check(reserve_list, firestore_collection, headless, need_send_email=False) start_check(reserve_list, firestore_collection, headless, need_send_email=False)
reserve_list = MONGO_STORE_MANAGER.get_all_successful_items_for_day() reserve_list = MONGO_STORE_MANAGER.get_all_successful_items_for_day()
start_check(reserve_list, firestore_collection, headless, need_send_email=True) # start_check(reserve_list, firestore_collection, headless, need_send_email=True)
def start_check(reserve_list, firestore_collection, headless: bool, need_send_email: bool): def start_check(reserve_list, firestore_collection, headless: bool, need_send_email: bool):
src/notification/mailer.py
+5 -3
View File
@@ -25,9 +25,11 @@ class Mailer:
aws_access_key_id="AKIAU67WLKZUXEHXDJB5", aws_access_key_id="AKIAU67WLKZUXEHXDJB5",
aws_secret_access_key=secret) aws_secret_access_key=secret)
def send_email(self, result: AcceptedResultPojo): def send_email(self, result: AcceptedResultPojo, to_all: bool = False):
recipients = ['panleicim@gmail.com', 'kamenonly@gmail.com', 'tangliang0411@gmail.com'] if to_all:
# recipients = ['panleicim@gmail.com'] recipients = ['panleicim@gmail.com', 'kamenonly@gmail.com', 'tangliang0411@gmail.com']
else:
recipients = ['panleicim@gmail.com']
mytemplate = Template(filename=config.ROOT_DIR + "/templates/appointment_results.html") mytemplate = Template(filename=config.ROOT_DIR + "/templates/appointment_results.html")
self.logger.info("send email to " + str(recipients)) self.logger.info("send email to " + str(recipients))