bugfix: decoding None values

idatsy · Aug 24, 2024 · b834eae · b834eae
1 parent 285949d
commit b834eae
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/py_questdb/__init__.py b/py_questdb/__init__.py
@@ -0,0 +1,3 @@
+from py_questdb.db import QuestDB
+
+__all__ = ['QuestDB']
diff --git a/py_questdb/db.py b/py_questdb/db.py
@@ -142,7 +142,9 @@ def parse_and_yield_query_response(
 
         for row in response.dataset:
             converted_row = {
-                field.name: converter(value) for field, converter, value in zip(response.columns, type_converter, row)
+                field.name: converter(value) if value is not None else None
+                for field, converter, value
+                in zip(response.columns, type_converter, row)
             }
 
             yield into_type(**converted_row) if into_type else converted_row