Support BigInt columns (#54)

Co-authored-by: Sergio Durban Belmonte <[email protected]>

Author: sdurban

local_data_api/models.py

@@ -42,6 +42,8 @@ def from_value(cls, value: Any) -> Field:
             return cls(stringValue=str(value))
         elif type(value).__name__.endswith('PGobject'):
             return cls(stringValue=str(value))
+        elif type(value).__name__.endswith('BigInteger'):
+            return cls(longValue=int(str(value)))
         else:
             raise Exception(f'unsupported type {type(value)}: {value} ')
 

tests/test_models.py

@@ -70,6 +70,15 @@ def __str__(self) -> str:
 
     assert Field.from_value(PGobject("{}")) == Field(stringValue="{}")
 
+    class BigInteger:
+        def __init__(self, val: int):
+            self._val: int = val
+
+        def __str__(self) -> int:
+            return self._val
+
+    assert Field.from_value(BigInteger("55")) == Field(longValue=55)
+
     class Dummy:
         pass