warehack.ing/WireViz
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Refactor #55

Browse Source
This commit is contained in:
Daniel Rojas 2020-07-04 17:16:00 +02:00
parent ebf1e5a6f2
commit 144c99e021
1 changed files with 9 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
src/wireviz/wireviz.py
View File

@ -197,17 +197,16 @@ def parse(yaml_input, file_out=None, generate_bom=False, return_types: (None, st
harness.output(filename=file_out, fmt=('png', 'svg'), gen_bom=generate_bom, view=False)
if return_types is not None:
# gather the harness data and return it as a single data type
if isinstance(return_types, str):
if return_types.lower() == 'png':
return harness.png
else:
# gather the harness data and return it as a series of tuples
returns = []
if isinstance(return_types, str): # only one return type speficied
return_types = [return_types]
return_types = [t.lower() for t in return_types]
if 'png' in return_types:
returns.append(harness.png)
return tuple(returns)
return tuple(returns) if len(returns) != 1 else returns
def parse_file(yaml_file, file_out=None, generate_bom=False):