translate.py

#
# AUTHOR: Zappastuff - 2021-APR
#
# REQUIRES:
#   - Python 3.9.x
#   - deep_translator library (pip3 install deep_translator)
#

import argparse
import os
import re
import sys
from deep_translator import GoogleTranslator

CHARS_MAX = 4500
SEPARATOR = "\x0D"

def parse_command_line():
    my_parser = argparse.ArgumentParser(description='Translates Solasta game terms')
    my_parser.add_argument('input_folder',
                        type=str,
                        help='input folder')
    my_parser.add_argument('-c', '--code',
                        type=str,
                        required=True,
                        help='language code')

    return my_parser.parse_args()


def display_progress(count, total, status=''):
    bar_len = 80
    filled_len = int(round(bar_len * count / float(total)))

    percents = round(100.0 * count / float(total), 1)
    bar = '=' * filled_len + '-' * (bar_len - filled_len)

    sys.stdout.write('[%s] %s%s (%s) (%i)\r' % (bar, percents, '%', status, count))
    sys.stdout.flush() 


def unpack_record(record):
    term = ""
    text = ""
    try:
        (term, text) = record.split("=", 1)
        text = text.strip()
    except:
        term = record

    return term, text if text != "" else "EMPTY"


def get_records(filename):
    try:
        line_count = 0
        line_total = sum(1 for line in open(filename))
        with open(filename, "rt", encoding="utf-8") as f:
            record = "\n"
            while record:
                display_progress(line_count, line_total, filename)
                record = f.readline()
                line_count += 1
                if record: yield unpack_record(record)
            print()

    except FileNotFoundError:
        print("ERROR")


def translate_text(text, code):
    text = text.replace("\\n", "{99}")
    if len(text) > 3 and len(text) <= CHARS_MAX:
        translated = GoogleTranslator(source="auto", target=code).translate(text) 
    else:
        translated = text
    translated = translated.replace("{99}", "\\n")

    return translated


# kiddos: this is ugly ;-)
r2 = re.compile(r"<i> (.*?) </i>")
r3 = re.compile(r"<b> (.*?) </b>")
r4 = re.compile(r"< (.*?)>")
r5 = re.compile(r"<(.*?) >")

def fix_translated_format(text):
    text = r2.sub(r"<i>\1</i>", text)
    text = r3.sub(r"<b>\1</b>", text) 
    text = r4.sub(r"<\1>", text)
    text = r5.sub(r"<\1>", text) 
    return text


def translate_file(input_file, output_file, code):
    with open(output_file, "wt", encoding="utf-8") as f:

        for term, text in get_records(input_file):
            translated = translate_text(text, code)
            fixed = fix_translated_format(translated)
            f.write(f"{term}={fixed}\n")


def translate_folder(root_folder_name, folder_name, code):
    root_output_name = code if code != "pt" else "pt-BR"
    os.mkdir(folder_name.replace(root_folder_name, root_output_name))
    for filename in os.listdir(folder_name):
        input_file = os.path.join(folder_name, filename)
        if os.path.isfile(input_file):
            output_file = f"{root_output_name}\\{input_file[3:-7]}-{root_output_name}.txt"
            translate_file(input_file, output_file, code)
        else:
            translate_folder(root_folder_name, input_file, code)

def main():
    # args = parse_command_line()
    # translate_folder(args.input_folder, args.input_folder, args.code)

    translate_folder("en", "en", "pt")
    translate_folder("en", "en", "es")
    translate_folder("en", "en", "fr")
    translate_folder("en", "en", "de")
    translate_folder("en", "en", "it")

if __name__ == "__main__":
    main()