update log file

exportExcelToDB_SZ.py

@@ -1,118 +1,36 @@
 """
-    下载的深圳交易所的数据
+    存储深圳交易所股票列表数据
 
-    表头和上海交易所的数据格式不一致，所以分开存储
+    不确定其数据爬取规则，防止 IP 被封
+    暂时使用该方案，获取股票列表数据
+    —— 下载excel,收到导入到数据库
 """
-import pandas as pd
-import pymysql
-from pymysql import Error
 from pathlib import Path
+from datetime import datetime
+from MySQLHelper import MySQLHelper
+from LogHelper import LogHelper
+import pandas as pd
 import os
 import logging
-from datetime import datetime
 import sys
 import csv
 import chardet
-import re
-from typing import List, Dict, Union, Tuple, Optional
 
-# 配置日志
-logging.basicConfig(
-    level=logging.INFO,
-    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
-    handlers=[
-        logging.StreamHandler(sys.stdout),
-        logging.FileHandler('stock_data_import.log')
-    ]
+
+# 创建配置实例
+logHelper = LogHelper(
+    level=logging.DEBUG,  # 设置日志级别为 DEBUG
+    format='%(asctime)s [%(levelname)s] %(message)s'  # 自定义格式
 )
+
+# # 添加处理器
+logHelper.add_console_handler()  # 默认输出到 stdout
+logHelper.add_file_handler('Debug.log')  # 添加文件日志
+
+# # 应用配置
+logHelper.setup()
 logger = logging.getLogger('StockDataImporter')
 
-class MySQLHelper:
-    """MySQL 数据库操作助手类"""
-    def __init__(self, host: str, user: str, password: str, database: str, 
-                 port: int = 3306, charset: str = 'utf8mb4'):
-        self.host = host
-        self.user = user
-        self.password = password
-        self.database = database
-        self.port = port
-        self.charset = charset
-        self.connection = None
-        self.cursor = None
-
-    def connect(self) -> bool:
-        try:
-            self.connection = pymysql.connect(
-                host=self.host,
-                user=self.user,
-                password=self.password,
-                database=self.database,
-                port=self.port,
-                charset=self.charset,
-                cursorclass=pymysql.cursors.DictCursor
-            )
-            self.cursor = self.connection.cursor()
-            logger.info("MySQL数据库连接成功")
-            return True
-        except Error as e:
-            logger.error(f"连接MySQL数据库失败: {e}")
-            return False
-
-    def close(self) -> None:
-        if self.cursor:
-            self.cursor.close()
-        if self.connection:
-            self.connection.close()
-        logger.info("MySQL数据库连接已关闭")
-
-    def execute_query(self, sql: str, params: Union[Tuple, List, Dict, None] = None) -> List[Dict]:
-        try:
-            self.cursor.execute(sql, params)
-            return self.cursor.fetchall()
-        except Error as e:
-            logger.error(f"查询执行失败: {e}")
-            return []
-
-    def execute_update(self, sql: str, params: Union[Tuple, List, Dict, None] = None) -> int:
-        try:
-            affected_rows = self.cursor.execute(sql, params)
-            self.connection.commit()
-            return affected_rows
-        except Error as e:
-            self.connection.rollback()
-            logger.error(f"更新执行失败: {e}")
-            return 0
-
-    def execute_many(self, sql: str, params_list: List[Union[Tuple, List, Dict]]) -> int:
-        try:
-            affected_rows = self.cursor.executemany(sql, params_list)
-            self.connection.commit()
-            return affected_rows
-        except Error as e:
-            self.connection.rollback()
-            logger.error(f"批量执行失败: {e}")
-            return 0
-
-    def get_one(self, sql: str, params: Union[Tuple, List, Dict, None] = None) -> Optional[Dict]:
-        try:
-            self.cursor.execute(sql, params)
-            return self.cursor.fetchone()
-        except Error as e:
-            logger.error(f"获取单条记录失败: {e}")
-            return None
-
-    def table_exists(self, table_name: str) -> bool:
-        sql = "SHOW TABLES LIKE %s"
-        result = self.execute_query(sql, (table_name,))
-        return len(result) > 0
-
-    def __enter__(self):
-        self.connect()
-        return self
-
-    def __exit__(self, exc_type, exc_val, exc_tb):
-        self.close()
-
 class StockDataImporter:
     """股票数据导入工具（支持新版CSV格式）"""
     
@@ -128,13 +46,13 @@ class StockDataImporter:
         'A股总股本': 'a_total_shares',
         'A股流通股本': 'a_circulating_shares',
         'B股代码': 'b_stock_code',
-        'B股简称': 'b_stock_short_name',
+        'B股 简 称': 'b_stock_short_name',
         'B股上市日期': 'b_listing_date',
         'B股总股本': 'b_total_shares',
         'B股流通股本': 'b_circulating_shares',
-        '地区': 'region',
-        '省份': 'province',
-        '城市': 'city',
+        '地      区': 'region',
+        '省    份': 'province',
+        '城     市': 'city',
         '所属行业': 'industry',
         '公司网址': 'website',
         '未盈利': 'unprofitable',