tag-game/lua_update_script_unmodified.py at master · grintoul1/tag-game · 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
def get_map_values(map_file):
    values = {}
    with open(map_file, 'r') as content:
        for line in content:
            if "gPlayerPartyCount" in line:
                values['gPlayerPartyCount'] = line.strip().split()[0]
            elif "gPlayerParty" in line:
                values['gPlayerParty'] = line.strip().split()[0]
            elif "gPokemonStorage" in line and not "gPokemonStoragePtr" in line:
                values['gPokemonStorage'] = line.strip().split()[0]
            elif "gSpeciesInfo" in line:
                values['gSpeciesInfo'] = line.strip().split()[0]

    return values


def replace_values(lua_script, new_values):
    party_count = f"local partyCount={new_values['gPlayerPartyCount']} -- gPlayerPartyCount\n"
    party_loc = f"local partyloc={new_values['gPlayerParty']} -- gPlayerParty\n"
    storage_loc = f"local storageLoc={new_values['gPokemonStorage']} -- gPokemonStorage\n"
    species_info = f"local speciesInfo={new_values['gSpeciesInfo']} -- gSpeciesInfo\n"

    with open(lua_script, 'r') as content:
        lines = content.readlines()

    for line_no, line in enumerate(lines):
        if 'gPlayerPartyCount' in line:
            lines[line_no] = lines[line_no].replace(line, party_count)
        elif 'gPlayerParty' in line:
            lines[line_no] = lines[line_no].replace(line, party_loc)
        elif 'gSpeciesInfo' in line:
            lines[line_no] = lines[line_no].replace(line, storage_loc)
        elif 'gPokemonStorage' in line:
            lines[line_no] = lines[line_no].replace(line, species_info)

    with open(lua_script, 'w') as content:
        content.writelines(lines)


if __name__ == '__main__':
    new_values = get_map_values("pokeemerald.map")
    replace_values("calc_export.lua", new_values)