Serialize NaNs as None

Author: s-boomi

src/psup_stac_converter/utils/models.py

@@ -1,8 +1,9 @@
 import datetime as dt
 from typing import Literal, Optional
 
+import pandas as pd
 from geojson_pydantic import Feature
-from pydantic import BaseModel, field_validator
+from pydantic import BaseModel, field_serializer, field_validator
 
 
 class CubedataDimensionBase(BaseModel):
@@ -22,6 +23,12 @@ class CubedataDimensionBase(BaseModel):
     reference_system: Optional[str | int]
     unit: Optional[str] = None
 
+    @field_serializer("extent", "values", mode="plain")
+    def ser_number(
+        self, value: list[str | int | float | None]
+    ) -> list[str | int | float | None]:
+        return [element if pd.notnull(element) else None for element in value]
+
 
 class CubedataVariable(BaseModel):
     """A *Variable Object* defines a variable (or a multi-dimensional array).
@@ -54,6 +61,12 @@ class CubedataVariable(BaseModel):
     values: Optional[list[str | int | float]] = None
     unit: Optional[str] = None
 
+    @field_serializer("extent", mode="plain")
+    def ser_number(
+        self, value: list[str | int | float | None]
+    ) -> list[str | int | float | None]:
+        return [element if pd.notnull(element) else None for element in value]
+
 
 class HorizontalSpatialRasterDimension(CubedataDimensionBase):
     """A spatial raster dimension in one of the horizontal (x or y) directions.