commit b2f43210de3476f4e0f9d3f2a4503bf7133d405b
author Lukasz Mrugala <lukaszx.mrugala@intel.com> Fri Apr 19 10:37:20 2024 +0000
committer Alberto Escolar <alberto.escolar.piedras@nordicsemi.no> Tue May 07 09:38:00 2024 +0200
tree b0a1fac89d77aca0532785977ae8a5bb4be33a76
parent 3d00574d335a32e2ef5798f55fc1e687edf3b322

scripts: Loader change

CSafeLoader used instead of yaml.safe_load and SafeLoader.
C implementation is faster.

Signed-off-by: Lukasz Mrugala <lukaszx.mrugala@intel.com>

scripts/list_hardware.py

diff --git a/scripts/list_hardware.py b/scripts/list_hardware.py
index d1f31a4..cdb0e05 100755
--- a/scripts/list_hardware.py
+++ b/scripts/list_hardware.py
@@ -12,14 +12,19 @@
 import yaml
 import re
 
+try:
+    from yaml import CSafeLoader as SafeLoader
+except ImportError:
+    from yaml import SafeLoader
+
 
 SOC_SCHEMA_PATH = str(Path(__file__).parent / 'schemas' / 'soc-schema.yml')
 with open(SOC_SCHEMA_PATH, 'r') as f:
-    soc_schema = yaml.safe_load(f.read())
+    soc_schema = yaml.load(f.read(), Loader=SafeLoader)
 
 ARCH_SCHEMA_PATH = str(Path(__file__).parent / 'schemas' / 'arch-schema.yml')
 with open(ARCH_SCHEMA_PATH, 'r') as f:
-    arch_schema = yaml.safe_load(f.read())
+    arch_schema = yaml.load(f.read(), Loader=SafeLoader)
 
 SOC_YML = 'soc.yml'
 ARCHS_YML_PATH = PurePath('arch/archs.yml')
@@ -35,7 +40,7 @@
             return
 
         try:
-            data = yaml.safe_load(soc_yaml)
+            data = yaml.load(soc_yaml, Loader=SafeLoader)
             pykwalify.core.Core(source_data=data,
                                 schema_data=soc_schema).validate()
         except (yaml.YAMLError, pykwalify.errors.SchemaError) as e:
@@ -188,7 +193,7 @@
 
         if Path(archs_yml).is_file():
             with Path(archs_yml).open('r') as f:
-                archs = yaml.safe_load(f.read())
+                archs = yaml.load(f.read(), Loader=SafeLoader)
 
             try:
                 pykwalify.core.Core(source_data=archs, schema_data=arch_schema).validate()