Add types and converters for Sick and Fracas problems

Author: XanderVertegaal

backend/problem/services.py

@@ -0,0 +1,70 @@
+import json
+from .models import Problem
+from .types import FracasProblem, SickProblem
+
+
+def get_sick_problems() -> list[SickProblem]:
+    """
+    Retrieves all Problem objects of type 'SICK' from the database
+    and converts them into SickProblem instances.
+    """
+    sick_problems: list[SickProblem] = []
+    problem_objects = Problem.objects.filter(type=Problem.ProblemType.SICK)
+
+    for problem_obj in problem_objects:
+        try:
+            problem_data = json.loads(problem_obj.content)
+            problem = SickProblem(
+                    pair_id=problem_data["pair_ID"],
+                    sentence_one=problem_data["sentence_A"],
+                    sentence_two=problem_data["sentence_B"],
+                    entailment_label=problem_data["entailment_label"],
+                    relatedness_score=float(problem_data["relatedness_score"]),
+                )
+            sick_problems.append(problem)
+        except json.JSONDecodeError:
+            print(
+                f"Warning: Could not parse JSON content for Problem ID {problem_obj.id}"
+            )
+        except TypeError as e:
+            print(
+                f"Warning: Could not create SickProblem for Problem ID {problem_obj.id}: {e}"
+            )
+
+    return sick_problems
+
+
+def get_fracas_problems() -> list[FracasProblem]:
+    """
+    Retrieves all Problem objects of type 'Fracas' from the database
+    and converts them into FracasProblem instances.
+    """
+    fracas_problems: list[FracasProblem] = []
+    problem_objects = Problem.objects.filter(type=Problem.ProblemType.FRACAS)
+
+    for problem_obj in problem_objects:
+        try:
+            problem_data = json.loads(problem_obj.content)
+            problem = FracasProblem(
+                fracas_id=problem_data["fracas_id"],
+                question=problem_data["question"],
+                hypothesis=problem_data["hypothesis"],
+                answer=problem_data["answer"],
+                fracas_answer=problem_data["fracas_answer"],
+                fracas_non_standard=problem_data["fracas_non_standard"],
+                note=problem_data["note"],
+                section_name=problem_data["section_name"],
+                subsection_name=problem_data["subsection_name"],
+                premises=problem_data.get("premises", []),
+            )
+            fracas_problems.append(problem)
+        except json.JSONDecodeError:
+            print(
+                f"Warning: Could not parse JSON content for Problem ID {problem_obj.id}"
+            )
+        except TypeError as e:
+            print(
+                f"Warning: Could not create FracasProblem for Problem ID {problem_obj.id}: {e}"
+            )
+
+    return fracas_problems

backend/problem/types.py

@@ -0,0 +1,29 @@
+from typing import Literal
+from dataclasses import dataclass, field
+
+
+@dataclass(frozen=True)
+class SickProblem:
+    pair_id: int
+    sentence_one: str
+    sentence_two: str
+    entailment_label: Literal["neutral", "contradiction", "entailment"]
+    relatedness_score: float
+
+
+@dataclass(frozen=True)
+class FracasProblem:
+    fracas_id: int
+    question: str
+    hypothesis: str
+    answer: str
+    fracas_answer: Literal["yes", "no", "unknown", "undefined"]
+    fracas_non_standard: bool
+    note: str
+    section_name: str
+    subsection_name: str
+    premises: list[str] = field(default_factory=list)
+
+
+
+