e6data-python-connector 2.3.14__py3-none-any.whl → 2.3.15__py3-none-any.whl

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.dist-info → e6data_python_connector-2.3.15.dist-info}/METADATA

@@ -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.dist-info → e6data_python_connector-2.3.15.dist-info}/RECORD

@@ -5,7 +5,7 @@ e6data_python_connector/connection_pool.py,sha256=Gzr3z4tJJV4cKUYWdyRqVtmJEvh6ZF
 e6data_python_connector/constants.py,sha256=AwACS9iUaYj84oRRHj9Waep9LGbNHxcE07BABV7s45Y,1406
 e6data_python_connector/datainputstream.py,sha256=DittJuNzTAVUiQ0ZzSFSfMnFC7YuHieF0YqjYt2dpZU,28187
 e6data_python_connector/date_time_utils.py,sha256=0n_3CchQTvXUHFIqWcenng9sW0l2A6sPLcDny_chDME,21283
-e6data_python_connector/dialect.py,sha256=yqtfWpS6JE-yDylysR9TzUTpT3kRHn6Rr94h2WifEoQ,12078
+e6data_python_connector/dialect.py,sha256=f-LrzauT0IdrbK965w2sLgd5OHv5L_hmzntyBJsc1A8,12883
 e6data_python_connector/e6data_grpc.py,sha256=AwTuRauFiZjIrAIpHdX87JcBxGSVee2-R1PsRN6gbSk,63747
 e6data_python_connector/exceptions.py,sha256=vrUqdzMq3LjbCmvr5_TcRq9eKmdqGfe-uVjR0BJQf5U,382
 e6data_python_connector/strategy.py,sha256=whAC-C031XFirP71hXKqdiMTJRIUUEvc34tuo2W8k-0,6539
@@ -53,7 +53,7 @@ test/test_connection_pool_e2e.py,sha256=UdqFbiFRY93tnp7DBR68CwZKfXD-gxiZH47qG2Ap
 test/test_current_implementation.py,sha256=2tbLsfzzsiopVrwaNfUqbOzZzPmC6S0M3M2kFR-mZ-I,4295
 test/test_decimal128_binary_parsing.py,sha256=eAtIvy8IvZszxhz9EoJET7-YA4KbY6IZAcIpsSmu6rQ,6512
 test/test_decimal128_parsing.py,sha256=ODSG4337GaWL0kdaQig5XQvmlSFnBkqhIat7Cs9VchI,9644
-test/test_dialect_columns.py,sha256=KdlPDVm2oITcSGL7WtGExn-Z76g0L4wgLK5hKGpo5Vk,2841
+test/test_dialect_columns.py,sha256=EDMBKi-DyUf_lDZWF8EiqghvkHatReYjiq9R_Ql2jL0,4482
 test/test_fix.py,sha256=d53VIY6oXpkn1SZGCi7xM2nRC_JveLkUouHa8CSOJDE,820
 test/test_improved_parsing.py,sha256=dWVpGjnfi_PWdZEV1vM8Ps5n0YvDM04VGM708fKBSUM,4001
 test/test_known_case.py,sha256=q4CVeU1ht3y_zWrdkZrvGHhU7zE5hNN91NMZVfdBZ_c,1854
@@ -77,9 +77,9 @@ test/tests_grpc.py,sha256=VRqJwW2BVRpL3c0BnwCKvtowogAS_0JYa1Ss4ynqTss,5916
 test/validate_decimal128.py,sha256=8m7FFD3nOHFkPKpN1kCd5MDDzfeEEJnMnOoSeXmX_I0,3320
 test/validate_implementation.py,sha256=2keEodBGxztdG1kgavTq-p-cLw1L9UNQLkXTfIijBNQ,5569
 test/verify_decimal_fix.py,sha256=n7XUV4dbo7kTdx-mVqhha9Nr6_bkDudwplHTdV-rM_Y,1051
-e6data_python_connector-2.3.14.dist-info/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
-e6data_python_connector-2.3.14.dist-info/METADATA,sha256=shgSca_ESgKD_FBBP05Fv3cQ7yA2XI0pwxPdmMa72UA,27193
-e6data_python_connector-2.3.14.dist-info/WHEEL,sha256=jB7zZ3N9hIM9adW7qlTAyycLYW9npaWKLRzaoVcLKcM,91
-e6data_python_connector-2.3.14.dist-info/entry_points.txt,sha256=MDK4cAOND0preLvq2Tlpnz00wXnP_Y_UopqJsAf3U7Y,77
-e6data_python_connector-2.3.14.dist-info/top_level.txt,sha256=nGwT_-fOD03WiucD15uscVhsUR42BzvBM2ddlnAMS4A,36
-e6data_python_connector-2.3.14.dist-info/RECORD,,
+e6data_python_connector-2.3.15.dist-info/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
+e6data_python_connector-2.3.15.dist-info/METADATA,sha256=KX_EQB6Gxp36F6CSJn0rJGJh4AZwTJknmQeK5VyydAQ,27193
+e6data_python_connector-2.3.15.dist-info/WHEEL,sha256=jB7zZ3N9hIM9adW7qlTAyycLYW9npaWKLRzaoVcLKcM,91
+e6data_python_connector-2.3.15.dist-info/entry_points.txt,sha256=MDK4cAOND0preLvq2Tlpnz00wXnP_Y_UopqJsAf3U7Y,77
+e6data_python_connector-2.3.15.dist-info/top_level.txt,sha256=nGwT_-fOD03WiucD15uscVhsUR42BzvBM2ddlnAMS4A,36
+e6data_python_connector-2.3.15.dist-info/RECORD,,

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()