Merge pull request #999 from Mause/bleeding

fix: add new type names for 0.10.3
Mause · May 24, 2024 · 5b15b2b · 5b15b2b
2 parents 48f91e9 + 8e0ca8e
commit 5b15b2b
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/duckdb_engine/datatypes.py b/duckdb_engine/datatypes.py
@@ -16,7 +16,7 @@
 from sqlalchemy.ext.compiler import compiles
 from sqlalchemy.sql import sqltypes, type_api
 from sqlalchemy.sql.type_api import TypeEngine
-from sqlalchemy.types import BigInteger, Integer, SmallInteger
+from sqlalchemy.types import BigInteger, Integer, SmallInteger, String
 
 # INTEGER	INT4, INT, SIGNED	-2147483648	2147483647
 # SMALLINT	INT2, SHORT	-32768	32767
@@ -181,14 +181,22 @@ def __init__(self, fields: Dict[str, TV]):
     "uhugeint": UHugeInteger,
     "tinyint": TinyInteger,
     "utinyint": UTinyInteger,
+    "int8": BigInteger,
+    "int4": Integer,
+    "int2": SmallInteger,
+    "timetz": sqltypes.TIME,
+    "timestamptz": sqltypes.TIMESTAMP,
+    "float4": sqltypes.FLOAT,
+    "float8": sqltypes.FLOAT,
     "usmallint": USmallInteger,
     "uinteger": UInteger,
     "ubigint": UBigInteger,
     "timestamp_s": sqltypes.TIMESTAMP,
     "timestamp_ms": sqltypes.TIMESTAMP,
     "timestamp_ns": sqltypes.TIMESTAMP,
     "enum": sqltypes.Enum,
-    "json": sqltypes.JSON,
+    "bool": sqltypes.BOOLEAN,
+    "varchar": String,
 }