Support BigInt columns (#54)

Co-authored-by: Sergio Durban Belmonte <[email protected]>
koxudaxi · Apr 3, 2020 · aa0a72f · aa0a72f
1 parent 2ed6146
commit aa0a72f
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/local_data_api/models.py b/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} ')
 

diff --git a/tests/test_models.py b/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