Paris7/l-systems
Archived
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add proper argument parser

Browse source
This commit is contained in:
Mylloon 2024-04-17 15:45:37 +02:00
parent 5d142d87dd
commit 7103a296ba
Signed by: Anri
GPG key ID: A82D63DFF8D1317F
1 changed files with 18 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

29
converter.py
View file

@ -1,7 +1,6 @@
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 pprint import pp as pprint
from sys import argv, stderr
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 import ElementTree
@ -38,7 +37,6 @@ class System:
represented as a couple (member, interpretation) represented as a couple (member, interpretation)
""" """
# TODO: Est-ce que on garde ça ??
# Interpretation of extra symbols added if necessary # Interpretation of extra symbols added if necessary
extra_symbols = ["[", "]"] extra_symbols = ["[", "]"]
for _, substitution in substitutions: for _, substitution in substitutions:
@ -107,18 +105,27 @@ def lsystems_xml(systems: list[System]) -> ElementTree.Element:
if __name__ == "__main__": if __name__ == "__main__":
if len(argv) != 2: parser = ArgumentParser(description="Generate XML representation of L-systems")
pprint(f"Syntax error:\n\tpython {argv[0]} file.csv", stream=stderr) parser.add_argument(
exit(1) "file", metavar="FILE", help="CSV file containing L-system data"
# TODO: Check if we have to dump to stdout or we write into a file )
# use `argparse` for cli arguments parser.add_argument(
"-o", "--output", metavar="OUTPUT_FILE", help="Specify output file for XML"
)
args = parser.parse_args()
# Read data # Read data
lsystems = data_reader(argv[1]) lsystems = data_reader(args.file)
# Generate XML # Generate XML
xml = lsystems_xml(lsystems) xml = lsystems_xml(lsystems)
# Print XML # Output XML
dom = xml_parser(ElementTree.tostring(xml)) dom = xml_parser(ElementTree.tostring(xml))
print(dom.toprettyxml(encoding="UTF-8").decode()) pretty_xml = dom.toprettyxml(encoding="UTF-8").decode()
if args.output:
with open(args.output, "w") as output_file:
output_file.write(pretty_xml)
else:
print(pretty_xml)