Models, Project files, some pytests, requirements.. etcHEADmaster

1 files changed, 17 insertions, 0 deletions

diff --git a/jb/models/api_response.py b/jb/models/api_response.py
new file mode 100644
index 0000000..6b29e51
--- /dev/null
+++ b/jb/models/api_response.py
@@ -0,0 +1,17 @@
+from pydantic import BaseModel, ConfigDict, Field, model_validator
+
+from jb.models.assignment import Assignment
+from jb.models.hit import Hit
+
+
+class AssignmentResponse(BaseModel):
+    model_config = ConfigDict(extra="forbid", validate_assignment=True)
+
+    assignment: Assignment = Field(alias="Assignment")
+    hit: Hit = Field(alias="HIT")
+
+    @model_validator(mode="after")
+    def check_consistent_hit_id(self) -> "AssignmentResponse":
+        if self.hit.id != self.assignment.hit_id:
+            raise ValueError("Inconsistent Hit IDs")
+        return self