python srt翻译中文to英文

pip install googletrans==4.0.0-rc1

import os
import time
from googletrans import Translator

def translate_srt(input_file, output_file, retry_count=3):
    with open(input_file, 'r', encoding='utf-8') as file:
        content = file.read()

    for attempt in range(retry_count):
        try:
            translator = Translator()
            source_language = translator.detect(content).lang if content else 'en'
            # Set the source language to 'en' (English) if the content is empty

            if source_language != 'zh-CN':
                # If the detected language is not Chinese, specify it explicitly
                translator = Translator(src=source_language, dest='en')
            translated_content = translator.translate(content, dest='en').text
            break  # Break out of the loop if translation is successful
        except Exception as e:
            print(f"Translation error for file {input_file} (Attempt {attempt + 1}): {str(e)}")
            time.sleep(5)  # Add a delay before retrying

    else:
        # If all attempts fail, keep the original content
        translated_content = content
        print(f"Translation failed for file {input_file} after {retry_count} attempts.")

    with open(output_file, 'w', encoding='utf-8') as file:
        file.write(translated_content)

def translate_and_create_ensrt_directory(directory_path):
    ensrt_directory = os.path.join(directory_path, 'ensrt')

    if not os.path.exists(ensrt_directory):
        os.makedirs(ensrt_directory)

    for filename in os.listdir(directory_path):
        if filename.endswith('.srt'):
            input_file_path = os.path.join(directory_path, filename)
            output_file_path = os.path.join(ensrt_directory, f'en_{filename}')

            translate_srt(input_file_path, output_file_path)

if __name__ == "__main__":
    target_directory = r"D:\Work_Tools\民间故事"
    translate_and_create_ensrt_directory(target_directory)
    print("翻译并创建ensrt目录完成！")

final

import os
import time
from googletrans import Translator

def translate_srt(input_file, output_file, retry_count=3):
    with open(input_file, 'r', encoding='utf-8') as file:
        content = file.read()

    for attempt in range(retry_count):
        try:
            translator = Translator()
            source_language = translator.detect(content).lang if content else 'en'
            # Set the source language to 'en' (English) if the content is empty

            if source_language != 'zh-CN':
                # If the detected language is not Chinese, specify it explicitly
                translator = Translator(src=source_language, dest='en')
            translated_content = translator.translate(content, dest='en').text
            break  # Break out of the loop if translation is successful
        except Exception as e:
            print(f"Translation error for file {input_file} (Attempt {attempt + 1}): {str(e)}")
            time.sleep(5)  # Add a delay before retrying

    else:
        # If all attempts fail, return None
        print(f"Translation failed for file {input_file} after {retry_count} attempts.")
        return None

    with open(output_file, 'w', encoding='utf-8') as file:
        file.write(translated_content)

    return output_file

def translate_and_create_ensrt_directory(directory_path):
    ensrt_directory = os.path.join(directory_path, 'ensrt')

    if not os.path.exists(ensrt_directory):
        os.makedirs(ensrt_directory)

    for filename in os.listdir(directory_path):
        if filename.endswith('.srt'):
            input_file_path = os.path.join(directory_path, filename)
            output_file_path = os.path.join(ensrt_directory, f'en_{filename}')

            translated_file = translate_srt(input_file_path, output_file_path)

            if translated_file:
                print(f"Translation successful for file {input_file_path}.")
            else:
                print(f"Translation failed for file {input_file_path}. File not added to 'ensrt' directory.")

if __name__ == "__main__":
    target_directory = r"D:\Work_Tools\民间故事"
    translate_and_create_ensrt_directory(target_directory)
    print("翻译并创建ensrt目录完成！")