mirror of
https://github.com/sjpiper145/MakerSkillTree.git
synced 2025-10-16 11:52:50 +02:00
93 lines
2.8 KiB
Python
Executable file
93 lines
2.8 KiB
Python
Executable file
#!/bin/env python
|
|
|
|
import argparse
|
|
import os
|
|
import yaml
|
|
|
|
|
|
def load_yaml(file_path):
|
|
with open(file_path) as file:
|
|
return yaml.safe_load(file)
|
|
|
|
|
|
def load_svg_template(file_path):
|
|
with open(file_path) as file:
|
|
return file.read()
|
|
|
|
|
|
def make_text_multiline(text: list, length1: int = 18, length2: int = 25) -> str:
|
|
"""Split long one-line strings into multiple-lines strings with a maximum length."""
|
|
|
|
# To follow the shape of the hexagon the first line should
|
|
# be smaller, maybe to 18 characters.
|
|
|
|
desired_l = length1 # for line 1
|
|
processed_str_len = 0
|
|
|
|
space_indexes = [i for i, char in enumerate(text) if char == " "]
|
|
for i in range(1, len(space_indexes)):
|
|
if space_indexes[i] < desired_l + processed_str_len:
|
|
# substring fit in the line, keep searching
|
|
pass
|
|
else:
|
|
# substring is greater than permissible length
|
|
text[space_indexes[i - 1]] = " " # break line on previous space
|
|
desired_l = length2 # for lines 2, 3, 4, ...
|
|
processed_str_len += space_indexes[i - 1] - 1
|
|
|
|
return "".join(text)
|
|
|
|
|
|
def process_svg(template, data):
|
|
# Replace title and author
|
|
processed = template.replace("{{ title }}", data["title"])
|
|
processed = processed.replace("{{ author }}", data["author"])
|
|
|
|
displacement = (0, 9, 19, 29, 39, 49, 59, 69)
|
|
for i in range(10):
|
|
for j in range(7):
|
|
if i < 9:
|
|
box_number = i + displacement[j]
|
|
elif i == 9:
|
|
box_number = i + displacement[j + 1]
|
|
if j >= 5:
|
|
continue
|
|
value = data["row"][i][j]
|
|
splitted_string = make_text_multiline(list(value))
|
|
placeholder = f"{{{{ box_{box_number:03d} }}}}"
|
|
processed = processed.replace(placeholder, splitted_string)
|
|
|
|
return processed
|
|
|
|
|
|
def save_processed_svg(content, output_path):
|
|
with open(output_path, "w") as file:
|
|
file.write(content)
|
|
|
|
|
|
def main():
|
|
script_path = os.path.dirname(os.path.abspath(__file__))
|
|
svg_template_path = os.path.join(script_path, "skill_tree_template.svg.j2")
|
|
|
|
parser = argparse.ArgumentParser(description="Process SVG template with YAML data.")
|
|
parser.add_argument("input_yaml", help="Path to input YAML file")
|
|
parser.add_argument("output_svg", help="Path to output processed SVG file")
|
|
args = parser.parse_args()
|
|
|
|
# Load YAML data
|
|
yaml_data = load_yaml(args.input_yaml)
|
|
|
|
# Load SVG template
|
|
svg_template = load_svg_template(svg_template_path)
|
|
|
|
# Process SVG
|
|
processed_svg = process_svg(svg_template, yaml_data)
|
|
|
|
# Save processed SVG
|
|
save_processed_svg(processed_svg, args.output_svg)
|
|
|
|
print(f"Processed SVG saved to {args.output_svg}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|