Merge pull request #7 from espressif/test/from-idf-build-apps

Updated the test and added BoolStmt to __all__.

Author: hfudev

esp_bool_parser/__init__.py

@@ -9,9 +9,42 @@
 
 __version__ = '0.1.1'
 
-from .bool_parser import parse_bool_expr, register_addition_attribute
+import importlib
+
+from .bool_parser import BoolStmt, parse_bool_expr, register_addition_attribute
+from .utils import lazy_load
+
+__getattr__ = lazy_load(
+    importlib.import_module(__name__),
+    {
+        'parse_bool_expr': parse_bool_expr,
+        'register_addition_attribute': register_addition_attribute,
+        'BoolStmt': BoolStmt,
+    },
+    {
+        'IDF_PATH': '.constants',
+        'SUPPORTED_TARGETS': '.constants',
+        'PREVIEW_TARGETS': '.constants',
+        'ALL_TARGETS': '.constants',
+        'IDF_VERSION_MAJOR': '.constants',
+        'IDF_VERSION_MINOR': '.constants',
+        'IDF_VERSION_PATCH': '.constants',
+        'IDF_VERSION': '.constants',
+        'SOC_HEADERS': '.soc_header',
+    },
+)
 
 __all__ = [
+    'ALL_TARGETS',
+    'IDF_PATH',
+    'IDF_VERSION',
+    'IDF_VERSION_MAJOR',
+    'IDF_VERSION_MINOR',
+    'IDF_VERSION_PATCH',
+    'PREVIEW_TARGETS',
+    'SOC_HEADERS',
+    'SUPPORTED_TARGETS',
+    'BoolStmt',
     'parse_bool_expr',
     'register_addition_attribute',
 ]

esp_bool_parser/utils.py

@@ -1,12 +1,63 @@
 # SPDX-FileCopyrightText: 2025 Espressif Systems (Shanghai) CO LTD
 # SPDX-License-Identifier: Apache-2.0
 
+import importlib
 import typing as t
 
 from packaging.version import (
     Version,
 )
 
+_ModuleType: t.Any = type(importlib)
+
+
+def lazy_load(
+    base_module: _ModuleType, name_obj_dict: t.Dict[str, t.Any], obj_module_dict: t.Dict[str, str]
+) -> t.Callable[[str], t.Any]:
+    """
+    use __getattr__ in the __init__.py file to lazy load some objects
+
+    Args:
+        base_module (ModuleType): base package module
+        name_obj_dict (dict[str, any]): name, real object dict, used to store real objects,
+            no need to add lazy-load objects
+        obj_module_dict (dict[str, str]): dict of object name and module name
+
+    Returns:
+        __getattr__ function
+
+    Example:
+
+        ::
+
+            __getattr__ = lazy_load(
+                importlib.import_module(__name__),
+                {
+                    'IdfApp': IdfApp,
+                    'LinuxDut': LinuxDut,
+                    'LinuxSerial': LinuxSerial,
+                    'CaseTester': CaseTester,
+                },
+                {
+                    'IdfSerial': '.serial',
+                    'IdfDut': '.dut',
+                },
+            )
+    """
+
+    def __getattr__(object_name):
+        if object_name in name_obj_dict:
+            return name_obj_dict[object_name]
+        elif object_name in obj_module_dict:
+            module = importlib.import_module(obj_module_dict[object_name], base_module.__name__)
+            imported = getattr(module, object_name)
+            name_obj_dict[object_name] = imported
+            return imported
+        else:
+            raise AttributeError('Attribute %s not found in module %s', object_name, base_module.__name__)
+
+    return __getattr__
+
 
 class InvalidInput(SystemExit):
     """Invalid input from user"""

tests/test_bool_parser.py

@@ -3,7 +3,9 @@
 
 import pytest
 
+import esp_bool_parser.constants
 from esp_bool_parser.bool_parser import parse_bool_expr, register_addition_attribute
+from esp_bool_parser.utils import to_version
 
 
 @pytest.mark.parametrize(
@@ -87,3 +89,15 @@ def test_chain_rule(s, res):
     stmt = parse_bool_expr(s)
     result = stmt.get_value('', '')
     assert result == res
+
+
+def test_idf_version(monkeypatch):
+    monkeypatch.setattr(esp_bool_parser.constants, 'IDF_VERSION', to_version('5.9.0'))
+    statement = 'IDF_VERSION > "5.10.0"'
+    assert parse_bool_expr(statement).get_value('esp32', 'foo') is False
+
+    statement = 'IDF_VERSION < "5.10.0"'
+    assert parse_bool_expr(statement).get_value('esp32', 'foo') is True
+
+    statement = 'IDF_VERSION in  ["5.9.0"]'
+    assert parse_bool_expr(statement).get_value('esp32', 'foo') is True

tests/test_load.py

@@ -0,0 +1,9 @@
+# SPDX-FileCopyrightText: 2025 Espressif Systems (Shanghai) CO LTD
+# SPDX-License-Identifier: Apache-2.0
+
+
+def test_load():
+    import esp_bool_parser
+
+    for el in esp_bool_parser.__all__:
+        esp_bool_parser.__getattr__(el)

tests/test_utils.py

@@ -0,0 +1,19 @@
+# SPDX-FileCopyrightText: 2025 Espressif Systems (Shanghai) CO LTD
+# SPDX-License-Identifier: Apache-2.0
+
+import shutil
+
+import pytest
+
+
+@pytest.mark.skipif(not shutil.which('idf.py'), reason='idf.py not found')
+def test_idf_version_keywords_type():
+    from esp_bool_parser.constants import (
+        IDF_VERSION_MAJOR,
+        IDF_VERSION_MINOR,
+        IDF_VERSION_PATCH,
+    )
+
+    assert isinstance(IDF_VERSION_MAJOR, int)
+    assert isinstance(IDF_VERSION_MINOR, int)
+    assert isinstance(IDF_VERSION_PATCH, int)