diff --git a/netgen.py b/netgen.py index 5c15495c..defda8cd 100755 --- a/netgen.py +++ b/netgen.py @@ -100,10 +100,11 @@ def run(args: argparse.Namespace) -> None: else: print(f'Included Networks: {len(in_networks)}') output_file = output.resolve() - if not output_file.parent.exists(): - output_file.parent.mkdir(parents=True) if output_file.is_dir(): print('Can not write to directory.') + sys.exit(1) + if not output_file.parent.exists(): + output_file.parent.mkdir(parents=True) if output_file.exists(): with output_file.open('a', encoding='utf-8') as wfp: wfp.writelines(output_lines) diff --git a/raydata.py b/raydata.py index db56b6f6..796fd988 100755 --- a/raydata.py +++ b/raydata.py @@ -129,10 +129,11 @@ def run(args: argparse.Namespace) -> None: print(line, end='') else: output_file = output.resolve() - if not output_file.parent.exists(): - output_file.parent.mkdir(parents=True) if output_file.is_dir(): print('Can not write to directory.') + sys.exit(1) + if not output_file.parent.exists(): + output_file.parent.mkdir(parents=True) if output_file.exists(): with output_file.open('a', encoding='utf-8') as wfp: wfp.writelines(output_lines)