repo structure

TPF/scenario_exe_parser.py

@@ -0,0 +1,103 @@
+import xml.etree.ElementTree as ET
+import os
+import sys
+import json
+from collections import defaultdict
+from pathlib import Path
+
+# Get the directory of the current Python file
+current_directory = os.path.dirname(os.path.abspath(__file__))
+COMPONENT_DIR = os.path.join(current_directory, "Sensor_hub_repo", "components")
+
+def finalize_output(data_obj):
+    # Convert defaultdict to standard dict recursively
+    # This removes the <lambda> and <class 'list'> metadata
+    standard_dict = json.loads(json.dumps(data_obj))
+    
+    # Print ONLY the JSON string to stdout
+    #print(json.dumps(standard_dict, indent=4))
+    return standard_dict
+
+def parse_test_scenario(xml_file_path):
+    """
+    Parses a test scenario XML file and extracts the configuration and all 
+    test case IDs mapped to their execution commands.
+
+    Args:
+        xml_file_path (str): The path to the XML file to parse.
+
+    Returns:
+        dict: A dictionary in the format:
+              {
+                  'config': <config_value>, 
+                  'test_cases': {
+                      <test_case_id>: <test_exec_command>,
+                      ...
+                  }
+              }
+              Returns an empty dictionary on error.
+    """
+    if not os.path.exists(xml_file_path):
+        print(f"Error: File not found at '{xml_file_path}'")
+        return {}
+
+    try:
+        # 1. Parse the XML file
+        tree = ET.parse(xml_file_path)
+        root = tree.getroot()
+    except ET.ParseError as e:
+        print(f"Error: Failed to parse XML file. Details: {e}")
+        return {}
+    except Exception as e:
+        print(f"An unexpected error occurred during file parsing: {e}")
+        return {}
+
+    # Initialize the final structured output
+    parsed_data = {
+        'config': '',
+        'test_cases': {}
+    }
+
+    # 2. Extract the mandatory <config> value
+    config_element = root.find('config')
+    if config_element is not None and config_element.text:
+        parsed_data['config'] = config_element.text.strip()
+    
+    # 3. Iterate over all <test_case> elements and extract ID and Exec
+    for tc in root.findall('test_case'):
+        tc_id_element = tc.find('test_case_id')
+        tc_exec_element = tc.find('test_exec')
+
+        # Use strip() and check against None for safety, even if validation passed
+        tc_id = tc_id_element.text.strip() if tc_id_element is not None and tc_id_element.text else "UNKNOWN_ID"
+        tc_exec = tc_exec_element.text.strip() if tc_exec_element is not None and tc_exec_element.text else "UNKNOWN_EXEC"
+        
+        # Add to the test_cases dictionary
+        parsed_data['test_cases'][tc_id] = tc_exec
+
+    return parsed_data
+
+if __name__ == "__main__":
+    # Define a default path to test against
+    default_test_file = 'sample_scenario.xml'
+    
+    # Allow passing the file path as a command-line argument for flexibility
+    file_to_check = sys.argv[1] if len(sys.argv) > 1 else print({})
+    file_path = os.path.join(COMPONENT_DIR, file_to_check)
+
+    print(f"--- XML Test Scenario Parser ---")
+    print(f"Parsing file: {file_to_check}\n")
+
+    # Run the parser
+    scenario_data = parse_test_scenario(file_path)
+    
+    # Print results
+    # if scenario_data:
+    #     print("✅ Parsing Successful. Extracted Data Structure:")
+    #     print(f"CONFIG: {scenario_data['config']}")
+    #     print("\nTEST CASES:")
+        # for test_id, command in scenario_data['test_cases'].items():
+        #     print(f"  - {test_id}:\n    '{command}'")
+    
+    print(finalize_output(scenario_data))
+    #return finalize_output(scenario_data['test_cases'])