xmlFixer.py

'''

WoT randomizer XML fixer written by KptKosmit91
does changes to Source XML files to work with the randomizer
update tool only - used for updating the randomizer for new WoT versions

'''

import xml.etree.ElementTree as ET
import xmlMethods as xml
from configLoader import Config as conf
import os
import fileMethods as fm


# maps = []

# def loadMaps():
#     folder = conf.mapsPath
#     for n in os.listdir(folder):
#         if n.lower() != "_default_.xml" and n.lower() != "_list_.xml" and n.lower() != "hangar_v3.xml" and n.lower() != "1002_ai_test.xml":
#             maps.append(folder+n)

# loadMaps()

# for m in maps:
#     f = open(m, "r")
#     mapName = m.replace(conf.mapsPath, "").replace("	","")
#     text=f.read()
#     text = text.replace(mapName,"root")

#     f.close()

#     f = open(m, "w")
#     f.write(text)
#     f.close()

def getTankFilePaths():
    tanks = []
    for f in conf.countryFolders:
        folder = conf.tanksPath+conf.countryFolders[f]+"/"
        if os.path.exists(folder):
            for n in os.listdir(folder):
                if n.lower() != "components" and n.lower() != "customization.xml" and n.lower() != "list.xml":
                    tanks.append(folder+n)

        folder = conf.tanksPath.replace("Source/", "SourceOverrides/")+conf.countryFolders[f]+"/"
        if os.path.exists(folder):
            for n in os.listdir(folder):
                if n.lower() != "components" and n.lower() != "customization.xml" and n.lower() != "list.xml":
                    tanks.append(folder+n)

    return tanks

def getCustomizationFilePaths():
    files = []
    folder = conf.customizationPath

    # KK91: this code is stupid, i'm not gonna lie
    if os.path.exists(folder):
        for n in os.listdir(folder):
            if n.endswith(".xml"):
                files.append(folder + n)
            else:
                for n2 in os.listdir(folder + n):
                    if n2.endswith(".xml"):
                        files.append(folder + n + "/" + n2)

    return files

def work(tank):
    print(f"Post-processing {tank}")
    tree = ET.parse(tank)
    root = tree.getroot()

    chassis = root.find("chassis").findall("*")

    xml.insertElement(xml.IsWheeledTag, "false", root)

    for c in chassis:
        nonTrack = c.find("wheels").find("wheel").find("nonTrack")
        if nonTrack != None and nonTrack.text.lower() == "true":
            xml.insertElement(xml.IsWheeledTag, "true", root)
            break

    for t in root.find("turrets0").findall("*"):
        for g in t.find("guns").findall("*"):
            if xml.elementExists("multiGunEffects", g):
                xml.insertElement(xml.IsDoubleGunTag, "true", g)
            else:
                xml.insertElement(xml.IsDoubleGunTag, "false", g)

    newtree = ET.ElementTree(root)

    newtree.write(tank)


print("Running Randomizer XML fixer")

print("\nWill fix tank XMLs. This may take a while")

tanks = getTankFilePaths()

print("Start Pre-processing")
for t in tanks:
    print(f"Pre-processing {t}")

    f = open(t, "r")
    text=f.read()
    text = text.replace("	","").replace("<xmlns:xmlref>http://bwt/xmlref</xmlns:xmlref>","")

    f.close()

    f = open(t, "w")
    f.write(text)
    f.close()


print("\nStart Post-processing")

for t in tanks:
    work(t)

customization = getCustomizationFilePaths()


print("\nStart Fixing Customization files")

for t in customization:
    print(f"Fixing Customization file {t}")

    f = open(t, "r")
    text = f.read()
    text = text.replace("	", "").replace("<xmlns:xmlref>http://bwt/xmlref</xmlns:xmlref>", "")

    f.close()

    f = open(t, "w")
    f.write(text)
    f.close()

print("\nFixing files complete")