Reduce the size of score set control variant responses.

Author: jstone-dev

src/mavedb/view_models/clinical_control.py

@@ -39,7 +39,7 @@ class Config:
 
 
 class SavedClinicalControlWithMappedVariants(SavedClinicalControl):
-    mapped_variants: Sequence["SavedMappedVariant"]
+    mapped_variants: Sequence["MappedVariantForClinicalControl"]
 
 
 # Properties to return to non-admin clients
@@ -48,7 +48,7 @@ class ClinicalControl(SavedClinicalControl):
 
 
 class ClinicalControlWithMappedVariants(SavedClinicalControlWithMappedVariants):
-    mapped_variants: Sequence["MappedVariant"]
+    pass
 
 
 class ClinicalControlOptions(BaseModel):
@@ -57,7 +57,7 @@ class ClinicalControlOptions(BaseModel):
 
 
 # ruff: noqa: E402
-from mavedb.view_models.mapped_variant import MappedVariant, SavedMappedVariant, MappedVariantCreate
+from mavedb.view_models.mapped_variant import MappedVariantCreate, MappedVariantForClinicalControl
 
 # ClinicalControlUpdate.model_rebuild()
 SavedClinicalControlWithMappedVariants.model_rebuild()

src/mavedb/view_models/mapped_variant.py

@@ -80,6 +80,22 @@ class MappedVariantWithControls(SavedMappedVariantWithControls):
     gnomad_variants: Sequence["GnomADVariant"]
 
 
+class MappedVariantForClinicalControl(BaseModel):
+    variant_urn: str
+
+    class Config:
+        from_attributes = True
+
+    @model_validator(mode="before")
+    def generate_score_set_urn_list(cls, data: Any):
+        if not hasattr(data, "variant_urn") and hasattr(data, "variant"):
+            try:
+                data.__setattr__("variant_urn", None if not data.variant else data.variant.urn)
+            except AttributeError as exc:
+                raise ValidationError(f"Unable to create {cls.__name__} without attribute: {exc}.")  # type: ignore
+        return data
+
+
 # ruff: noqa: E402
 from mavedb.view_models.clinical_control import ClinicalControlBase, ClinicalControl, SavedClinicalControl
 from mavedb.view_models.gnomad_variant import GnomADVariantBase, GnomADVariant, SavedGnomADVariant