添加本地数据备选: 在线获取失败时使用本地SZ#399673.txt文件

cat-fly/auto_report.py

@@ -71,14 +71,14 @@ class DataFetcher:
     
     @staticmethod
     def fetch_recent_2months():
-        """获取近2个月数据"""
+        """获取近2个月数据 - 优先在线获取，失败则使用本地数据"""
         end_date = datetime.now()
         start_date = end_date - timedelta(days=70)  # 2个月+10天缓冲
         
         print(f"获取数据: {start_date.strftime('%Y-%m-%d')} 至 {end_date.strftime('%Y-%m-%d')}")
         
+        # 首先尝试在线获取
         try:
-            # 使用akshare获取30分钟K线
             df = ak.index_zh_a_hist_min_em(
                 symbol="399673",
                 period="30",
@@ -86,34 +86,103 @@ class DataFetcher:
                 end_date=end_date.strftime('%Y%m%d%H%M%S')
             )
             
-            if df is None or len(df) == 0:
-                print("❌ 未获取到数据")
+            if df is not None and len(df) > 0:
+                df = df.rename(columns={
+                    '时间': 'datetime',
+                    '开盘': 'open',
+                    '收盘': 'close',
+                    '最高': 'high',
+                    '最低': 'low',
+                    '成交量': 'volume'
+                })
+                
+                df['datetime'] = pd.to_datetime(df['datetime'])
+                df = df.set_index('datetime').sort_index()
+                
+                backtest_start = end_date - timedelta(days=60)
+                df_backtest = df[df.index >= backtest_start]
+                
+                print(f"✅ 在线数据获取成功: 共{len(df_backtest)}条30分钟K线")
+                print(f"   数据区间: {df_backtest.index[0]} 至 {df_backtest.index[-1]}")
+                return df_backtest
+        except Exception as e:
+            print(f"⚠️ 在线数据获取失败: {e}")
+            print("   尝试使用本地数据...")
+        
+        # 在线获取失败，使用本地数据
+        return DataFetcher._load_local_data(start_date, end_date)
+    
+    @staticmethod
+    def _load_local_data(start_date, end_date):
+        """从本地文件加载数据"""
+        local_file = 'SZ#399673.txt'
+        
+        try:
+            if not os.path.exists(local_file):
+                print(f"❌ 本地文件不存在: {local_file}")
                 return None
             
-            # 标准化列名
-            df = df.rename(columns={
-                '时间': 'datetime',
-                '开盘': 'open',
-                '收盘': 'close',
-                '最高': 'high',
-                '最低': 'low',
-                '成交量': 'volume'
-            })
+            print(f"正在从本地文件读取数据: {local_file}")
+            
+            # 读取文本格式数据
+            data_list = []
+            encodings = ['gbk', 'gb2312', 'utf-8', 'latin-1']
+            lines = None
+            
+            for encoding in encodings:
+                try:
+                    with open(local_file, 'r', encoding=encoding) as f:
+                        lines = f.readlines()
+                    print(f"✅ 成功使用编码: {encoding}")
+                    break
+                except UnicodeDecodeError:
+                    continue
+            
+            if lines is None:
+                raise ValueError("无法读取文件，尝试了多种编码格式都失败")
+            
+            # 跳过前两行（标题行）
+            for line in lines[2:]:
+                line = line.strip()
+                if not line:
+                    continue
+                    
+                parts = line.split()
+                if len(parts) >= 7:
+                    try:
+                        # 格式: 日期 时间 开盘 最高 最低 收盘 成交量 成交额
+                        date_time_str = f"{parts[0]} {parts[1]}"
+                        datetime_obj = pd.to_datetime(date_time_str, format='%Y/%m/%d %H%M')
+                        
+                        data_list.append({
+                            'datetime': datetime_obj,
+                            'open': float(parts[2]),
+                            'high': float(parts[3]),
+                            'low': float(parts[4]),
+                            'close': float(parts[5]),
+                            'volume': float(parts[6])
+                        })
+                    except (ValueError, IndexError) as e:
+                        continue
+            
+            if not data_list:
+                raise ValueError("文本文件中没有解析到有效数据")
             
-            df['datetime'] = pd.to_datetime(df['datetime'])
+            df = pd.DataFrame(data_list)
             df = df.set_index('datetime').sort_index()
             
-            # 只保留最近2个月的数据用于回测
+            # 只保留最近2个月的数据
             backtest_start = end_date - timedelta(days=60)
             df_backtest = df[df.index >= backtest_start]
             
-            print(f"✅ 数据获取成功: 共{len(df_backtest)}条30分钟K线")
-            print(f"   数据区间: {df_backtest.index[0]} 至 {df_backtest.index[-1]}")
+            print(f"✅ 本地数据加载成功: 共{len(df_backtest)}条30分钟K线")
+            if len(df_backtest) > 0:
+                print(f"   数据区间: {df_backtest.index[0]} 至 {df_backtest.index[-1]}")
             
             return df_backtest
             
         except Exception as e:
-            print(f"❌ 数据获取失败: {e}")
+            print(f"❌ 本地数据加载失败: {e}")
             return None