136 lines
5.5 KiB
Python
136 lines
5.5 KiB
Python
#!/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()) |