#!/usr/bin/env python3
"""
AI回复修复验证脚本
验证settings变量作用域问题是否已修复
"""

import asyncio
import sys
import os

# 添加项目根目录到Python路径
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))

from ai_reply_engine import ai_reply_engine
from db_manager import db_manager

def setup_test_account():
    """设置测试账号"""
    test_cookie_id = "test_fix_001"
    
    # 配置AI回复设置
    ai_settings = {
        'ai_enabled': True,
        'model_name': 'qwen-plus',
        'api_key': 'test-api-key',  # 测试用假密钥
        'base_url': 'https://dashscope.aliyuncs.com/compatible-mode/v1',
        'max_discount_percent': 10,
        'max_discount_amount': 100,
        'max_bargain_rounds': 3,
        'custom_prompts': ''
    }
    
    success = db_manager.save_ai_reply_settings(test_cookie_id, ai_settings)
    return test_cookie_id if success else None

def test_settings_variable_scope():
    """测试settings变量作用域问题"""
    print("🔧 测试settings变量作用域修复...")
    
    test_cookie_id = setup_test_account()
    if not test_cookie_id:
        print("   ❌ 测试账号设置失败")
        return False
    
    # 测试数据
    test_item_info = {
        'title': '测试商品',
        'price': 100,
        'desc': '测试商品描述'
    }
    
    test_chat_id = "test_chat_fix_001"
    
    # 清理测试数据
    try:
        with db_manager.lock:
            cursor = db_manager.conn.cursor()
            cursor.execute('DELETE FROM ai_conversations WHERE cookie_id = ? AND chat_id = ?', 
                         (test_cookie_id, test_chat_id))
            db_manager.conn.commit()
    except:
        pass
    
    print(f"   测试账号: {test_cookie_id}")
    print(f"   测试对话: {test_chat_id}")
    
    # 测试1: 普通消息（非议价）
    print(f"\n1️⃣ 测试普通消息处理...")
    try:
        reply = ai_reply_engine.generate_reply(
            message="你好",
            item_info=test_item_info,
            chat_id=test_chat_id,
            cookie_id=test_cookie_id,
            user_id="test_user",
            item_id="test_item"
        )
        
        # 由于使用测试API密钥，预期会失败，但不应该出现settings变量错误
        print(f"   普通消息测试完成（预期API调用失败）")
        
    except Exception as e:
        error_msg = str(e)
        if "cannot access local variable 'settings'" in error_msg:
            print(f"   ❌ settings变量作用域问题仍然存在: {error_msg}")
            return False
        else:
            print(f"   ✅ settings变量作用域问题已修复（其他错误: {error_msg[:50]}...）")
    
    # 测试2: 议价消息
    print(f"\n2️⃣ 测试议价消息处理...")
    try:
        # 先添加一些议价记录，测试轮数限制逻辑
        for i in range(3):  # 添加3轮议价记录
            ai_reply_engine.save_conversation(
                chat_id=test_chat_id,
                cookie_id=test_cookie_id,
                user_id="test_user",
                item_id="test_item",
                role="user",
                content=f"第{i+1}次议价",
                intent="price"
            )
        
        # 现在测试第4轮议价（应该被拒绝）
        reply = ai_reply_engine.generate_reply(
            message="能再便宜点吗？",
            item_info=test_item_info,
            chat_id=test_chat_id,
            cookie_id=test_cookie_id,
            user_id="test_user",
            item_id="test_item"
        )
        
        if reply and "不能再便宜" in reply:
            print(f"   ✅ 议价轮数限制正常工作: {reply}")
        else:
            print(f"   ⚠️  议价消息处理完成，但结果可能不符合预期")
        
    except Exception as e:
        error_msg = str(e)
        if "cannot access local variable 'settings'" in error_msg:
            print(f"   ❌ settings变量作用域问题仍然存在: {error_msg}")
            return False
        else:
            print(f"   ✅ settings变量作用域问题已修复（其他错误: {error_msg[:50]}...）")
    
    # 测试3: 验证settings获取
    print(f"\n3️⃣ 测试settings获取...")
    try:
        settings = db_manager.get_ai_reply_settings(test_cookie_id)
        print(f"   ✅ settings获取成功:")
        print(f"      AI启用: {settings.get('ai_enabled')}")
        print(f"      最大议价轮数: {settings.get('max_bargain_rounds')}")
        print(f"      最大优惠百分比: {settings.get('max_discount_percent')}%")
        
    except Exception as e:
        print(f"   ❌ settings获取失败: {e}")
        return False
    
    return True

def main():
    """主测试函数"""
    print("🚀 AI回复settings变量修复验证")
    print("=" * 50)
    
    # 测试修复
    fix_ok = test_settings_variable_scope()
    
    if fix_ok:
        print("\n" + "=" * 50)
        print("🎉 修复验证成功！")
        print("\n✅ 修复内容:")
        print("   • settings变量作用域问题已解决")
        print("   • 议价轮数限制功能正常")
        print("   • AI回复流程完整")
        print("\n💡 说明:")
        print("   • 由于使用测试API密钥，AI调用会失败")
        print("   • 但不会再出现settings变量错误")
        print("   • 配置真实API密钥后即可正常使用")
    else:
        print("\n❌ 修复验证失败，请检查代码")

if __name__ == "__main__":
    main()