fix location serializing

cancervariants · Jan 2, 2024 · 5106be0 · 5106be0
1 parent 7f102a5
commit 5106be0
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/gene/database/dynamodb.py b/src/gene/database/dynamodb.py
@@ -10,6 +10,7 @@
 import click
 from boto3.dynamodb.conditions import Key
 from botocore.exceptions import ClientError
+from pydantic import BaseModel
 
 from gene.database.database import (
     AWS_ENV_VAR_NAME,
@@ -382,6 +383,12 @@ def add_record(self, record: Dict, src_name: SourceName) -> None:
         label_and_type = f"{concept_id.lower()}##identity"
         record["label_and_type"] = label_and_type
         record["item_type"] = "identity"
+        for i, location in enumerate(record.get("locations", [])):
+            if isinstance(location, BaseModel):
+                record["locations"][i] = location.model_dump(
+                    mode="json", exclude_none=True
+                )
+
         try:
             self.batch.put_item(Item=record)
         except ClientError as e: