Added a logging utility class and supplemented, standardized the logging output for all modules.

utils/logger.py

@@ -0,0 +1,57 @@
+import os
+import logging
+from logging.handlers import RotatingFileHandler
+
+def setup_logger(name, log_file=None, level=logging.INFO):
+    """
+    设置统一的日志记录器
+    
+    Args:
+        name: 日志记录器名称
+        log_file: 日志文件路径，如果为None则只输出到控制台
+        level: 日志级别
+    
+    Returns:
+        logger: 配置好的日志记录器
+    """
+    # 创建日志记录器
+    logger = logging.getLogger(name)
+    logger.setLevel(level)
+    
+    # 统一的日志格式
+    formatter = logging.Formatter(
+        '%(asctime)s [%(name)s] [%(levelname)s] %(message)s',
+        datefmt='%Y-%m-%d %H:%M:%S'
+    )
+    
+    # 添加控制台处理器
+    console_handler = logging.StreamHandler()
+    console_handler.setFormatter(formatter)
+    logger.addHandler(console_handler)
+    
+    # 如果指定了日志文件，添加文件处理器
+    if log_file:
+        # 确保日志目录存在
+        log_dir = os.path.dirname(log_file)
+        if log_dir and not os.path.exists(log_dir):
+            os.makedirs(log_dir)
+            
+        # 使用 RotatingFileHandler 进行日志轮转
+        file_handler = RotatingFileHandler(
+            log_file,
+            maxBytes=10*1024*1024,  # 10MB
+            backupCount=5,
+            encoding='utf-8'
+        )
+        file_handler.setFormatter(formatter)
+        logger.addHandler(file_handler)
+    
+    return logger
+
+# 创建默认的应用日志记录器
+app_logger = setup_logger('weibo_analysis', 'logs/app.log')
+spider_logger = setup_logger('spider', 'logs/spider.log')
+model_logger = setup_logger('model', 'logs/model.log')
+
+# 导出日志记录器
+__all__ = ['setup_logger', 'app_logger', 'spider_logger', 'model_logger'] 

utils/query.py

@@ -1,6 +1,6 @@
-import getpass
 import pymysql
-import logging
+from getpass import getpass
+from utils.logger import app_logger as logging
 
 # 配置日志
 logging.basicConfig(
@@ -28,7 +28,7 @@ def get_db_connection_interactive():
         port = 3306
     
     user = input(" 3. 用户名 (默认: root): ") or "root"
-    password = getpass.getpass(" 4. 密码 (默认: 12345678): ") or "12345678"
+    password = getpass(" 4. 密码 (默认: 12345678): ") or "12345678"
     db_name = input(" 5. 数据库名 (默认: Weibo_PublicOpinion_AnalysisSystem): ") or "Weibo_PublicOpinion_AnalysisSystem"
     
     logging.info(f"尝试连接到数据库: {user}@{host}:{port}/{db_name}")