--overwrite bug fix on mkdir logic.

DUNE-DAQ · Feb 14, 2024 · 1031b64 · 1031b64
1 parent 706bb80
commit 1031b64
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/apps/ta_dump.py b/apps/ta_dump.py
@@ -399,7 +399,9 @@ def main():
         plot_iter += 1
         plot_dir = f"{data.run_id}-{data.file_index}_figures_{plot_iter:04}"
     print(f"Saving outputs to ./{plot_dir}/")
-    if not overwrite:  # Can only create new dirs.
+    # If overwriting and it does exist, don't need to make it.
+    # So take the inverse to mkdir.
+    if not (overwrite and os.path.isdir(plot_dir)):
         os.mkdir(plot_dir)
     os.chdir(plot_dir)
 

diff --git a/apps/tp_dump.py b/apps/tp_dump.py
@@ -381,7 +381,9 @@ def main():
         plot_iter += 1
         plot_dir = f"{data.run_id}-{data.file_index}_figures_{plot_iter:04}"
     print(f"Saving outputs to ./{plot_dir}/")
-    if not overwrite:  # Can only create new dirs.
+    # If overwriting and it does exist, don't need to make it.
+    # So take the inverse to mkdir.
+    if not (overwrite and os.path.isdir(plot_dir)):
         os.mkdir(plot_dir)
     os.chdir(plot_dir)