Test rule parsing output

src/pyrdfrules/common/result/result.py

@@ -0,0 +1,40 @@
+
+from pyrdfrules.common.rule.resultrule import ResultRule
+from pyrdfrules.common.rule.rule.body import RuleBody
+from pyrdfrules.common.rule.rule.head import RuleHead
+from pyrdfrules.common.rule.ruleset import Ruleset
+
+
+class Result():
+
+    ruleset: Ruleset = None
+    """Ruleset generated by RDFRules.
+    """
+
+    def __init__(self, data: dict) -> None:
+        self.id = id
+        self.data = data
+
+        self._parse_data()
+
+        pass
+
+    def _parse_data(self):
+
+        rules = []
+
+        for item in self.data:                
+            match item:
+                case {'body': _, 'head': __, 'measures': ___}:
+                    # Item is a rule
+
+                    rule = ResultRule.model_construct(item)
+
+                    print(rule)
+
+                    pass
+                case _: 
+                    pass
+
+    def get_ruleset(self) -> Ruleset:
+        pass

src/pyrdfrules/common/rule/rule.py → src/pyrdfrules/common/rule/resultrule.py

@@ -6,15 +6,15 @@
 from pyrdfrules.common.rule.rule.body import RuleBody
 from pyrdfrules.common.rule.rule.head import RuleHead
 
-class Rule(BaseModel):
+class ResultRule(BaseModel):
     """
     Rule mined by RDFRules.
     """
 
     """
     Represents the list of the parts of the rule in its body.
     """
-    body: RuleBody
+    body: List[RuleBody]
 
     """
     Head of the rule.

src/pyrdfrules/common/rule/rule/body.py

@@ -2,6 +2,8 @@
 from pydantic import BaseModel
 
 class RuleBody(BaseModel):
+    graphs: str
 
+    object: str
 
-    items: List[any]
+    items: List[dict]

src/pyrdfrules/common/rule/rule/head.py

@@ -1,6 +1,16 @@
 from typing import List
 from pydantic import BaseModel
 
+from pyrdfrules.common.rule.rule.object import Object
+from pyrdfrules.common.rule.rule.predicate import Predicate
+from pyrdfrules.common.rule.rule.subject import Subject
+
 class RuleHead(BaseModel):
 
-    item: any
+    graphs: List[str]
+
+    object: Object
+
+    predicate: Predicate
+
+    subject: Subject

src/pyrdfrules/common/rule/rule/object.py

@@ -0,0 +1,8 @@
+from pydantic import BaseModel
+
+
+class Object(BaseModel):
+
+    type: str
+
+    value: str

src/pyrdfrules/common/rule/rule/predicate.py

@@ -0,0 +1,11 @@
+from typing import Optional
+from pydantic import BaseModel
+
+
+class Predicate(BaseModel):
+
+    localName: str
+
+    nameSpace: str
+
+    prefix: Optional[str]

src/pyrdfrules/common/rule/rule/subject.py

@@ -0,0 +1,8 @@
+from pydantic import BaseModel
+
+
+class Subject(BaseModel):
+
+    type: str
+
+    value: str

src/pyrdfrules/common/rule/ruleset.py

@@ -1,7 +1,8 @@
 from typing import List
 from pydantic import BaseModel
 
-from pyrdfrules.common.rule.rule import Rule
+from pyrdfrules.common.rule.resultrule import ResultRule
+
 
 class Ruleset(BaseModel):
     """
@@ -14,7 +15,7 @@ class Ruleset(BaseModel):
     """
     List of rules generated by RDFRules.
     """
-    rules: List[Rule]
+    rules: List[ResultRule]
 
     """
     True if the ruleset is complete.
@@ -32,7 +33,7 @@ class Ruleset(BaseModel):
     """
     __iterator_index: int = 0
 
-    def get_rules(self) -> List[Rule]:
+    def get_rules(self) -> List[ResultRule]:
         # return all rules
         pass
 
@@ -47,7 +48,7 @@ def __iter__(self):
         self.__iterator_index = 0
         return self
 
-    def __next__(self) -> Rule | None:
+    def __next__(self) -> ResultRule | None:
         idx = self.__iterator_index + 1
 
         if len(self.rules) < idx:

src/pyrdfrules/common/task/task_runner.py

@@ -51,7 +51,7 @@ def create_task_from_string(self, task: str) -> Task:
         return task
 
     def run_task(self, task: Task) -> Generator[Task, None, None]:
-        """Runs the task to completion.
+        """Runs a task step.
         """
 
         yield from self.task_updater.run(task)

src/tests/test_parsing.py

@@ -0,0 +1,28 @@
+import json
+import logging
+import unittest
+
+from pyrdfrules.common.result.result import Result
+
+
+
+class TestRuleParsing(unittest.TestCase):
+
+    def test_rule_parsing(self):
+        """
+        Tests parsing of a rule.
+        """
+
+        data = '[{"body": [{"graphs": ["<dbpedia>"], "object": {"type": "variable", "value": "?b"}, "predicate": {"localName": "subsequentWork", "nameSpace": "http://dbpedia.org/ontology/", "prefix": "dbo"}, "subject": {"type": "variable", "value": "?c"}}, {"graphs": ["<dbpedia>"], "object": {"type": "variable", "value": "?a"}, "predicate": {"localName": "artist", "nameSpace": "http://dbpedia.org/ontology/", "prefix": "dbo"}, "subject": {"type": "variable", "value": "?c"}}], "head": {"graphs": ["<yago>"], "object": {"type": "variable", "value": "?b"}, "predicate": "<created>", "subject": {"type": "variable", "value": "?a"}}, "measures": [{"name": "Support", "value": 845}, {"name": "HeadSupport", "value": 5803}, {"name": "PcaBodySize", "value": 871}, {"name": "PcaConfidence", "value": 0.9701492537313433}, {"name": "HeadSize", "value": 5803}, {"name": "HeadCoverage", "value": 0.14561433741168361}]}]'
+
+        rules = json.loads(data)
+
+        print(rules)
+
+        result = Result(rules)
+
+        print(rules)
+
+
+if __name__ == '__main__':
+    unittest.main()

src/tests/test_pipeline.py

@@ -158,6 +158,8 @@ def test_pipeline_execution(self):
             print(step)
             self.assertIsNotNone(step, "Should not be None")
 
+        print(task.result)
+
         app.stop()
 
 if __name__ == '__main__':