From aee4ec1b4c9412549ffcf52dacca7a47976d62ea Mon Sep 17 00:00:00 2001 From: Daniel Rojas Date: Fri, 22 May 2020 17:28:11 +0200 Subject: [PATCH] Add output functionality to module --- src/wireviz.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) diff --git a/src/wireviz.py b/src/wireviz.py index b63dae9..254bdad 100644 --- a/src/wireviz.py +++ b/src/wireviz.py @@ -281,3 +281,23 @@ class Cable: s = s + '}' return s + +def output(objects, print_to_screen=False): + with open('output/output.dot','w') as f: + with open('input/header.dot','r') as infile: + for line in infile: + f.write(line) + f.write('\n\n') + + for o in objects: + f.write(o.graphviz() + '\n') + + f.write('\n\n') + with open('input/footer.dot','r') as infile: + for line in infile: + f.write(line) + + if print_to_screen == True: + with open('output/output.dot','r') as f: + for line in f: + print(line)