e6data-python-connector 2.3.14__tar.gz → 2.3.15__tar.gz

{e6data_python_connector-2.3.14/e6data_python_connector.egg-info → e6data_python_connector-2.3.15}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: e6data-python-connector
-Version: 2.3.14
+Version: 2.3.15
 Summary: Client for the e6data distributed SQL Engine.
 Home-page: https://github.com/e6x-labs/e6data-python-connector
 Author: e6data, Inc.
@@ -39,7 +39,7 @@ Dynamic: summary
 
 # e6data Python Connector
 
-![version](https://img.shields.io/badge/version-2.3.14-blue.svg)
+![version](https://img.shields.io/badge/version-2.3.15-blue.svg)
 
 ## Introduction
 

{e6data_python_connector-2.3.14 → e6data_python_connector-2.3.15}/README.md

@@ -1,6 +1,6 @@
 # e6data Python Connector
 
-![version](https://img.shields.io/badge/version-2.3.14-blue.svg)
+![version](https://img.shields.io/badge/version-2.3.15-blue.svg)
 
 ## Introduction
 

{e6data_python_connector-2.3.14 → e6data_python_connector-2.3.15}/e6data_python_connector/dialect.py

@@ -148,6 +148,13 @@ _type_map = {
     'struct': types.String,
     'uniontype': types.String,
     'decimal': E6dataDecimal,
+    'numeric': E6dataDecimal,
+    # aliases for the short type names the e6data engine actually emits
+    'int': types.Integer,
+    'long': types.BigInteger,
+    'short': types.SmallInteger,
+    'bool': types.Boolean,
+    'real': types.Float,
 }
 
 
@@ -328,7 +335,17 @@ class E6dataDialect(default.DefaultDialect):
             row = dict()
             row["name"] = column.get('fieldName')
             field_type = str(column.get('fieldType')).lower()
-            row["type"] = _type_map.get(field_type, types.String)
+            # Engine types may carry parameters: 'decimal(7,2)', 'varchar(16)',
+            # 'array<int>'. Strip them so the base type maps; otherwise the exact
+            # key lookup misses and the column silently defaults to String.
+            base_type = field_type.split('(', 1)[0].split('<', 1)[0].strip()
+            mapped = _type_map.get(base_type)
+            if mapped is None:
+                _logger.warning(
+                    "e6data dialect: unmapped column type %r -> defaulting to String",
+                    column.get('fieldType'))
+                mapped = types.String
+            row["type"] = mapped
             rows.append(row)
         return rows
 

{e6data_python_connector-2.3.14 → e6data_python_connector-2.3.15/e6data_python_connector.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: e6data-python-connector
-Version: 2.3.14
+Version: 2.3.15
 Summary: Client for the e6data distributed SQL Engine.
 Home-page: https://github.com/e6x-labs/e6data-python-connector
 Author: e6data, Inc.
@@ -39,7 +39,7 @@ Dynamic: summary
 
 # e6data Python Connector
 
-![version](https://img.shields.io/badge/version-2.3.14-blue.svg)
+![version](https://img.shields.io/badge/version-2.3.15-blue.svg)
 
 ## Introduction
 

{e6data_python_connector-2.3.14 → e6data_python_connector-2.3.15}/setup.py

@@ -12,7 +12,7 @@
 
 import setuptools
 
-VERSION = (2, 3, 14,)
+VERSION = (2, 3, 15,)
 
 
 def get_long_desc():

{e6data_python_connector-2.3.14 → e6data_python_connector-2.3.15}/test/test_dialect_columns.py

@@ -82,5 +82,37 @@ class TestDialectGetColumns(unittest.TestCase):
         self.assertIs(columns[0]["type"], types.String)
 
 
+    def test_get_columns_maps_engine_short_names_and_parameterized_types(self):
+        # The e6data engine emits short names ('int') and parameterized types
+        # ('decimal(7,2)', 'varchar(16)', 'array<int>'). These must resolve to the
+        # right SQLAlchemy type, not silently default to String.
+        client = FakeClient([
+            {"fieldName": "sk", "fieldType": "int"},
+            {"fieldName": "big", "fieldType": "long"},
+            {"fieldName": "small", "fieldType": "short"},
+            {"fieldName": "flag", "fieldType": "bool"},
+            {"fieldName": "price", "fieldType": "decimal(7,2)"},
+            {"fieldName": "name", "fieldType": "varchar(16)"},
+            {"fieldName": "code", "fieldType": "char(1)"},
+            {"fieldName": "tags", "fieldType": "array<int>"},
+        ])
+        connection = FakeSQLAlchemyConnection(client)
+        dialect = E6dataDialect()
+        dialect.catalog_name = "lakehouse"
+
+        with patch.object(dialect_module, "Connection", FakeSQLAlchemyConnection):
+            columns = dialect.get_columns(connection, "items", "sales")
+
+        by_name = {c["name"]: c["type"] for c in columns}
+        self.assertIs(by_name["sk"], types.Integer)
+        self.assertIs(by_name["big"], types.BigInteger)
+        self.assertIs(by_name["small"], types.SmallInteger)
+        self.assertIs(by_name["flag"], types.Boolean)
+        self.assertIs(by_name["price"], E6dataDecimal)
+        self.assertIs(by_name["name"], types.String)
+        self.assertIs(by_name["code"], types.String)
+        self.assertIs(by_name["tags"], types.String)  # complex type normalizes, not a miss
+
+
 if __name__ == "__main__":
     unittest.main()