From 9de33b49bd10988ff176bfbd23d5f2e1ebcdfafe Mon Sep 17 00:00:00 2001 From: "Jason R. Jones" Date: Mon, 13 Jul 2020 12:58:33 -0400 Subject: [PATCH] Make file output more consistently behaved. --- src/wireviz/Harness.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/src/wireviz/Harness.py b/src/wireviz/Harness.py index 28f8b74..ca9f8e6 100644 --- a/src/wireviz/Harness.py +++ b/src/wireviz/Harness.py @@ -9,6 +9,7 @@ from wireviz.wv_helper import awg_equiv, mm2_equiv, tuplelist2tsv, nested, \ graphviz_line_breaks, remove_line_breaks from collections import Counter from typing import List +from pathlib import Path class Harness: @@ -279,13 +280,13 @@ class Harness: data.seek(0) return data.read() - def output(self, filename, directory='_output', view=False, cleanup=True, fmt=('pdf', )): + def output(self, filename: (str, Path), view=False, cleanup=True, fmt=('pdf', )): # graphical output graph = self.create_graph() for f in fmt: graph.format = f - graph.render(filename=filename, directory=directory, view=view, cleanup=cleanup) - graph.save(filename=f'{filename}.gv', directory=directory) + graph.render(filename=filename, view=view, cleanup=cleanup) + graph.save(filename=f'{filename}.gv') # bom output bom_list = self.bom_list() with open(f'{filename}.bom.tsv', 'w') as file: