add boundaries, internal and items logic for mesh_components' relations table

Author: MaxNumerique

src/opengeodeweb_back/geode_objects/geode_brep.py

@@ -79,6 +79,15 @@ def mesh_components(self) -> ComponentRegistry:
     def collection_components(self) -> ComponentRegistry:
         return self.brep.collection_components()
 
+    def boundaries(self, id: og.uuid) -> list[og.ComponentID]:
+        return self.brep.boundaries(id)
+
+    def internals(self, id: og.uuid) -> list[og.ComponentID]:
+        return self.brep.internals(id)
+
+    def items(self, id: og.uuid) -> list[og.ComponentID]:
+        return self.brep.items(id)
+
     def inspect(self) -> og_inspector.BRepInspectionResult:
         return og_inspector.inspect_brep(self.brep)
 

src/opengeodeweb_back/geode_objects/geode_model.py

@@ -26,3 +26,12 @@ def mesh_components(self) -> ComponentRegistry: ...
 
     @abstractmethod
     def collection_components(self) -> ComponentRegistry: ...
+
+    @abstractmethod
+    def boundaries(self, id: og.uuid) -> list[og.ComponentID]: ...
+
+    @abstractmethod
+    def internals(self, id: og.uuid) -> list[og.ComponentID]: ...
+
+    @abstractmethod
+    def items(self, id: og.uuid) -> list[og.ComponentID]: ...

src/opengeodeweb_back/geode_objects/geode_section.py

@@ -81,6 +81,15 @@ def mesh_components(self) -> ComponentRegistry:
     def collection_components(self) -> ComponentRegistry:
         return self.section.collection_components()
 
+    def boundaries(self, id: og.uuid) -> list[og.ComponentID]:
+        return self.section.boundaries(id)
+
+    def internals(self, id: og.uuid) -> list[og.ComponentID]:
+        return self.section.internals(id)
+
+    def items(self, id: og.uuid) -> list[og.ComponentID]:
+        return self.section.items(id)
+
     def inspect(self) -> og_inspector.SectionInspectionResult:
         return og_inspector.inspect_section(self.section)
 

src/opengeodeweb_back/routes/models/blueprint_models.py

@@ -44,12 +44,17 @@ def model_components() -> flask.Response:
             geode_id = id.string()
             component_name = geode_id
             viewer_id = uuid_to_flat_index[geode_id]
-
+            boundaries = model.boundaries(id)
+            boundaries_uuid = [boundary.id().string() for boundary in boundaries]
+            internals = model.internals(id)
+            internals_uuid = [internal.id().string() for internal in internals]
             mesh_component_object = {
                 "viewer_id": viewer_id,
                 "geode_id": geode_id,
                 "name": component_name,
                 "type": component_type,
+                "boundaries": boundaries_uuid,
+                "internals": internals_uuid,
             }
             mesh_components.append(mesh_component_object)
 
@@ -59,10 +64,13 @@ def model_components() -> flask.Response:
         component_type = collection_component.get()
         for id in ids:
             geode_id = id.string()
+            items = model.items(id)
+            items_uuid = [item.id().string() for item in items]
             collection_component_object = {
                 "geode_id": geode_id,
                 "name": geode_id,
                 "type": component_type,
+                "items": items_uuid,
             }
             collection_components.append(collection_component_object)
 

tests/test_models_routes.py

@@ -38,13 +38,22 @@ def test_model_components(client: FlaskClient) -> None:
         assert isinstance(mesh_component["viewer_id"], int)
         assert isinstance(mesh_component["name"], str)
         assert isinstance(mesh_component["type"], str)
+        assert isinstance(mesh_component["boundaries"], list)
+        for boundary_uuid in mesh_component["boundaries"]:
+            assert isinstance(boundary_uuid, str)
+        assert isinstance(mesh_component["internals"], list)
+        for internal_uuid in mesh_component["internals"]:
+            assert isinstance(internal_uuid, str)
     assert "collection_components" in response.get_json()
     collection_components = response.get_json()["collection_components"]
     assert isinstance(collection_components, list)
     for collection_component in collection_components:
         assert isinstance(collection_component, object)
         assert isinstance(collection_component["geode_id"], str)
         assert isinstance(collection_component["name"], str)
+        assert isinstance(collection_component["items"], list)
+        for item_uuid in collection_component["items"]:
+            assert isinstance(item_uuid, str)
 
 
 def test_export_project_route(client: FlaskClient, tmp_path: Path) -> None: