chg: [main] detect case data_subfolder using sysdiagnose.log

src/sysdiagnose/utils/base.py

@@ -1,11 +1,12 @@
 from abc import ABC, abstractmethod
-import os
-import json
-from pathlib import Path
 from datetime import datetime
-import re
 from functools import cached_property
+from pathlib import Path
 from sysdiagnose.utils.logger import logger
+import glob
+import json
+import os
+import re
 
 
 class SysdiagnoseConfig:
@@ -44,8 +45,13 @@ def __init__(self, module_filename: str, config: SysdiagnoseConfig, case_id: str
 
         self.case_data_folder = config.get_case_data_folder(case_id)
         os.makedirs(self.case_data_folder, exist_ok=True)
-        case_data_folder_dirlist = os.listdir(self.case_data_folder)
-        self.case_data_subfolder = os.path.join(self.case_data_folder, [item for item in case_data_folder_dirlist if 'sysdiagnose_' in item][0])
+
+        # Search for the 'sysdiagnose.log' file and return the parent folder
+        log_files = glob.glob(os.path.join(self.case_data_folder, '**', 'sysdiagnose.log'), recursive=True)
+        if log_files:
+            self.case_data_subfolder = os.path.dirname(log_files[0])
+        else:
+            self.case_data_subfolder = self.case_data_folder
 
         self.case_parsed_data_folder = config.get_case_parsed_data_folder(case_id)
         os.makedirs(self.case_parsed_data_folder, exist_ok=True)

src/sysdiagnose/utils/misc.py

@@ -53,7 +53,7 @@ def load_plist_string_as_json(plist_string: str) -> dict:
     return json_serializable(plist)
 
 
-def load_plist_bytes_as_json(plist_bytes: bytes):
+def load_plist_bytes_as_json(plist_bytes: bytes) -> dict:
     plist = nska_deserialize.deserialize_plist_from_string(plist_bytes, full_recurse_convert_nska=True, format=dict)
     return json_serializable(plist)