fix: support binary type (#17)

Author: hantmac

databend_sqlalchemy/databend_dialect.py

@@ -52,20 +52,6 @@ def __init__(self, key_type, value_type):
         super(MAP, self).__init__()
 
 
-_type_map = {
-    "ARRAY": sqlalchemy.types.ARRAY,
-    "BOOLEAN": sqlalchemy.types.Boolean,
-    "DATETIME": sqlalchemy.types.DATETIME,
-    "DATE": sqlalchemy.types.DATE,
-    "DOUBLE": sqlalchemy.types.Float,
-    "FLOAT": sqlalchemy.types.Float,
-    "INT64": sqlalchemy.types.Integer,
-    "INTEGER": sqlalchemy.types.Integer,
-    "NUMERIC": sqlalchemy.types.Numeric,
-    "STRING": sqlalchemy.types.String,
-    "TIMESTAMP": sqlalchemy.types.TIMESTAMP,
-}
-
 # Type converters
 ischema_names = {
     "int": INTEGER,
@@ -90,6 +76,7 @@ def __init__(self, key_type, value_type):
     "json": JSON,
     "varchar": VARCHAR,
     "boolean": BOOLEAN,
+    "binary": BINARY,
 }
 
 

tests/unit/test_databend_dialect.py

@@ -99,6 +99,7 @@ def getitem(self, idx):
             multi_column_row(["name1", "INT", "YES"]),
             multi_column_row(["name2", "date", "NO"]),
             multi_column_row(["name3", "boolean", "YES"]),
+            multi_column_row(["name4", "binary", "YES"])
         ]
 
         expected_query = """
@@ -135,6 +136,12 @@ def getitem(self, idx):
                     "nullable": True,
                     "default": None,
                 },
+                {
+                    "name": "name4",
+                    "type": sqlalchemy.types.BINARY,
+                    "nullable": True,
+                    "default": None,
+                },
             ]
             connection.execute.assert_called_once()
             assert str(connection.execute.call_args[0][0].compile()) == str(