i3dMapper_toXML.py

# pylama:ignore=W605,E221,C901
"""
        __ ______     __     _   _
        /    /      /    )   /  /|              /       /   ,
-------/----/-------\-------/| /-|----__----__-/----__-/--------__----__-
      /    /         \     / |/  |  /   ) /   /   /   /   /   /   ) /   )
_(___/____/______(____/___/__/___|_(___/_(___/___(___/___/___/___/_(___/_
                                                                      /
    i3dMapper_toXML.py - v1.0.2                                   (_ /

Version History:
 v1.0.2 - Initial Release
"""

import argparse
import xml.etree.ElementTree as ET
import os
import subprocess
import sys


def enter_key_exit():
    """ Exit when enter key pressed """
    print("Press ENTER to close terminal.")
    try:
        input()
    except BaseException:
        pass
    exit()


def clean_path(baseFolder, filename):
    baseFolderTuple = os.path.splitdrive(baseFolder)
    return os.path.normpath(os.path.join(baseFolderTuple[0], baseFolderTuple[1], filename))


print("-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-")
print("  i3dMapper_toXML v1.0.2")
print("    by JTSModding")
print("-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-\n")

parser = argparse.ArgumentParser(description='Export i3d Mapping direct to XML')
parser.add_argument('file', metavar='file', nargs=1, type=argparse.FileType('r', encoding='utf-8'))

if not os.path.isfile('i3dMapper.py'):
    print("ERROR: i3dMapper.py is not in the script folder. Unable to continue")
    enter_key_exit()

try:
    args = parser.parse_args()
except BaseException:
    enter_key_exit()

try:
    if not args.file[0].name.endswith("xml"):
        raise BaseException
    thisShopItemXML = ET.fromstring(args.file[0].read())
    args.file[0].close()
except BaseException:
    print("ERROR: Unable to read / parse file '" + os.path.basename(args.file[0].name) + "'")
    enter_key_exit()

thisShopItemFile   = os.path.basename(args.file[0].name)
thisShopItemFolder = os.path.dirname(os.path.abspath(args.file[0].name))
thisShopItemAbs    = clean_path(thisShopItemFolder, thisShopItemFile)
i3dFileAbs         = ""

try:
    ET.tostring(thisShopItemXML, encoding='unicode')
    for thisTag in thisShopItemXML.findall(".//base/filename"):
        if thisTag.text is None:
            raise Exception("NOT FOUND: i3d File Not Found in XML")
        i3dFileAbs = clean_path(thisShopItemFolder, thisTag.text)

    if not os.path.isfile(i3dFileAbs):
        raise Exception("NOT FOUND: i3d File Not Found")

    i3dMap = subprocess.run([
        sys.executable,
        'i3dMapper.py',
        '--no-pretty-print',
        i3dFileAbs
    ], stdout=subprocess.PIPE)

    i3dMapping = ET.fromstring(i3dMap.stdout)

    thisShopItemI3D = thisShopItemXML.find(".//i3dMappings")

    if thisShopItemI3D is not None:
        print("FOUND: i3dMappings Section, re-writing")
        for child in list(thisShopItemI3D):
            thisShopItemI3D.remove(child)

        for thisMap in i3dMapping.findall(".//i3dMapping"):
            thisShopItemI3D.append(thisMap)
    else:
        print("NOT FOUND: i3dMappings Section, creating one instead")
        thisShopItemXML.append(i3dMapping)

    ET.indent(thisShopItemXML, space='    ')
    storeXML_string = ET.tostring(thisShopItemXML, encoding='unicode')
    storeXML_string = storeXML_string.replace(">", ">")
    storeXML_string = "<?xml version='1.0' encoding='utf-8'?>\n" + storeXML_string

    xmlWriter = open(thisShopItemAbs, "w")
    xmlWriter.write(storeXML_string)
    xmlWriter.close()

    print("SUCCESS: saved mapping to xml file!")

except BaseException as err:
    print("UNRECOVERABLE ERROR: " + str(err))
    enter_key_exit()

if sys.stdout.isatty():
    # Don't pause on finish if we re-directed to a file.
    enter_key_exit()