Avoid a crash when output dir exists

ESMValGroup · Oct 14, 2022 · 416b028 · 416b028
1 parent 0f5bda8
commit 416b028
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/esmvalcore/_main.py b/esmvalcore/_main.py
@@ -402,9 +402,17 @@ def run(self,
         cfg = read_config_user_file(config_file, recipe.stem, kwargs)
 
         # Create run dir
-        if os.path.exists(cfg['run_dir']):
-            print("ERROR: run_dir {} already exists, aborting to "
-                  "prevent data loss".format(cfg['output_dir']))
+        run_dir = Path(cfg['run_dir'])
+        if run_dir.exists():
+            suffix = 1
+            run_dir = Path(f"{run_dir}-{suffix}")
+            while run_dir.exists():
+                suffix += 1
+                run_dir = Path(f"{str(run_dir).rsplit('-', 1)[0]}-{suffix}")
+                if suffix > 1000:
+                    print("ERROR: run_dir {} already exists, aborting to "
+                          "prevent data loss".format(cfg['output_dir']))
+        cfg['run_dir'] = str(run_dir)
         os.makedirs(cfg['run_dir'])
 
         # configure logging