From e3d78f3e01fdaa2ba424a5724f481c9bb48321b1 Mon Sep 17 00:00:00 2001 From: Daniel Rojas Date: Sun, 24 May 2020 13:41:18 +0200 Subject: [PATCH] Improve output function --- src/wireviz.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/src/wireviz.py b/src/wireviz.py index 323c045..a4096bb 100644 --- a/src/wireviz.py +++ b/src/wireviz.py @@ -146,10 +146,11 @@ class Harness: return dot - def output(self, filename, format='pdf', view=True): + def output(self, filename, directory='_output', view=False, cleanup=True, format='pdf'): d = self.create_graph() - d.format = format - d.render(filename, view=view) + for f in format: + d.format = f + d.render(filename=filename, directory=directory, view=view, cleanup=cleanup) class Node: