update 数据库根据env文件来选择

.gitignore

@@ -1,2 +1,3 @@
 venv
-.idea
+.idea
+.env

common/db/mysql_help.py

@@ -4,15 +4,19 @@
 """
 数据库连接及操作
 """
+import os
 import logging
-
 import pymysql
+from dotenv import load_dotenv
+
+load_dotenv(verbose=True)
+env = os.getenv('env')
 
 
 class MysqlHelper:
     @classmethod
-    def connect_mysql(cls, env='test', machine='test'):
-        if machine == 'aliyun_hk':
+    def connect_mysql(cls):
+        if env == 'hk':
             # 创建一个 Connection 对象，代表了一个数据库连接
             connection = pymysql.connect(
                 host="rm-j6cz4c6pt96000xi3.mysql.rds.aliyuncs.com",  # 数据库IP地址，内网地址
@@ -49,10 +53,10 @@ class MysqlHelper:
         return connection
 
     @classmethod
-    def get_values(cls, sql, env='test', machine='test'):
+    def get_values(cls, sql):
         try:
             # 连接数据库
-            connect = cls.connect_mysql(env, machine)
+            connect = cls.connect_mysql()
             # 返回一个 Cursor对象
             mysql = connect.cursor()
 
@@ -71,10 +75,10 @@ class MysqlHelper:
             logging.error(f"get_values异常:{e}\n")
 
     @classmethod
-    def insert_values(cls, sql, value, env='test', machine='test'):
+    def insert_values(cls, sql, value):
         try:
             # 连接数据库
-            connect = cls.connect_mysql(env, machine)
+            connect = cls.connect_mysql()
             # 返回一个 Cursor对象
             mysql = connect.cursor()
 
@@ -89,9 +93,9 @@ class MysqlHelper:
             logging.error(f"insert_values异常:{e}\n")
 
     @classmethod
-    def update_values(cls, sql, env='test', machine='test'):
+    def update_values(cls, sql):
         # 连接数据库
-        connect = cls.connect_mysql(env, machine)
+        connect = cls.connect_mysql()
         # 返回一个 Cursor对象
         mysql = connect.cursor()
 
@@ -117,5 +121,5 @@ if __name__ == "__main__":
     # edit_data = MysqlHelper.edit_data(sql=sql_statement)
     # print(edit_data)
 
-    get_data = MysqlHelper.get_values("select * from crawler_user", "dev", "local")
+    get_data = MysqlHelper.get_values("select * from crawler_user")
     print(get_data)