丁云鹏 преди 8 месеца
родител
ревизия
12632ddd6d
променени са 1 файла, в които са добавени 3 реда и са изтрити 1 реда
  1. 3 1
      utils/mysql_db.py

+ 3 - 1
utils/mysql_db.py

@@ -3,6 +3,8 @@
 数据库连接及操作
 """
 import pymysql
+from loguru import logger
+logger.add("/app/logs/consumption.log", rotation="10 MB")
 
 class MysqlHelper:
     @classmethod
@@ -57,7 +59,7 @@ class MysqlHelper:
             return res
         except Exception as e:
             # 发生错误时回滚
-            print(f"Error executing SQL: {e}")
+            logger.error(f"Error executing SQL: {e}")
             connect.rollback()
             return None  # 返回 None 表示失败
         finally: