.env

Author: iach526526

.env.example

@@ -0,0 +1,7 @@
+# 這是一個範例檔案，請複製一份 .env.example 並命名為 .env
+# 並將以下的設定值填入
+MYSQL_USER=資料庫使用者
+MYSQL_PASSWORD=資料庫使用者密碼
+MYSQL_PORT=資料庫連接埠
+MYSQL_DATABASE=資料庫名稱
+HOST=資料庫主機位址

.gitignore

@@ -12,3 +12,4 @@ read.txt
 nohup.out
 pointLog.out
 flakLog.out
+.env

cog/core/secret.py

@@ -1,13 +1,21 @@
-# IMPORTANT!!!!Must ignore git!!!
 import mysql.connector
+import os
+from dotenv import load_dotenv
 
+load_dotenv(f'{os.getcwd()}/.env')
+
+
+HOSTIP = os.getenv('HOST')
+PORT= os.getenv('MYSQL_PORT')
+DBUSER = os.getenv('MYSQL_USER')
+DBNAME= os.getenv('MYSQL_DATABASE')
+PWD = os.getenv('MYSQL_PASSWORD')
 def connect():
-    connection = mysql.connector.connect(
-        host = "",
-        port = "",
-        user = "",
-        passwd = "",
-        database = "",
-        auth_plugin = "mysql_native_password"
+    connection = mysql.connector.connect(host=HOSTIP,
+                                        port=PORT,
+                                        user=DBUSER,  
+                                        passwd=PWD,
+                                        database=DBNAME
     )
+
     return connection