70 lines
2.3 KiB
Python
70 lines
2.3 KiB
Python
|
import os
|
|||
|
|
import logging
|
|||
|
|
from excel_stock_importer import ExcelStockImporter
|
|||
|
|
|
|||
|
|
# 设置日志
|
|||
|
|
logging.basicConfig(level=logging.INFO)
|
|||
|
|
|
|||
|
|
def main():
|
|||
|
|
# 数据库配置
|
|||
|
|
db_config = {
|
|||
|
|
'host': '127.0.0.1',
|
|||
|
|
'database': 'fintech',
|
|||
|
|
'user': 'root',
|
|||
|
|
'password': 'secret',
|
|||
|
|
'port': 3306
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# Excel文件路径 - 尝试不同的路径
|
|||
|
|
excel_files = [
|
|||
|
|
"stock_list.xlsx", # 当前目录
|
|||
|
|
"./finance/stock_list.xlsx", # 当前目录（明确指定）
|
|||
|
|
os.path.join(os.getcwd(), "./finance/stock_list.xlsx"), # 绝对路径
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
print("=== 当前工作目录 ===")
|
|||
|
|
print(f"工作目录: {os.getcwd()}")
|
|||
|
|
print(f"目录内容: {os.listdir('.')}")
|
|||
|
|
|
|||
|
|
# 初始化导入器
|
|||
|
|
importer = ExcelStockImporter(db_config)
|
|||
|
|
|
|||
|
|
found_file = None
|
|||
|
|
for excel_file in excel_files:
|
|||
|
|
if os.path.exists(excel_file):
|
|||
|
|
found_file = excel_file
|
|||
|
|
print(f"找到文件: {excel_file}")
|
|||
|
|
break
|
|||
|
|
else:
|
|||
|
|
print(f"文件不存在: {excel_file}")
|
|||
|
|
|
|||
|
|
if not found_file:
|
|||
|
|
print("未找到stock_list.xlsx文件，请检查：")
|
|||
|
|
print("1. 文件名是否正确（包括扩展名）")
|
|||
|
|
print("2. 文件是否在当前目录")
|
|||
|
|
print("3. 文件权限是否正常")
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
# 验证数据
|
|||
|
|
print("\n=== 验证数据 ===")
|
|||
|
|
validation_result = importer.validate_data(found_file)
|
|||
|
|
print(f"数据验证结果: {'通过' if validation_result['valid'] else '失败'}")
|
|||
|
|
|
|||
|
|
if 'total_rows' in validation_result:
|
|||
|
|
print(f"总行数: {validation_result['total_rows']}")
|
|||
|
|
print(f"行业列表: {validation_result.get('industries', [])}")
|
|||
|
|
print(f"交易所列表: {validation_result.get('exchanges', [])}")
|
|||
|
|
|
|||
|
|
if validation_result.get('duplicate_stocks'):
|
|||
|
|
print(f"重复股票代码: {validation_result['duplicate_stocks']}")
|
|||
|
|
|
|||
|
|
if validation_result.get('valid'):
|
|||
|
|
# 导入数据
|
|||
|
|
print("\n=== 导入数据 ===")
|
|||
|
|
success = importer.import_to_database(found_file)
|
|||
|
|
print(f"导入结果: {'成功' if success else '失败'}")
|
|||
|
|
else:
|
|||
|
|
print(f"数据验证失败: {validation_result.get('message', '未知错误')}")
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
main()
|