From 4543ee95c6141f56a39580e9b3b2262ab3b2e283 Mon Sep 17 00:00:00 2001
From: Andrew Player <andrewplayer3@gmail.com>
Date: Wed, 28 Feb 2024 12:07:37 -0600
Subject: [PATCH] added setup_directories to utils

---
 src/asf_tools/watermasking/utils.py | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/src/asf_tools/watermasking/utils.py b/src/asf_tools/watermasking/utils.py
index 15b56bd4..511323da 100644
--- a/src/asf_tools/watermasking/utils.py
+++ b/src/asf_tools/watermasking/utils.py
@@ -55,5 +55,15 @@ def remove_temp_files(temp_files: list):
     for file in temp_files:
         try:
             os.remove(file)
-        except:
-            print('Error removing temporary file. Skipping it...')
\ No newline at end of file
+        except Exception as e:
+            print(f'Caught {e} while removing temporary file: {file}. Skipping it...')
+
+
+def setup_directories(dirs: list[str]):
+    """Setup the directories necessary for running the script."""
+    for dir in dirs:
+        try:
+            os.mkdir(dir)
+        except FileExistsError as e:
+            # Directories already exists.
+            pass
\ No newline at end of file