mirror of
https://github.com/tradecatlabs/vibe-coding-cn.git
synced 2026-08-01 05:07:45 +00:00
tukuaiai
91 lines
3.1 KiB
Python
Executable File
91 lines
3.1 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
"""
|
|
轻量级批量翻译工具：将 Markdown/文本从一种语言翻译到另一种语言。
|
|
默认使用 deep-translator 封装的 Google Translator，自带简单的代码块保护。
|
|
用法示例：
|
|
python translate.py --input ../../i18n/zh/README.md --src-lang zh --tgt-lang en --output ../../i18n/en/README.md
|
|
"""
|
|
|
|
import argparse
|
|
import os
|
|
import sys
|
|
from typing import List
|
|
|
|
try:
|
|
from deep_translator import GoogleTranslator
|
|
except ImportError as exc: # pragma: no cover - 运行前请安装依赖
|
|
sys.stderr.write("[错误] 缺少依赖 deep-translator，请先 `pip install -r requirements.txt`\n")
|
|
raise
|
|
|
|
|
|
def translate_blocks(lines: List[str], src: str, tgt: str) -> List[str]:
|
|
"""逐段翻译，保持代码块原样，减少上下文丢失。"""
|
|
translated: List[str] = []
|
|
in_code = False
|
|
buffer: List[str] = []
|
|
translator = GoogleTranslator(source=src, target=tgt)
|
|
|
|
def flush_buffer():
|
|
if not buffer:
|
|
return
|
|
text = "\n".join(buffer)
|
|
try:
|
|
result = translator.translate(text)
|
|
except Exception as exc: # pragma: no cover
|
|
raise RuntimeError(f"翻译失败: {exc}") from exc
|
|
translated.extend(result.split("\n"))
|
|
buffer.clear()
|
|
|
|
for line in lines:
|
|
if line.strip().startswith("```"):
|
|
flush_buffer()
|
|
in_code = not in_code
|
|
translated.append(line)
|
|
continue
|
|
if in_code:
|
|
translated.append(line)
|
|
continue
|
|
# 空行作为段落分割，先刷新再保留空行
|
|
if not line.strip():
|
|
flush_buffer()
|
|
translated.append(line)
|
|
continue
|
|
buffer.append(line)
|
|
|
|
flush_buffer()
|
|
return translated
|
|
|
|
|
|
def main() -> int:
|
|
parser = argparse.ArgumentParser(description="批量翻译文本/Markdown，保护代码块")
|
|
parser.add_argument("--input", required=True, help="源文件路径")
|
|
parser.add_argument("--output", required=True, help="目标文件路径")
|
|
parser.add_argument("--src-lang", required=True, help="源语言代码，如 zh")
|
|
parser.add_argument("--tgt-lang", required=True, help="目标语言代码，如 en")
|
|
parser.add_argument("--overwrite", action="store_true", help="允许覆盖已有输出文件")
|
|
args = parser.parse_args()
|
|
|
|
if not os.path.isfile(args.input):
|
|
sys.stderr.write(f"[错误] 源文件不存在: {args.input}\n")
|
|
return 1
|
|
if os.path.exists(args.output) and not args.overwrite:
|
|
sys.stderr.write(f"[错误] 目标文件已存在，加 --overwrite 才会覆盖: {args.output}\n")
|
|
return 1
|
|
|
|
with open(args.input, "r", encoding="utf-8") as f:
|
|
lines = f.read().splitlines()
|
|
|
|
translated = translate_blocks(lines, args.src_lang, args.tgt_lang)
|
|
|
|
os.makedirs(os.path.dirname(args.output), exist_ok=True)
|
|
with open(args.output, "w", encoding="utf-8") as f:
|
|
f.write("\n".join(translated) + "\n")
|
|
|
|
print(f"[完成] {args.input} -> {args.output} ({args.src_lang} -> {args.tgt_lang})")
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__": # pragma: no cover
|
|
sys.exit(main())
|