feat: add partner lab models and update lab order command to validate results

canvas_sdk/commands/commands/lab_order.py

@@ -1,4 +1,10 @@
+from typing import Literal
+
+from django.db.models.query_utils import Q
+from pydantic_core import InitErrorDetails
+
 from canvas_sdk.commands.base import _BaseCommand as BaseCommand
+from canvas_sdk.v1.data.lab import LabPartner, LabPartnerTest
 
 
 class LabOrderCommand(BaseCommand):
@@ -20,6 +26,60 @@ class Meta:
     fasting_required: bool = False
     comment: str | None = None
 
+    def _get_error_details(
+        self, method: Literal["originate", "edit", "delete", "commit", "enter_in_error"]
+    ) -> list[InitErrorDetails]:
+        errors = super()._get_error_details(method)
+
+        lab_partner_obj = None
+        if self.lab_partner:
+            lab_partner_obj = (
+                LabPartner.objects.filter(Q(name=self.lab_partner) | Q(id=self.lab_partner))
+                .values("id", "dbid")
+                .first()
+            )
+            if not lab_partner_obj:
+                errors.append(
+                    self._create_error_detail(
+                        "value",
+                        f"lab partner with Id or Name {self.lab_partner} not found",
+                        self.lab_partner,
+                    )
+                )
+
+        if self.tests_order_codes:
+            if not lab_partner_obj:
+                errors.append(
+                    self._create_error_detail(
+                        "value", "lab partner is required to find tests", self.tests_order_codes
+                    )
+                )
+            else:
+                tests = LabPartnerTest.objects.filter(
+                    Q(order_code__in=self.tests_order_codes) | Q(id__in=self.tests_order_codes),
+                    lab_partner_id=lab_partner_obj["dbid"],
+                ).values_list("id", "order_code")
+
+                if tests.count() != len(self.tests_order_codes):
+                    missing_tests = [
+                        test
+                        for test in self.tests_order_codes
+                        if not any(
+                            test == str(test_id) or test == order_code
+                            for test_id, order_code in tests
+                        )
+                    ]
+
+                    errors.append(
+                        self._create_error_detail(
+                            "value",
+                            f"tests with order codes {missing_tests} not found",
+                            self.tests_order_codes,
+                        )
+                    )
+
+        return errors
+
     @property
     def values(self) -> dict:
         """The Lab Order command's field values."""

canvas_sdk/v1/data/lab.py

@@ -311,3 +311,41 @@ class Meta:
 
     def __str__(self) -> str:
         return f"{self.ontology_test_name}: f{self.ontology_test_code}"
+
+
+class LabPartner(models.Model):
+    """A class representing a lab partner."""
+
+    class Meta:
+        managed = False
+        db_table = "canvas_sdk_data_lab_partner_001"
+
+    objects: models.Manager["LabPartner"]
+
+    id = models.UUIDField()
+    dbid = models.BigIntegerField(primary_key=True)
+    name = models.CharField(max_length=256)
+    active = models.BooleanField()
+    electronic_ordering_enabled = models.BooleanField()
+    keywords = models.TextField()
+    default_lab_account_number = models.CharField(max_length=256)
+
+
+class LabPartnerTest(models.Model):
+    """A class representing a lab partner's test."""
+
+    class Meta:
+        managed = False
+        db_table = "canvas_sdk_data_lab_partner_test_001"
+
+    objects: models.Manager["LabPartnerTest"]
+
+    id = models.UUIDField()
+    dbid = models.BigIntegerField(primary_key=True)
+    lab_partner = models.ForeignKey(
+        "LabPartner", on_delete=models.DO_NOTHING, related_name="available_tests"
+    )
+    order_code = models.CharField(max_length=256, blank=True)
+    order_name = models.TextField()
+    keywords = models.TextField(blank=True)
+    cpt_code = models.CharField(max_length=256, blank=True, null=True)