Merge pull request #5 from kszucs/nullable

noissue nullable columns handled at convert.py
kszucs · Sep 29, 2017 · 9868a35 · 9868a35
2 parents 71b5ad4 + 18c653f
commit 9868a35
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/pandahouse/convert.py b/pandahouse/convert.py
@@ -12,9 +12,9 @@
            'uint64': 'UInt64',
            'uint32': 'UInt32',
            'uint16': 'UInt16',
+           'uint8': 'UInt8',
            'float64': 'Float64',
            'float32': 'Float32',
-           'uint8': 'UInt8',
            'int64': 'Int64',
            'int32': 'Int32',
            'int16': 'Int16',
@@ -26,7 +26,11 @@
 CH2PD = itemmap(reversed, MAPPING)
 CH2PD['Null'] = 'object'
 
+NULLABLE_COLS = ['UInt64', 'UInt32', 'UInt16', 'UInt8', 'Float64', 'Float32',
+                 'Int64', 'Int32', 'Int16', 'Int8']
 
+for col in NULLABLE_COLS:
+    CH2PD['Nullable({})'.format(col)] = CH2PD[col]
 PY3 = sys.version_info[0] == 3