Initial commit

2026-01-19 09:51:07 +08:00
commit 9ef82393be
174 changed files with 22285 additions and 0 deletions
@@ -0,0 +1,136 @@
+#!/usr/bin/env python3
+"""
+更新后的测试脚本
+测试新的缓存和智能分块功能
+"""
+
+import asyncio
+import sys
+from pathlib import Path
+
+# 添加 src 目录到路径
+sys.path.insert(0, str(Path(__file__).parent / "src"))
+
+from src.translator import EPUBTranslator
+from src.utils import load_config, setup_logging
+from rich.console import Console
+
+
+async def test_new_features():
+    """测试新功能"""
+    console = Console()
+    
+    epub_file = "The ingenuity gap - Facing the economic, environmental, and other challenges of an increasingly complex and unpredictable future_副本.epub"
+    
+    if not Path(epub_file).exists():
+        console.print(f"[red]文件不存在: {epub_file}[/red]")
+        return
+    
+    try:
+        # 加载配置
+        config = load_config('config/config.json')
+        setup_logging(config)
+        
+        console.print("[bold blue]测试新的 EPUB 翻译功能[/bold blue]\n")
+        
+        # 测试1: 缓存功能
+        console.print("[cyan]1. 测试缓存功能...[/cyan]")
+        
+        # 启用缓存的翻译器
+        translator_with_cache = EPUBTranslator(config, use_cache=True)
+        
+        console.print("[yellow]首次运行（无缓存）...[/yellow]")
+        result1 = await translator_with_cache.translate_epub(epub_file, test_mode=True)
+        
+        console.print("[yellow]第二次运行（应该使用缓存）...[/yellow]")
+        result2 = await translator_with_cache.translate_epub(epub_file, test_mode=True)
+        
+        # 检查缓存命中
+        if (result1.get('preface', {}).get('from_cache') or 
+            result1.get('chapter', {}).get('from_cache')):
+            console.print("[green]✓ 缓存功能正常工作[/green]")
+        else:
+            console.print("[yellow]缓存可能是首次使用[/yellow]")
+        
+        # 测试2: 禁用缓存
+        console.print("\n[cyan]2. 测试禁用缓存...[/cyan]")
+        
+        translator_no_cache = EPUBTranslator(config, use_cache=False)
+        result3 = await translator_no_cache.translate_epub(epub_file, test_mode=True)
+        
+        console.print("[green]✓ 禁用缓存功能正常[/green]")
+        
+        # 测试3: 智能分块
+        console.print("\n[cyan]3. 测试智能分块功能...[/cyan]")
+        
+        from src.epub_parser import EPUBParser
+        from src.text_processor import TextProcessor
+        
+        parser = EPUBParser(epub_file)
+        processor = TextProcessor(config)
+        
+        # 获取第一个章节
+        content_items = parser.extract_translatable_content(['chapter'])
+        if content_items:
+            first_chapter = content_items[0]
+            paragraphs = processor.extract_paragraphs(first_chapter['content'])
+            
+            console.print(f"章节段落数: {len(paragraphs)}")
+            
+            # 测试智能分块
+            chunks = processor.create_smart_chunks(paragraphs)
+            
+            console.print(f"生成翻译块数: {len(chunks)}")
+            console.print(f"配置的块大小: {config['translation']['chunk_size']} 字符")
+            
+            # 显示块统计
+            for i, chunk in enumerate(chunks[:3], 1):  # 只显示前3个
+                total_chars = sum(len(p['text']) for p in chunk)
+                console.print(f"  块 {i}: {len(chunk)} 个段落, {total_chars} 字符")
+            
+            console.print("[green]✓ 智能分块功能正常[/green]")
+        
+        # 测试4: 翻译失败处理
+        console.print("\n[cyan]4. 测试翻译失败处理...[/cyan]")
+        
+        # 这里我们可以模拟一个翻译失败的情况
+        # 通过检查配置中的 never_fallback_to_original 设置
+        never_fallback = config['translation'].get('never_fallback_to_original', True)
+        console.print(f"翻译失败时不回填原文: {'是' if never_fallback else '否'}")
+        
+        if never_fallback:
+            console.print("[green]✓ 翻译失败处理配置正确[/green]")
+        else:
+            console.print("[yellow]⚠ 建议启用 never_fallback_to_original[/yellow]")
+        
+        # 显示缓存统计
+        console.print("\n[cyan]5. 缓存统计信息...[/cyan]")
+        
+        if translator_with_cache.cache:
+            stats = translator_with_cache.cache.get_cache_stats()
+            console.print(f"缓存文件数: {stats.get('total_files', 0)}")
+            console.print(f"缓存大小: {stats.get('total_size_mb', 0)} MB")
+            console.print(f"缓存目录: {stats.get('cache_directory', '')}")
+        
+        console.print("\n[bold green]🎉 所有新功能测试完成！[/bold green]")
+        
+        # 询问是否进行完整翻译
+        console.print("\n[yellow]是否进行完整翻译测试？这将使用缓存加速翻译过程。[/yellow]")
+        response = input("输入 'yes' 继续完整翻译，其他任意键退出: ")
+        
+        if response.lower() == 'yes':
+            console.print("\n[cyan]开始完整翻译（使用缓存）...[/cyan]")
+            output_file = await translator_with_cache.translate_epub(epub_file, test_mode=False)
+            console.print(f"\n[bold green]🎉 翻译完成！[/bold green]")
+            console.print(f"输出文件: {output_file}")
+        else:
+            console.print("[yellow]已跳过完整翻译[/yellow]")
+            
+    except Exception as e:
+        console.print(f"[red]测试过程出错: {e}[/red]")
+        import traceback
+        console.print(traceback.format_exc())
+
+
+if __name__ == "__main__":
+    asyncio.run(test_new_features())