better import name

This commit is contained in:
Mylloon 2024-04-17 21:23:26 +02:00
parent 9d9cacb83d
commit e90064466d
Signed by: Anri
GPG key ID: A82D63DFF8D1317F

View file

@ -2,7 +2,9 @@ from argparse import ArgumentParser
from csv import reader as csv_reader from csv import reader as csv_reader
from pprint import pformat from pprint import pformat
from xml.dom.minidom import parseString as xml_parser from xml.dom.minidom import parseString as xml_parser
from xml.etree import ElementTree from xml.etree.ElementTree import Element as XMLElement
from xml.etree.ElementTree import SubElement as XMLSubElement
from xml.etree.ElementTree import tostring as element_stringifier
class System: class System:
@ -49,28 +51,28 @@ class System:
def __repr__(self): def __repr__(self):
return pformat(self.__dict__, compact=True, width=120, sort_dicts=False) return pformat(self.__dict__, compact=True, width=120, sort_dicts=False)
def to_xml(self) -> ElementTree.Element: def to_xml(self) -> XMLElement:
"""Convert the current system into an XML element""" """Convert the current system into an XML element"""
system = ElementTree.Element("lsystem") system = XMLElement("lsystem")
base = ElementTree.SubElement(system, "name") base = XMLSubElement(system, "name")
base.text = self.name base.text = self.name
base = ElementTree.SubElement(system, "base") base = XMLSubElement(system, "base")
base.text = "".join(self.base) base.text = "".join(self.base)
axiom = ElementTree.SubElement(system, "axiom") axiom = XMLSubElement(system, "axiom")
axiom.text = self.axiom axiom.text = self.axiom
substitutions = ElementTree.SubElement(system, "substitutions") substitutions = XMLSubElement(system, "substitutions")
for member, substitution in self.substitutions: for member, substitution in self.substitutions:
sub_element = ElementTree.SubElement(substitutions, "substitution") sub_element = XMLSubElement(substitutions, "substitution")
sub_element.set("member", member) sub_element.set("member", member)
sub_element.text = substitution sub_element.text = substitution
interpretations = ElementTree.SubElement(system, "interpretations") interpretations = XMLSubElement(system, "interpretations")
for member, interpretation in self.interpretations: for member, interpretation in self.interpretations:
inter_element = ElementTree.SubElement(interpretations, "interpretation") inter_element = XMLSubElement(interpretations, "interpretation")
inter_element.set("member", member) inter_element.set("member", member)
inter_element.text = interpretation inter_element.text = interpretation
@ -96,10 +98,10 @@ def data_reader(path: str, delimiter: str = ","):
return res return res
def lsystems_xml(systems: list[System]) -> ElementTree.Element: def lsystems_xml(systems: list[System]) -> XMLElement:
"""Convert list of L-system structure into XML""" """Convert list of L-system structure into XML"""
root = ElementTree.Element("lsystems") root = XMLElement("lsystems")
for system in systems: for system in systems:
root.append(system.to_xml()) root.append(system.to_xml())
@ -125,7 +127,7 @@ if __name__ == "__main__":
xml = lsystems_xml(lsystems) xml = lsystems_xml(lsystems)
# Output XML # Output XML
dom = xml_parser(ElementTree.tostring(xml)) dom = xml_parser(element_stringifier(xml))
pretty_xml = dom.toprettyxml(encoding="UTF-8").decode() pretty_xml = dom.toprettyxml(encoding="UTF-8").decode()
if args.output: if args.output:
with open(args.output, "w") as output_file: with open(args.output, "w") as output_file: