From fce66f47250eb8376a3bf50ef04aa1d6d1f0f03f Mon Sep 17 00:00:00 2001 From: darkclip Date: Fri, 11 Apr 2025 05:47:58 +0800 Subject: [PATCH] update --- netgen.py | 5 +++-- raydata.py | 5 +++-- 2 files changed, 6 insertions(+), 4 deletions(-) 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)