feat: generate entities.py with RDFadapter

owl2yams/__init__.py

@@ -19,6 +19,7 @@
 )
 
 from owl2yams.generate_postcreate import generate_postcreate
+from owl2yams.generate_entities import generate_entities
 
 here = os.path.dirname(__file__)
 
@@ -274,6 +275,12 @@
             )
         )
 
+    with open(f"{cube_master_folder}/{cube_subfolder}/entities.py", "a") as f:
+        f.write(
+            generate_entities(
+                schema, entitytype_fragment_to_uri, relationtype_fragment_to_uri
+            )
+        )
     if args.only_cube:
         print("Cube correctly initialized")
         exit(0)

owl2yams/generate_entities.py

+from yams.schema import Schema
+from typing import Dict
+
+
+def generate_entities(
+    schema: Schema,
+    entitytype_fragment_to_uri: Dict[str, str],
+    relationtype_fragment_to_uri: Dict[str, str],
+) -> str:
+    entities_py_content = """
+from cubicweb.entities.adapters import EntityRDFAdapter
+from rdflib import URIRef, OWL
+"""
+
+    for fragment, uri in entitytype_fragment_to_uri.items():
+        entities_py_content += f"""
+
+class {fragment}RDFAdapter(EntityRDFAdapter):
+    def triples(self):
+        yield from super().triples()
+        yield(URIRef("{uri}"), OWL.sameAs, URIRef(self.uri))
+"""
+
+    return entities_py_content