From f0f96f83c2630e890a2cbcab53f77fd4c37e1684 Mon Sep 17 00:00:00 2001 From: Max Nanis Date: Thu, 19 Feb 2026 02:43:23 -0500 Subject: Models, Project files, some pytests, requirements.. etc --- jb/models/api_response.py | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) create mode 100644 jb/models/api_response.py (limited to 'jb/models/api_response.py') 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 -- cgit v1.2.3