Update converter

This commit is contained in:
Shadow15510 2021-11-01 10:36:20 +01:00
parent f189b3789e
commit 0e4ca5ef63
1 changed files with 18 additions and 3 deletions

View File

@ -3,7 +3,7 @@ from sys import argv
import xmltodict
def convert_to_string(filename, *doors):
def convert_to_string(filename, doors, misc):
output_filename, extension = filename.split('.')
char_list = " !\"#$%&'()*+,-./0123456789:;<=>?@ABCDEFGHIJKLMNOPQRSTUVWXYZ[\\]^_`abcdefghijklmnopqrstuvwxyz{|}~ "
@ -18,17 +18,32 @@ def convert_to_string(filename, *doors):
data = [[int(char_id) for char_id in line.split(",") if char_id] for line in data]
doors_coords = []
misc_coords = []
output = r""
for line_index, line in enumerate(data):
for char_index, char_id in enumerate(line):
output += char_list[char_id]
if char_list[char_id] in doors:
doors_coords.append(f"\t({char_index}, {line_index}, ,0, 0),")
if char_list[char_id] in misc:
misc_coords.append(f"# {char_list[char_id]} : ({char_index}; {line_index})")
output += "\n"
doors_coords = "\n".join(doors_coords)
misc_coords = "\n".join(misc_coords)
with open(f"{output_filename}.py", "w") as file:
file.write(f"{output_filename} = (r\"\"\"\n{output[:-1]}\"\"\",\n{doors_coords}\n)")
file.write(f"{output_filename} = (r\"\"\"\n{output[:-1]}\"\"\",\n{doors_coords}\n)\n\n{misc_coords}")
convert_to_string(*argv[1:])
filename, doors, misc = argv[1], "", ""
for arg in argv[2:]:
if arg.startswith("door"):
doors = arg.split("=", 1)[1]
elif arg.startswith("misc"):
misc = arg.split("=", 1)[1]
convert_to_string(filename, doors, misc)