Explorar el Código

transcode job id

丁云鹏 hace 1 mes
padre
commit
12632ddd6d
Se han modificado 1 ficheros con 3 adiciones y 1 borrados
  1. 3 1
      utils/mysql_db.py

+ 3 - 1
utils/mysql_db.py

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