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

Separated into cli() and main() to enable build script to call main()

Browse Source
This commit is contained in:
Jason R. Jones 2020-07-24 22:01:56 -04:00
parent 52f79fee40
commit 50ff1a046a
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/wireviz/wireviz.py
View File

@ -203,7 +203,7 @@ def parse(yaml_input: str, return_types: (None, str, Tuple[str]) = None) -> Any:
@click.argument('input_file', required=True)
@click.option('--prepend', '-p', default=None, help='a YAML file containing a library of templates and parts that may be referenced in the `input_file`')
@click.option('--out', '-o', multiple=True, default=['png'], help='specify one or more output types to be generated; currently supports "png" "svg", "html", "csv", "csv_excel", "csv_unix", "tsv", "tsv_excel"')
def main(input_file, prepend, out):
def cli(input_file, prepend, out):
"""
Take a YAML file containing a harness specification and, utilizing GraphViz,
translate that specification into various output formats.
@ -212,8 +212,12 @@ def main(input_file, prepend, out):
$>wireviz my_input_file.yml -o png -o svg
"""
main(input_file, prepend, out)
def main(input_file, prepend, out):
if not os.path.exists(input_file):
print(f'Error: input file {input_file} inaccessible or does not exist, check path')
print(f'Error: input file "{input_file}" inaccessible or does not exist, check path')
sys.exit(1)
with open_file_read(input_file) as fh:
@ -250,4 +254,4 @@ def main(input_file, prepend, out):
if __name__ == '__main__':
main()
cli()