Merge pull request #295 from someoneiscoding/master

Boris-code · web-flow · commit 0301b71cc488 · 2025-12-15T21:05:09.000+08:00
MysqlDB#__init__ 增加 set_session 参数
diff --git a/feapder/db/mysqldb.py b/feapder/db/mysqldb.py
@@ -41,7 +41,7 @@ def wapper(*args, **kwargs):
 
 class MysqlDB:
     def __init__(
-        self, ip=None, port=None, db=None, user_name=None, user_pass=None, charset="utf8mb4", **kwargs
+        self, ip=None, port=None, db=None, user_name=None, user_pass=None, charset="utf8mb4", set_session=None, **kwargs
     ):
         # 可能会改setting中的值，所以此处不能直接赋值为默认值，需要后加载赋值
         if not ip:
@@ -69,7 +69,9 @@ def __init__(
                 passwd=user_pass,
                 db=db,
                 charset=charset,
+                setsession=set_session,
                 cursorclass=cursors.SSCursor,
+                **kwargs
             )  # cursorclass 使用服务的游标，默认的在多线程下大批量插入数据会使内存递增
 
         except Exception as e:
diff --git a/tests/test_mysqldb.py b/tests/test_mysqldb.py
@@ -2,7 +2,10 @@
 
 
 db = MysqlDB(
-    ip="localhost", port=3306, db="feapder", user_name="feapder", user_pass="feapder123"
+    ip="localhost", port=3306, db="feapder", user_name="feapder", user_pass="feapder123", set_session=["SET time_zone='+08:00'"]
 )
 
-MysqlDB.from_url("mysql://feapder:feapder123@localhost:3306/feapder?charset=utf8mb4")
+MysqlDB.from_url("mysql://feapder:feapder123@localhost:3306/feapder?charset=utf8mb4")
+
+result = db.find("SELECT @@global.time_zone, @@session.time_zone, date_format(NOW(), '%Y-%m-%d %H:%i:%s')")
+print(f"Database timezone info: {result}")
