Add try except, where esmf files are not ready, preventing errors but still leaving a warning message

Author: minghangli-uni

ESMF_profiling/esmfFileParser.py

@@ -40,6 +40,7 @@
 # ===========================================================================
 import os
 import re
+import warnings
 from esmfRegion import ESMFRegion
 import time
 
@@ -76,22 +77,27 @@ def collect_runtime_tot(
 
 def _list_esmf_files(dir_path, profile_prefix, esmf_summary, summary_profile):
     """Lists ESMF files based on a prefix."""
-    files = os.listdir(dir_path)
-    if not esmf_summary:
-        # ESMF_Profile.xxxx
-        matching_files = [
-            file
-            for file in files
-            if file.startswith(profile_prefix) and file != summary_profile
+    try:
+        files = os.listdir(dir_path)
+        if not esmf_summary:
+            # ESMF_Profile.xxxx
+            matching_files = [
+                file
+                for file in files
+                if file.startswith(profile_prefix) and file != summary_profile
+            ]
+            matching_files.sort(key=lambda x: int(x[len(profile_prefix) :]))
+        else:
+            # ESMF_Profile.summary
+            matching_files = [summary_profile]
+
+        matching_files_path = [
+            os.path.join(dir_path, matching_file) for matching_file in matching_files
         ]
-        matching_files.sort(key=lambda x: int(x[len(profile_prefix) :]))
-    else:
-        # ESMF_Profile.summary
-        matching_files = [summary_profile]
+    except FileNotFoundError:
+        warnings.warn(f"Directory {dir_path} does not exist! Skipping!")
+        matching_files_path = []
 
-    matching_files_path = [
-        os.path.join(dir_path, matching_file) for matching_file in matching_files
-    ]
     return matching_files_path