Parcourir la source

修改資料庫連線,每次重連,避免斷線

Mike il y a 3 ans
Parent
commit
45480da057
1 fichiers modifiés avec 19 ajouts et 8 suppressions
  1. 19 8
      is_close_changed_notice.py

+ 19 - 8
is_close_changed_notice.py

@@ -38,6 +38,21 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
+
+# 取得資料庫連線
+def get_db_connection():
+    # Choozmo DB
+    # db = dataset.connect("mysql://choozmo:pAssw0rd@db.ptt.cx:3306/hhh?charset=utf8mb4")
+
+    # 測試機 DB
+    db = dataset.connect('mysql://hhh7796hhh:lYmWsu^ujcA1@hhh-v57-cluster.cluster-cmab1ctkglka.ap-northeast-2.rds.amazonaws.com:3306/stage?charset=utf8mb4')
+
+    # 正式機 DB
+    # db = dataset.connect('mysql://hhh7796hhh:lYmWsu^ujcA1@hhh-v57-cluster.cluster-cmab1ctkglka.ap-northeast-2.rds.amazonaws.com:3306/xoops?charset=utf8mb4')
+
+    return db
+
+
 # 寄送逾期執行表單Email通知
 def hhh_send_mail(email, subject, html):
     SENDER = "Gorgeous Space <noreply@hhh.com.tw>"
@@ -72,17 +87,13 @@ def hhh_send_mail(email, subject, html):
         print("Email sent! Message ID:"),
         print(response['MessageId'])
 
-# Choozmo DB
-# db = dataset.connect("mysql://choozmo:pAssw0rd@db.ptt.cx:3306/hhh?charset=utf8mb4")
 
-# 測試機 DB
-db = dataset.connect('mysql://hhh7796hhh:lYmWsu^ujcA1@hhh-v57-cluster.cluster-cmab1ctkglka.ap-northeast-2.rds.amazonaws.com:3306/stage?charset=utf8mb4')
-
-# 正式機 DB
-# db = dataset.connect('mysql://hhh7796hhh:lYmWsu^ujcA1@hhh-v57-cluster.cluster-cmab1ctkglka.ap-northeast-2.rds.amazonaws.com:3306/xoops?charset=utf8mb4')
 
 @app.get("/is_close_changed_notice")
-async def is_close_changed_notice(exf_id: str = '', is_close: str = ''):   
+async def is_close_changed_notice(exf_id: str = '', is_close: str = ''):
+    # 取得資料庫連線
+    db = get_db_connection()
+
     subject = ""
 
     # 取得合約資料