support null values for optional fields

Signed-off-by: wiseaidev <[email protected]>

Author: wiseaidev

aredis_om/model/model.py

@@ -1316,6 +1316,8 @@ async def get(cls, pk: Any) -> "HashModel":
         if not document:
             raise NotFoundError
         try:
+            # restore none values
+            document = {key: val if val != "0" else None for key, val in document.items()}
             result = cls.parse_obj(document)
         except TypeError as e:
             log.warning(
@@ -1332,14 +1334,14 @@ async def get(cls, pk: Any) -> "HashModel":
     @no_type_check
     def _get_value(cls, *args, **kwargs) -> Any:
         """
-        Always send None as an empty string.
+        Always send None as a zero string: "0" to handle Optional int and float fields.
 
         TODO: We do this because redis-py's hset() method requires non-null
         values. Is there a better way?
         """
         val = super()._get_value(*args, **kwargs)
         if val is None:
-            return ""
+            return "0"
         return val
 
     @classmethod

tests/test_hash_model.py

@@ -48,6 +48,8 @@ class Member(BaseHashModel):
         last_name: str = Field(index=True)
         email: str = Field(index=True)
         join_date: datetime.date
+        height: Optional[int] = None
+        weight: Optional[float] = None
         age: int = Field(index=True, sortable=True)
         bio: str = Field(index=True, full_text_search=True)