x

Author: soburi

alt_core_api/tools/protoc_gen_arduinoif/core.py

@@ -28,10 +28,7 @@ def _iter_generated_files(
             continue
 
         if proto_file.enum_type and not proto_file.service:
-            yield from EnumRenderer(
-                proto_file.name,
-                list(proto_file.enum_type),
-            ).iter_headers()
+            yield from EnumRenderer(proto_file.name, list(proto_file.enum_type))
 
         for service in proto_file.service:
             service_model = ServiceModelBuilder.build(
@@ -40,7 +37,7 @@ def _iter_generated_files(
                 list(proto_file.enum_type),
                 context,
             )
-            yield from ServiceRenderer(service_model).iter_headers()
+            yield from ServiceRenderer(service_model)
 
 
 def main() -> int:

alt_core_api/tools/protoc_gen_arduinoif/enum_renderer.py

@@ -17,7 +17,7 @@ def __init__(self, proto_name: str, enums: List[EnumDescriptorProto]) -> None:
         self._proto_name = proto_name
         self._enums = enums
 
-    def iter_headers(self) -> Iterator[Tuple[str, str]]:
+    def __iter__(self) -> Iterator[Tuple[str, str]]:
         yield self._header_name(), self.render_header_content()
 
     def _header_name(self) -> str:

alt_core_api/tools/protoc_gen_arduinoif/service_renderer.py

@@ -45,7 +45,7 @@ class ServiceRenderer:
     def __init__(self, model: ServiceModel) -> None:
         self._model = model
 
-    def iter_headers(self) -> Iterator[Tuple[str, str]]:
+    def __iter__(self) -> Iterator[Tuple[str, str]]:
         for header_attr, surface, gate_attr in _HEADER_ORDER:
             if gate_attr and not getattr(self._model, gate_attr):
                 continue

alt_core_api/tools/protoc_gen_arduinoif/tests/test_renderer_templates.py

@@ -89,7 +89,7 @@ def _sample_model() -> ServiceModel:
 def test_renderer_templates_emit_expected_fragments() -> None:
     model = _sample_model()
     renderer = ServiceRenderer(model)
-    rendered = dict(renderer.iter_headers())
+    rendered = dict(renderer)
 
     ifc_content = rendered["example_interface.hpp"]
     assert "class ExampleInterface" in ifc_content
@@ -117,7 +117,7 @@ def test_renderer_template_iter_order_is_stable() -> None:
     model = _sample_model()
     renderer = ServiceRenderer(model)
 
-    names = [name for name, _ in renderer.iter_headers()]
+    names = [name for name, _ in renderer]
     assert names == [
         "example_interface.hpp",
         "example_api.hpp",

alt_core_api/tools/protoc_gen_arduinoif/tests/test_service_model.py

@@ -94,7 +94,7 @@ def test_build_service_model_has_expected_headers_and_groups(tmp_path: Path) ->
 
 def test_render_service_headers_uses_stable_order(tmp_path: Path) -> None:
     model = _hardware_serial_model(tmp_path)
-    rendered_names = [name for name, _ in ServiceRenderer(model).iter_headers()]
+    rendered_names = [name for name, _ in ServiceRenderer(model)]
     assert rendered_names == [
         "hardware_serial_interface.hpp",
         "hardware_serial_api.hpp",