855 协议版本-调整完毕内容

db/base.py

@@ -1,5 +1,5 @@
 # -*- coding: utf-8 -*-
-import logging
+from loguru import logger
 from typing import List, Dict, Any, Optional, Tuple, Union
 
 from db.connection import DBConnectionManager
@@ -7,12 +7,13 @@ from db.connection import DBConnectionManager
 
 class BaseDBOperator:
     """基础数据库操作类"""
-    
+
     def __init__(self, db_manager: DBConnectionManager):
         self.db_manager = db_manager
-        self.LOG = logging.getLogger(self.__class__.__name__)
-    
-    def execute_query(self, sql: str, params: Optional[tuple] = None, fetch_one: bool = False) -> Union[List[Dict], Dict, None]:
+        self.LOG = logger
+
+    def execute_query(self, sql: str, params: Optional[tuple] = None, fetch_one: bool = False) -> Union[
+        List[Dict], Dict, None]:
         """执行查询SQL"""
         conn = self.db_manager.get_mysql_connection()
         try:
@@ -26,7 +27,7 @@ class BaseDBOperator:
             return None
         finally:
             conn.close()
-    
+
     def execute_update(self, sql: str, params: Optional[tuple] = None) -> bool:
         """执行更新SQL"""
         conn = self.db_manager.get_mysql_connection()
@@ -41,12 +42,12 @@ class BaseDBOperator:
             return False
         finally:
             conn.close()
-    
+
     def execute_batch(self, sql: str, params_list: List[tuple]) -> bool:
         """批量执行SQL"""
         if not params_list:
             return True
-            
+
         conn = self.db_manager.get_mysql_connection()
         try:
             with conn.cursor() as cursor:
@@ -59,12 +60,12 @@ class BaseDBOperator:
             return False
         finally:
             conn.close()
-    
+
     def execute_transaction(self, operations: List[Tuple[str, tuple]]) -> bool:
         """执行事务"""
         if not operations:
             return True
-            
+
         conn = self.db_manager.get_mysql_connection()
         try:
             with conn.cursor() as cursor:
@@ -77,4 +78,4 @@ class BaseDBOperator:
             conn.rollback()
             return False
         finally:
-            conn.close()
+            conn.close()