Add files to chanlun_1

main.py

@@ -0,0 +1,234 @@
+"""
+缠论分析系统主程序入口
+"""
+
+import argparse
+import logging
+import sys
+import os
+from datetime import datetime
+
+# 添加项目根目录到Python路径
+sys.path.append(os.path.dirname(os.path.abspath(__file__)))
+
+from data.data_fetcher import DataFetcher
+from data.data_processor import DataProcessor
+from core.chan_analyzer import ChanAnalyzer
+from web.app import run_app
+
+# 配置日志
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
+    handlers=[
+        logging.StreamHandler(sys.stdout),
+        logging.FileHandler('chan_analysis.log', encoding='utf-8')
+    ]
+)
+
+logger = logging.getLogger(__name__)
+
+
+def run_command_line_analysis(args):
+    """运行命令行分析"""
+    try:
+        logger.info("开始命令行缠论分析...")
+        
+        # 初始化组件
+        data_fetcher = DataFetcher(args.exchange)
+        data_processor = DataProcessor()
+        
+        # 获取数据
+        logger.info(f"获取 {args.symbol} {args.timeframe} 数据...")
+        klines = data_fetcher.fetch_klines(
+            symbol=args.symbol,
+            timeframe=args.timeframe,
+            limit=args.limit
+        )
+        
+        if klines.empty:
+            logger.error("未获取到数据")
+            return
+            
+        # 数据处理
+        logger.info("清理和验证数据...")
+        if not data_processor.validate_klines(klines):
+            logger.warning("数据验证失败，继续处理...")
+            
+        klines = data_processor.clean_klines(klines)
+        
+        # 缠论分析
+        logger.info("开始缠论分析...")
+        analyzer = ChanAnalyzer(klines)
+        analysis_summary = analyzer.run_full_analysis(args.fractal_strength)
+        
+        # 输出分析结果
+        print("\n" + "="*50)
+        print("缠论分析结果摘要")
+        print("="*50)
+        
+        # 数据信息
+        data_info = analysis_summary['data_info']
+        print(f"\n数据信息:")
+        print(f"  原始K线数量: {data_info['original_klines']}")
+        print(f"  处理后K线数量: {data_info['processed_klines']}")
+        print(f"  数据时间范围: {data_info['date_range']['start']} 至 {data_info['date_range']['end']}")
+        
+        # 分型信息
+        fractal_info = analysis_summary['fractal_info']
+        print(f"\n分型信息:")
+        print(f"  总分型数量: {fractal_info['total']}")
+        print(f"  顶分型: {fractal_info['top']}")
+        print(f"  底分型: {fractal_info['bottom']}")
+        
+        # 笔信息
+        stroke_info = analysis_summary['stroke_info']
+        print(f"\n笔信息:")
+        print(f"  总笔数量: {stroke_info['total']}")
+        print(f"  上升笔: {stroke_info['up']}")
+        print(f"  下降笔: {stroke_info['down']}")
+        
+        # 线段信息
+        segment_info = analysis_summary['segment_info']
+        print(f"\n线段信息:")
+        print(f"  总线段数量: {segment_info['total']}")
+        print(f"  上升线段: {segment_info['up']}")
+        print(f"  下降线段: {segment_info['down']}")
+        
+        # 中枢信息
+        cb_info = analysis_summary['central_bank_info']
+        print(f"\n中枢信息:")
+        print(f"  总中枢数量: {cb_info['total']}")
+        if cb_info['levels']:
+            print("  级别分布:")
+            for level, count in cb_info['levels'].items():
+                print(f"    {level}: {count}个")
+        
+        # 买卖点信息
+        signal_info = analysis_summary['trading_signal_info']
+        print(f"\n买卖点信息:")
+        print(f"  总信号数量: {signal_info['total']}")
+        print(f"  买点: {signal_info['buy_points']}")
+        print(f"  卖点: {signal_info['sell_points']}")
+        if signal_info['by_class']:
+            print("  类别分布:")
+            for class_name, count in signal_info['by_class'].items():
+                print(f"    {class_name}: {count}个")
+        
+        # 获取最新信号
+        latest_signals = analyzer.get_latest_signals(24)  # 最近24小时
+        if latest_signals:
+            print(f"\n最近24小时信号 ({len(latest_signals)}个):")
+            for signal in latest_signals:
+                signal_type = "买点" if signal.signal_type == 'buy' else "卖点"
+                class_type = {"first": "一类", "second": "二类", "third": "三类"}.get(signal.point_class, "未知")
+                print(f"  {signal.timestamp.strftime('%m-%d %H:%M')} - {class_type}{signal_type} - 价格:{signal.price:.2f} - 强度:{signal.strength:.3f}")
+        
+        # 当前市场结构
+        market_structure = analyzer.get_current_market_structure()
+        if market_structure:
+            print(f"\n当前市场结构:")
+            print(f"  当前价格: {market_structure.get('current_price', 'N/A')}")
+            trend_map = {'upward': '上升', 'downward': '下降', 'sideways': '震荡', 'unclear': '不明'}
+            print(f"  趋势方向: {trend_map.get(market_structure.get('trend'), '未知')}")
+            phase_map = {'consolidation': '盘整', 'trending': '趋势', 'unknown': '未知'}
+            print(f"  市场阶段: {phase_map.get(market_structure.get('market_phase'), '未知')}")
+        
+        # 保存结果
+        if args.export:
+            export_results(analyzer, args)
+            
+        logger.info("命令行分析完成")
+        
+    except Exception as e:
+        logger.error(f"命令行分析失败: {e}")
+        raise
+
+
+def export_results(analyzer: ChanAnalyzer, args):
+    """导出分析结果"""
+    try:
+        results = analyzer.export_results()
+        
+        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+        export_dir = f"exports/chan_analysis_{args.symbol.replace('/', '_')}_{timestamp}"
+        os.makedirs(export_dir, exist_ok=True)
+        
+        # 保存各种分析结果
+        for name, df in results.items():
+            if not df.empty:
+                filename = f"{export_dir}/{name}.csv"
+                df.to_csv(filename, encoding='utf-8-sig')
+                logger.info(f"导出 {name} 到 {filename}")
+        
+        # 保存分析摘要
+        with open(f"{export_dir}/summary.txt", 'w', encoding='utf-8') as f:
+            f.write(f"缠论分析结果摘要\n")
+            f.write(f"交易对: {args.symbol}\n")
+            f.write(f"时间周期: {args.timeframe}\n")
+            f.write(f"分析时间: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}\n")
+            f.write(f"数据数量: {args.limit}\n")
+            f.write(f"分型强度: {args.fractal_strength}\n")
+        
+        logger.info(f"分析结果已导出到 {export_dir}")
+        
+    except Exception as e:
+        logger.error(f"导出结果失败: {e}")
+
+
+def main():
+    """主程序入口"""
+    parser = argparse.ArgumentParser(description='缠论分析系统')
+    
+    # 运行模式
+    parser.add_argument('--mode', choices=['web', 'cli'], default='web',
+                       help='运行模式：web(Web界面) 或 cli(命令行)')
+    
+    # 数据参数
+    parser.add_argument('--symbol', default='BTC/USDT',
+                       help='交易对 (默认: BTC/USDT)')
+    parser.add_argument('--timeframe', default='1h',
+                       help='时间周期 (默认: 1h)')
+    parser.add_argument('--limit', type=int, default=500,
+                       help='获取的K线数量 (默认: 500)')
+    parser.add_argument('--exchange', default='binance',
+                       help='交易所 (默认: binance)')
+    
+    # 分析参数
+    parser.add_argument('--fractal-strength', type=int, default=1,
+                       help='分型强度要求 (默认: 1)')
+    
+    # 输出参数
+    parser.add_argument('--export', action='store_true',
+                       help='是否导出分析结果')
+    
+    # Web参数
+    parser.add_argument('--port', type=int, default=8050,
+                       help='Web服务端口 (默认: 8050)')
+    parser.add_argument('--debug', action='store_true',
+                       help='是否启用调试模式')
+    
+    args = parser.parse_args()
+    
+    logger.info("缠论分析系统启动")
+    logger.info(f"运行模式: {args.mode}")
+    
+    try:
+        if args.mode == 'web':
+            logger.info(f"启动Web服务，端口: {args.port}")
+            print(f"\n缠论分析系统Web界面")
+            print(f"访问地址: http://localhost:{args.port}")
+            print("按 Ctrl+C 停止服务\n")
+            run_app(debug=args.debug, port=args.port)
+        else:
+            run_command_line_analysis(args)
+            
+    except KeyboardInterrupt:
+        logger.info("用户中断，程序退出")
+    except Exception as e:
+        logger.error(f"程序执行失败: {e}")
+        sys.exit(1)
+
+
+if __name__ == "__main__":
+    main()