(3/x) SQLAlchemy 1.3 exposes the underlying connection / cursor with

different property names than SQLAlchemy > 1.4. So here I wrapped the logic
in a public method.

Signed-off-by: Jesse Whitehouse <[email protected]>

Author: Jesse Whitehouse

src/databricks/sqlalchemy/dialect/__init__.py

@@ -7,7 +7,7 @@
 import sqlalchemy
 from sqlalchemy import types, processors, event
 from sqlalchemy.engine import default, Engine
-from sqlalchemy.exc import DatabaseError
+from sqlalchemy.exc import DatabaseError, SQLAlchemyError
 from sqlalchemy.engine import reflection
 
 from databricks import sql
@@ -154,9 +154,7 @@ def get_columns(self, connection, table_name, schema=None, **kwargs):
             "date": DatabricksDate,
         }
 
-        with self.get_driver_connection(
-            connection
-        )._dbapi_connection.dbapi_connection.cursor() as cur:
+        with self.get_connection_cursor(connection) as cur:
             resp = cur.columns(
                 catalog_name=self.catalog,
                 schema_name=schema or self.schema,
@@ -245,9 +243,7 @@ def get_indexes(self, connection, table_name, schema=None, **kw):
 
     def get_table_names(self, connection, schema=None, **kwargs):
         TABLE_NAME = 1
-        with self.get_driver_connection(
-            connection
-        )._dbapi_connection.dbapi_connection.cursor() as cur:
+        with self.get_connection_cursor(connection) as cur:
             sql_str = "SHOW TABLES FROM {}".format(
                 ".".join([self.catalog, schema or self.schema])
             )
@@ -258,9 +254,7 @@ def get_table_names(self, connection, schema=None, **kwargs):
 
     def get_view_names(self, connection, schema=None, **kwargs):
         VIEW_NAME = 1
-        with self.get_driver_connection(
-            connection
-        )._dbapi_connection.dbapi_connection.cursor() as cur:
+        with self.get_connection_cursor(connection) as cur:
             sql_str = "SHOW VIEWS FROM {}".format(
                 ".".join([self.catalog, schema or self.schema])
             )
@@ -292,6 +286,21 @@ def has_table(self, connection, table_name, schema=None, **kwargs) -> bool:
                 return False
             else:
                 raise e
+            
+   
+    def get_connection_cursor(self, connection):
+        """Added for backwards compatibility with 1.3.x
+        """
+        if hasattr(connection, "_dbapi_connection"):
+            return connection._dbapi_connection.dbapi_connection.cursor()
+        elif hasattr(connection, "raw_connection"):
+            return connection.raw_connection().cursor()
+        elif hasattr(connection, "connection"):
+            return connection.connection.cursor()
+        
+        raise SQLAlchemyError("Databricks dialect can't obtain a cursor context manager from the dbapi")
+
+
 
     @reflection.cache
     def get_schema_names(self, connection, **kw):