#!/usr/bin/env python3
"""
翻译功能测试脚本。

测试内容：
1. 翻译提示词配置加载
2. 同步翻译（索引场景）
3. 异步翻译（查询场景）
4. 不同提示词的使用
5. 缓存功能
6. DeepL Context参数使用
"""

import sys
import os
from pathlib import Path

# Add parent directory to path
sys.path.insert(0, str(Path(__file__).parent.parent))

from config import ConfigLoader
from query.translator import Translator
import logging

# Configure logging
logging.basicConfig(
    level=logging.INFO,
    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
)
logger = logging.getLogger(__name__)


def test_config_loading():
    """测试配置加载"""
    print("\n" + "="*60)
    print("测试1: 配置加载")
    print("="*60)
    
    try:
        config_loader = ConfigLoader()
        config = config_loader.load_config()
        
        print(f"✓ 配置加载成功")
        print(f"  翻译服务: {config.query_config.translation_service}")
        print(f"  翻译提示词配置:")
        for key, value in config.query_config.translation_prompts.items():
            print(f"    {key}: {value[:60]}..." if len(value) > 60 else f"    {key}: {value}")
        
        return config
    except Exception as e:
        print(f"✗ 配置加载失败: {e}")
        import traceback
        traceback.print_exc()
        return None


def test_translator_sync(config):
    """测试同步翻译（索引场景）"""
    print("\n" + "="*60)
    print("测试2: 同步翻译（索引场景）")
    print("="*60)
    
    if not config:
        print("✗ 跳过：配置未加载")
        return None
    
    try:
        translator = Translator(
            api_key=config.query_config.translation_api_key,
            use_cache=True,
            glossary_id=config.query_config.translation_glossary_id,
            translation_context=config.query_config.translation_context
        )
        
        # 测试商品标题翻译（使用product_title提示词）
        test_texts = [
            ("蓝牙耳机", "zh", "en", "product_title"),
            ("Wireless Headphones", "en", "zh", "product_title"),
        ]
        
        for text, source_lang, target_lang, prompt_type in test_texts:
            if prompt_type == "product_title":
                if target_lang == "zh":
                    prompt = config.query_config.translation_prompts.get('product_title_zh')
                else:
                    prompt = config.query_config.translation_prompts.get('product_title_en')
            else:
                if target_lang == "zh":
                    prompt = config.query_config.translation_prompts.get('default_zh')
                else:
                    prompt = config.query_config.translation_prompts.get('default_en')
            
            print(f"\n翻译测试:")
            print(f"  原文 ({source_lang}): {text}")
            print(f"  目标语言: {target_lang}")
            print(f"  提示词: {prompt[:50] if prompt else 'None'}...")
            
            result = translator.translate(
                text,
                target_lang=target_lang,
                source_lang=source_lang,
                prompt=prompt
            )
            
            if result:
                print(f"  结果: {result}")
                print(f"  ✓ 翻译成功")
            else:
                print(f"  ⚠ 翻译返回None（可能是mock模式或无API key）")
        
        return translator
        
    except Exception as e:
        print(f"✗ 同步翻译测试失败: {e}")
        import traceback
        traceback.print_exc()
        return None


def test_translator_async(config, translator):
    """测试异步翻译（查询场景）"""
    print("\n" + "="*60)
    print("测试3: 异步翻译（查询场景）")
    print("="*60)
    
    if not config or not translator:
        print("✗ 跳过：配置或翻译器未初始化")
        return
    
    try:
        query_text = "手机"
        target_langs = ['en']
        source_lang = 'zh'
        
        query_prompt = config.query_config.translation_prompts.get('query_zh')
        
        print(f"查询文本: {query_text}")
        print(f"目标语言: {target_langs}")
        print(f"提示词: {query_prompt}")
        
        # 异步模式（立即返回，后台翻译）
        results = translator.translate_multi(
            query_text,
            target_langs,
            source_lang=source_lang,
            context=config.query_config.translation_context,
            async_mode=True,
            prompt=query_prompt
        )
        
        print(f"\n异步翻译结果:")
        for lang, translation in results.items():
            if translation:
                print(f"  {lang}: {translation} (缓存命中)")
            else:
                print(f"  {lang}: None (后台翻译中...)")
        
        # 同步模式（等待完成）
        print(f"\n同步翻译（等待完成）:")
        results_sync = translator.translate_multi(
            query_text,
            target_langs,
            source_lang=source_lang,
            context=config.query_config.translation_context,
            async_mode=False,
            prompt=query_prompt
        )
        
        for lang, translation in results_sync.items():
            print(f"  {lang}: {translation}")
        
    except Exception as e:
        print(f"✗ 异步翻译测试失败: {e}")
        import traceback
        traceback.print_exc()


def test_cache():
    """测试缓存功能"""
    print("\n" + "="*60)
    print("测试4: 缓存功能")
    print("="*60)
    
    try:
        config_loader = ConfigLoader()
        config = config_loader.load_config()
        
        translator = Translator(
            api_key=config.query_config.translation_api_key,
            use_cache=True
        )
        
        test_text = "测试文本"
        target_lang = "en"
        source_lang = "zh"
        prompt = config.query_config.translation_prompts.get('default_zh')
        
        print(f"第一次翻译（应该调用API或返回mock）:")
        result1 = translator.translate(test_text, target_lang, source_lang, prompt=prompt)
        print(f"  结果: {result1}")
        
        print(f"\n第二次翻译（应该使用缓存）:")
        result2 = translator.translate(test_text, target_lang, source_lang, prompt=prompt)
        print(f"  结果: {result2}")
        
        if result1 == result2:
            print(f"  ✓ 缓存功能正常")
        else:
            print(f"  ⚠ 缓存可能有问题")
        
    except Exception as e:
        print(f"✗ 缓存测试失败: {e}")
        import traceback
        traceback.print_exc()


def test_context_parameter():
    """测试DeepL Context参数使用"""
    print("\n" + "="*60)
    print("测试5: DeepL Context参数")
    print("="*60)
    
    try:
        config_loader = ConfigLoader()
        config = config_loader.load_config()
        
        translator = Translator(
            api_key=config.query_config.translation_api_key,
            use_cache=False  # 禁用缓存以便测试
        )
        
        # 测试带context和不带context的翻译
        text = "手机"
        prompt = config.query_config.translation_prompts.get('query_zh')
        
        print(f"测试文本: {text}")
        print(f"提示词（作为context）: {prompt}")
        
        # 带context的翻译
        result_with_context = translator.translate(
            text,
            target_lang='en',
            source_lang='zh',
            prompt=prompt
        )
        print(f"\n带context翻译结果: {result_with_context}")
        
        # 不带context的翻译
        result_without_context = translator.translate(
            text,
            target_lang='en',
            source_lang='zh',
            prompt=None
        )
        print(f"不带context翻译结果: {result_without_context}")
        
        print(f"\n✓ Context参数测试完成")
        print(f"  注意：根据DeepL API，context参数影响翻译但不参与翻译本身")
        
    except Exception as e:
        print(f"✗ Context参数测试失败: {e}")
        import traceback
        traceback.print_exc()


def main():
    """主测试函数"""
    print("="*60)
    print("翻译功能测试")
    print("="*60)
    
    # 测试1: 配置加载
    config = test_config_loading()
    
    # 测试2: 同步翻译
    translator = test_translator_sync(config)
    
    # 测试3: 异步翻译
    test_translator_async(config, translator)
    
    # 测试4: 缓存功能
    test_cache()
    
    # 测试5: Context参数
    test_context_parameter()
    
    print("\n" + "="*60)
    print("测试完成")
    print("="*60)


if __name__ == '__main__':
    main()