Add reading material_id to get_pipes

nens · Nov 12, 2024 · b9ce3c9 · b9ce3c9
1 parent 14b1e9f
commit b9ce3c9
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/threedigrid_builder/interface/db.py b/threedigrid_builder/interface/db.py
@@ -373,7 +373,10 @@ def get_surfaces(self) -> Surfaces:
                 .join(
                     models.SurfaceMap, models.SurfaceMap.surface_id == models.Surface.id
                 )
-                .join(models.ConnectionNode, models.SurfaceMap.connection_node_id == models.ConnectionNode.id)
+                .join(
+                    models.ConnectionNode,
+                    models.SurfaceMap.connection_node_id == models.ConnectionNode.id,
+                )
                 .order_by(models.Surface.id)
                 .as_structarray()
             )
@@ -833,6 +836,7 @@ def get_pipes(self) -> Pipes:
             models.Pipe.exchange_thickness,
             models.Pipe.hydraulic_conductivity_out,
             models.Pipe.hydraulic_conductivity_in,
+            models.Pipe.material_id,
             case(
                 {
                     models.Pipe.friction_value.isnot(None)