#!/usr/bin/env python3
"""
快速验证DataFrame长度修复的脚本
"""

import sys
import os
import subprocess

def check_strategy_syntax():
    """检查策略文件语法"""
    print("🔍 检查策略文件语法...")
    
    strategy_file = "/Users/zhangkun/myTestFreqAI/freqtrade/templates/freqaiprimer.py"
    
    if not os.path.exists(strategy_file):
        print(f"❌ 策略文件不存在: {strategy_file}")
        return False
    
    # 检查Python语法
    try:
        result = subprocess.run([
            'python3', '-c', 
            f"import ast; ast.parse(open('{strategy_file}').read())"
        ], capture_output=True, text=True)
        
        if result.returncode == 0:
            print("✅ 策略文件语法正确")
            return True
        else:
            print(f"❌ 策略文件语法错误: {result.stderr}")
            return False
            
    except Exception as e:
        print(f"❌ 检查语法时出错: {e}")
        return False

def check_imports():
    """检查必要的导入"""
    print("📦 检查必要的导入...")
    
    imports_to_check = [
        "import pandas as pd",
        "import numpy as np",
        "from freqtrade.strategy.interface import IStrategy"
    ]
    
    for imp in imports_to_check:
        try:
            result = subprocess.run([
                'python3', '-c', imp
            ], capture_output=True, text=True)
            
            if result.returncode != 0:
                print(f"⚠️  导入警告: {imp}")
                print(f"   错误: {result.stderr.strip()}")
            else:
                print(f"✅ 导入正常: {imp}")
                
        except Exception as e:
            print(f"⚠️  检查导入时出错: {e}")
    
    return True

def check_dataframe_fixes():
    """检查DataFrame修复代码是否存在"""
    print("🔧 检查DataFrame修复代码...")
    
    strategy_file = "/Users/zhangkun/myTestFreqAI/freqtrade/templates/freqaiprimer.py"
    
    try:
        with open(strategy_file, 'r', encoding='utf-8') as f:
            content = f.read()
        
        # 检查关键修复代码
        fixes_found = []
        
        if "DataFrame长度不匹配" in content:
            fixes_found.append("✅ DataFrame长度验证代码")
        
        if "original_length" in content and "original_index" in content:
            fixes_found.append("✅ 原始长度记录")
        
        if "数据填充完成" in content:
            fixes_found.append("✅ 数据填充修复")
        
        if "数据截断完成" in content:
            fixes_found.append("✅ 数据截断修复")
        
        if "数据长度验证通过" in content:
            fixes_found.append("✅ 长度验证完成")
        
        if fixes_found:
            print("找到的修复代码:")
            for fix in fixes_found:
                print(f"  {fix}")
            return len(fixes_found) >= 3
        else:
            print("❌ 未找到DataFrame修复代码")
            return False
            
    except Exception as e:
        print(f"❌ 检查修复代码时出错: {e}")
        return False

def check_files():
    """检查必要文件是否存在"""
    print("📁 检查必要文件...")
    
    files_to_check = [
        "/Users/zhangkun/myTestFreqAI/freqtrade/templates/freqaiprimer.py",
        "/Users/zhangkun/myTestFreqAI/apply_dataframe_fix.sh",
        "/Users/zhangkun/myTestFreqAI/system_restart.sh"
    ]
    
    all_exist = True
    for file_path in files_to_check:
        if os.path.exists(file_path):
            print(f"✅ 存在: {os.path.basename(file_path)}")
        else:
            print(f"❌ 不存在: {os.path.basename(file_path)}")
            all_exist = False
    
    return all_exist

def main():
    """主函数"""
    print("🚀 DataFrame长度修复验证")
    print("=" * 40)
    
    # 切换到正确目录
    os.chdir('/Users/zhangkun/myTestFreqAI')
    
    # 运行各项检查
    checks = [
        ("文件检查", check_files),
        ("语法检查", check_strategy_syntax),
        ("导入检查", check_imports),
        ("修复代码检查", check_dataframe_fixes)
    ]
    
    passed = 0
    total = len(checks)
    
    for name, check_func in checks:
        print(f"\n{name}:")
        if check_func():
            passed += 1
        else:
            print(f"  ❌ {name}失败")
    
    print("\n" + "=" * 40)
    print(f"🎯 检查结果: {passed}/{total} 项通过")
    
    if passed == total:
        print("\n🎉 所有检查通过！修复已就绪")
        print("\n📋 下一步操作:")
        print("  1. 运行修复脚本: ./apply_dataframe_fix.sh")
        print("  2. 或者手动操作:")
        print("     • 停止服务: pkill -f freqtrade")
        print("     • 清理缓存: rm -rf user_data/data/*")
        print("     • 启动服务: python3 -m freqtrade trade -c config.json")
        return 0
    else:
        print("\n❌ 部分检查失败，请查看详细信息")
        return 1

if __name__ == "__main__":
    sys.exit(main())