support extraction from sh:or

Author: gtfierro

buildingmotif/utils.py

@@ -9,6 +9,7 @@
 
 import pyshacl  # type: ignore
 from rdflib import BNode, Graph, Literal, URIRef
+from rdflib.collection import Collection
 from rdflib.compare import _TripleCanonicalizer
 from rdflib.paths import ZeroOrOne
 from rdflib.term import Node
@@ -380,6 +381,22 @@ def _gather(predicate):
             if pvalue:
                 body.add((focus_param, property_path, pvalue))
 
+        for or_list in shape_graph.objects(shape_node, SH["or"]):
+            try:
+                options = list(Collection(shape_graph, or_list))
+            except Exception:
+                options = []
+            if not options:
+                continue
+            chosen = options[0]
+            if isinstance(chosen, URIRef):
+                if is_nodeshape(chosen):
+                    process_shape(chosen, focus_param)
+                else:
+                    body.add((focus_param, RDF.type, chosen))
+            elif isinstance(chosen, BNode):
+                process_shape(chosen, focus_param)
+
     process_shape(shape_name, root_param)
 
     return body, deps

tests/unit/test_utils.py

@@ -200,6 +200,29 @@ def test_inline_sh_nodes(shacl_engine):
     assert len(shape1_cbd) == 8
 
 
+def test_get_template_parts_from_shape_with_or():
+    shape_graph = Graph()
+    shape_graph.parse(
+        data=PREAMBLE
+        + """
+    :shape1 a owl:Class, sh:NodeShape ;
+        sh:or ( :option1 :option2 ) .
+    :option1 a owl:Class, sh:NodeShape ;
+        sh:property [
+            sh:path brick:hasPoint ;
+            sh:qualifiedValueShape [ sh:class brick:Zone_Air_Temperature_Sensor ] ;
+            sh:qualifiedMinCount 1 ;
+        ] .
+    :option2 a owl:Class, sh:NodeShape .
+    """
+    )
+    body, deps = get_template_parts_from_shape(MODEL["shape1"], shape_graph)
+    point_targets = list(body.objects(PARAM["name"], BRICK.hasPoint))
+    assert point_targets
+    for target in point_targets:
+        assert (target, A, BRICK.Zone_Air_Temperature_Sensor) in body
+
+
 def test_inline_sh_and(bm: BuildingMOTIF, shacl_engine):
     bm.shacl_engine = shacl_engine
     sg = Graph()