translation_convertor_rpytopo.py

import os
import re
import shutil
from fileinput import FileInput
from glob import glob

# ATTENTION: there must not be 2 equal key or value
# regex: https://www.w3schools.com/python/python_regex.asp
dict = {
    # search_text : replace_text
    # start
    r"\\" + '"': r"§§§§§§§§",
    # Effect
    r'" nointeract': r' [nointeract]"',
    r'" with fade': r' [withfade]"',
    r'" with dissolve': r' [withdissolve]"',
    r'" with slowdissolve': r' [withslowdissolve]"',
    r'" with hpunch': r' [withhpunch]"',
    r'" with flash': r' [withflash]"',
    r'" with vpunch': r' [withvpunch]"',
    r'" with Dissolve\(2.0\)': r' [withDissolve20]"',
    r'" with Dissolve\(1\)': r' [withDissolve1]"',
    r'" with Dissolve\(.3\)': r' [withDissolvey3]"',
    r'" \(multiple=2\)': r' [multiple2]"',
    r"    # nvl clear": r'msgid "[nvl_clear]"',
    r"    nvl clear": r'msgstr "[nvl_clear]"',
    # first
    r'    # "(.*?)" "(.*?)"': r'msgid "\1 [special_delimiter] \2"',
    r'    "(.*?)" "(.*?)"': r'msgstr "\1 [special_delimiter] \2"',
    r"    #": r"#",
    r'    old "(.*?)"': r'msgid "\1"',
    r'    new "(.*?)"': r'msgstr "\1"',
    # find:    # (.*?) "(.*?)"
    # replace:msgid "[$1] $2"
    r'# (.*?) "(.*?)"': r'msgid "[_\1_] \2"',
    # after
    # find:    (.*?) "(.*?)"
    # replace:msgstr "[$1] $2"
    r'    (.*?) "(.*?)"': r'msgstr "[_\1_] \2"',
    r'# "(.*?)"': r'msgid "\1"',
    r'    "(.*?)"': r'msgstr "\1"',
    # Comment
    r":\n\nmsgid": r":\nmsgid",
    r"rpy:(.*?)\ntranslate": r"rpy:\1 #-#-# translate",
    r"strings:\n\n# ": r"strings: #|#|# # ",
    r"\ntranslate": r"\n# §translate",
    r"updated at (.*?)-(.*?)-(.*?) (.*?):(.*?)\n\n# ": r"updated at \1-\2-\3 \4:\5 #|#|# # ",
    # end
    r"§§§§§§§§": r"\\" + '"',
}


# Creating a function to replace the text
def replacetext(search_text, replace_text, pathFile):
    # Read in the file
    with open(pathFile, "r+", encoding="utf8") as file:
        filedata = file.read()

    # c = re.compile(search_text)

    # Replace the target string
    # filedata = filedata.replace(search_text, replace_text)
    filedata = re.sub(search_text, replace_text, filedata)
    # TODO: to improve
    filedata = re.sub(r"\[_(.*?)\b (.*?)_\]", r"[_\1_s_\2_]", filedata)
    filedata = re.sub(r"\[_(.*?)\b (.*?)_\]", r"[_\1_s_\2_]", filedata)
    filedata = re.sub(r"\[_(.*?)\b (.*?)_\]", r"[_\1_s_\2_]", filedata)
    filedata = re.sub(r"\[_(.*?)\b (.*?)_\]", r"[_\1_s_\2_]", filedata)
    filedata = re.sub(r"\[_(.*?)\b (.*?)_\]", r"[_\1_s_\2_]", filedata)
    filedata = re.sub(r"\[_(.*?)\b (.*?)_\]", r"[_\1_s_\2_]", filedata)
    filedata = re.sub(r"\[_(.*?)\b (.*?)_\]", r"[_\1_s_\2_]", filedata)
    filedata = re.sub(r"\[_(.*?)\b (.*?)_\]", r"[_\1_s_\2_]", filedata)

    # Write the file out again
    with open(pathFile, "w", encoding="utf8") as file:
        file.write(filedata)
    return True


def replaceDictionary(pathFile, dict={}, reverse=False):
    newpathFile = fileRename(pathFile, extension=".po")
    print(pathFile)
    if reverse:
        for item in reversed(list(dict.items())):
            replacetext(pathFile=newpathFile, search_text=item[1], replace_text=item[0])
    else:
        for search_text in dict.keys():
            replacetext(
                pathFile=newpathFile,
                search_text=search_text,
                replace_text=dict[search_text],
            )


def getListFiles(extension=".po"):
    # Get the list of all files and directories
    path = "game/tl/"
    dir_list = glob(path + "/**/*" + extension, recursive=True)
    return dir_list


def rpytopo():
    for path in getListFiles(extension=".rpy"):
        replaceDictionary(path, dict=dict)


def potorpy():
    for path in getListFiles():
        replaceDictionary(path, dict=dict, reverse=True)


def fileRename(pathFile, extension=".rpy"):
    pre, ext = os.path.splitext(pathFile)
    shutil.copyfile(pathFile, pre + extension)
    return pre + extension


rpytopo()