Added TIME / TIMESTAMP support

gizmodata · Jul 9, 2024 · 9003f1f · 9003f1f
1 parent 22f1e84
commit 9003f1f
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/sqlalchemy_flight_sql_adbc_dialect/__init__.py b/src/sqlalchemy_flight_sql_adbc_dialect/__init__.py
@@ -288,6 +288,10 @@ def _get_column_type(self, data_type: str):
             return sqltypes.Integer
         elif data_type == 'DATE':
             return sqltypes.DateTime
+        elif data_type == "TIMESTAMP":
+            return sqltypes.TIMESTAMP
+        elif data_type == "TIME":
+            return sqltypes.TIME
         elif data_type == "BIGINT":
             return sqltypes.BigInteger
         elif re.match(pattern="^DECIMAL", string=data_type):