From 07b919edcc188efd6dc3c0caf39b44df1dd9ae7e Mon Sep 17 00:00:00 2001
From: paniy <louisowen@me.com>
Date: Thu, 16 Jan 2025 00:35:23 +0800
Subject: [PATCH] Update translate.py

---
 translate.py | 78 ++++++++++++++++++++--------------------------------
 1 file changed, 30 insertions(+), 48 deletions(-)

diff --git a/translate.py b/translate.py
index 27779e3..9bb1559 100644
--- a/translate.py
+++ b/translate.py
@@ -3,7 +3,6 @@ import time
 import json
 import os
 from googletrans import Translator, LANGUAGES
-import concurrent.futures
 
 def load_translation_cache(cache_file):
     if os.path.exists(cache_file):
@@ -15,70 +14,53 @@ def save_translation_cache(cache_file, translations):
     with open(cache_file, 'w', encoding='utf-8') as f:
         json.dump(translations, f, ensure_ascii=False, indent=2)
 
-def batch_translate(translator, texts, src_lang, dest_lang):
-    translations = {}
-    max_retries = 3
-    with concurrent.futures.ThreadPoolExecutor() as executor:
-        future_to_text = {executor.submit(translate_with_retry, translator, text, src_lang, dest_lang, max_retries): text for text in texts}
-        for future in concurrent.futures.as_completed(future_to_text):
-            text = future_to_text[future]
-            try:
-                translation = future.result()
-                translations[text] = translation
-            except Exception as e:
-                print(f"Translation failed for: {text}")
-                print(f"Error: {e}")
-    return translations
-
-def translate_with_retry(translator, text, src_lang, dest_lang, max_retries):
-    for attempt in range(max_retries):
-        try:
-            time.sleep(2)  # 增加延迟以避免请求过快
-            translation = translator.translate(text, src=src_lang, dest=dest_lang)
-            return translation.text
-        except Exception as e:
-            if attempt == max_retries - 1:
-                raise e
-            print(f"Retry {attempt + 1}/{max_retries} for: {text}")
-            time.sleep(5)  # 重试前等待更长时间
-
 def translate_po_file(input_file, output_file, target_lang):
     cache_file = f'po/cache_{target_lang}.json'
     translations = load_translation_cache(cache_file)
     
+    # 使用最新的 Translator，并设置更可靠的服务 URL
     translator = Translator(service_urls=['translate.google.com'])
     
     with open(input_file, 'r', encoding='utf-8') as f:
         content = f.read()
     
-    pattern = r'msgid "(.*?)"\nmsgstr "(.*?)"'
+    pattern = r'msgid "(.*?)"\nmsgstr ""'
     matches = re.finditer(pattern, content)
     
-    texts_to_translate = []
+    updated = False
     for match in matches:
         chinese_text = match.group(1)
         if chinese_text and any('\u4e00' <= char <= '\u9fff' for char in chinese_text):
-            if chinese_text not in translations:
-                texts_to_translate.append(chinese_text)
-    
-    if texts_to_translate:
-        new_translations = batch_translate(translator, texts_to_translate, 'zh-cn', target_lang)
-        translations.update(new_translations)
-        updated = True
-    else:
-        updated = False
-    
-    for match in matches:
-        chinese_text = match.group(1)
-        existing_translation = match.group(2)
-        translated_text = translations.get(chinese_text, existing_translation)
-        
-        if existing_translation != translated_text:
+            if chinese_text in translations:
+                translated_text = translations[chinese_text]
+                print(f"Using cached translation [{target_lang}]: {chinese_text} -> {translated_text}")
+            else:
+                try:
+                    # 增加重试机制
+                    max_retries = 3
+                    for attempt in range(max_retries):
+                        try:
+                            time.sleep(2)  # 增加延迟以避免请求过快
+                            translation = translator.translate(chinese_text, src='zh-cn', dest=target_lang)
+                            translated_text = translation.text
+                            translations[chinese_text] = translated_text
+                            updated = True
+                            print(f"New translation [{target_lang}]: {chinese_text} -> {translated_text}")
+                            break
+                        except Exception as e:
+                            if attempt == max_retries - 1:
+                                raise e
+                            print(f"Retry {attempt + 1}/{max_retries} for: {chinese_text}")
+                            time.sleep(5)  # 重试前等待更长时间
+                except Exception as e:
+                    print(f"Translation failed for: {chinese_text}")
+                    print(f"Error: {e}")
+                    continue
+                
             content = content.replace(
-                f'msgid "{chinese_text}"\nmsgstr "{existing_translation}"',
+                f'msgid "{chinese_text}"\nmsgstr ""',
                 f'msgid "{chinese_text}"\nmsgstr "{translated_text}"'
             )
-            updated = True
     
     if updated:
         save_translation_cache(cache_file, translations)