接入数据库

scripts/generate_sample_data.py

@@ -1,10 +1,17 @@
 """生成测试数据以便在无法连接Binance时使用"""
 
+import sys
+import os
+from pathlib import Path
+
+# 添加项目根目录 to path
+sys.path.append(str(Path(__file__).resolve().parent.parent))
+
 import pandas as pd
 import numpy as np
 from datetime import datetime, timedelta
-from pathlib import Path
 
+from data.database import DatabaseManager
 
 def generate_sample_data(
     symbol: str = 'BTC',
@@ -76,14 +83,13 @@ def generate_sample_data(
 
     df = pd.DataFrame(data)
 
-    # 保存到缓存
-    cache_dir = Path('./data/cache')
-    cache_dir.mkdir(parents=True, exist_ok=True)
+    # 保存到数据库
+    # 注意：我们这里假设DatabaseManager的默认路径在脚本运行时是相对于CWD的
+    # 如果从根目录运行，./data/cache/market_data.db 是正确的
+    db_manager = DatabaseManager()
+    db_manager.save_ohlcv(df, symbol, timeframe)
 
-    cache_file = cache_dir / f"{symbol}_{timeframe}_1y_now.csv"
-    df.to_csv(cache_file, index=False)
-
-    print(f"✓ 已生成 {len(df)} 行模拟数据并保存到：{cache_file}")
+    print(f"✓ 已生成 {len(df)} 行模拟数据并保存到数据库")
     return df