pokeemerald-latin/insert_text.py at master · krkpdf/pokeemerald-latin · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
# NOTE: for some reason the single line "I get it now!" in data/text/birch_speech.inc is not replaced??


import os
import re

progress = {}


script_dir = os.path.dirname(__file__)

folder_path_a = os.path.join("..", "pokeemerald-master-translated") + os.sep # output folder
folder_path_b = os.path.join(".", "text") + os.sep                           # input folder

#print(folder_path_a, folder_path_b)

def extract_literal(line):
    number = line.find(":")
    return (int(line[:number]), line[number+2:])


def extract_quoted_text(line):
    # Find the indices of the first and second quotation marks
    number = line.find(":")
    first_quote_index = line.find('"')
    second_quote_index = line.find('"', first_quote_index + 1)
    if first_quote_index != -1 and second_quote_index != -1 and number != -1:
        # Extract the text between the first and second quotation marks
        return (int(line[:number]), line[first_quote_index + 1:second_quote_index])
    else:
        return (-1, "")  # Return empty string if no quoted text found

def replace_quoted_text(line, text):
    first_quote_index = line.find('"')
    second_quote_index = line.find('"', first_quote_index + 1)

    return line[:first_quote_index+1] + text + line[second_quote_index:]


print("Starting Insertion!")

try :
    for root, _, files in os.walk(folder_path_b):
        for file in files:
            file_path = os.path.join(root, file)
            myfile_path = file_path[len(folder_path_b):]

            # figure out file ending
            file_path_base = os.path.join(folder_path_a, myfile_path)

            end_index = file_path_base.rfind(".")
            file_index = file_path_base.rfind(os.sep)

            mylist = os.listdir(file_path_base[:file_index])
            r = re.compile(file_path_base[file_index+1:end_index+1] + "*")
            myfile = list(filter(r.match, mylist))[0]

            if myfile.rfind(".") == -1 and "maps" in myfile_path:
                myfile = os.path.join(myfile, "scripts.inc")

            file_path_base = os.path.join(file_path_base[:file_index], myfile)

            empty = " "
            print(f"================= {(myfile_path + ' '):=<70}")

            if ".json" in myfile_path:
                with open(file_path_base, 'r', encoding='utf-8') as input_file:
                    content = [line for line in input_file.readlines()]

                with open(file_path, 'r', encoding='utf-8') as input_file:
                    new_content = [extract_literal(line) for line in input_file.readlines()]
                    new_content[-1] = (new_content[-1][0], new_content[-1][1] + "\n")

                    for i, t in new_content:
                        n = i-1
                        content[n] = t


                with open(file_path_base, 'w', encoding='utf-8') as output_file:
                    output_file.write("".join(content))


            elif "ipynb" not in myfile_path:
                with open(file_path_base, 'r', encoding='utf-8') as input_file:
                    content = [line for line in input_file.readlines()]

                with open(file_path, 'r', encoding='utf-8') as input_file:
                    if ".json" in myfile_path:
                        new_content = [extract_literal(line) for line in input_file.readlines()]

                        for i, t in new_content:
                            n = i-1
                            content[n] = t

                    else:
                        new_content = [extract_quoted_text(line) for line in input_file.readlines()]
                        new_content = [c for c in new_content if c[0] != -1]

                        for i, t in new_content:
                            n = i-1
                            content[n] = replace_quoted_text(content[n], t)

                with open(file_path_base, 'w', encoding='utf-8') as output_file:
                    output_file.write("".join(content))

except Exception as e:
    print(e)

print("Insertion Done!")