changing drivers to support hive, presto and trino with sqlalchemy>=2.0 (#448)

Author: nicholas-miles

pyhive/sqlalchemy_hive.py

@@ -13,11 +13,19 @@
 
 import re
 from sqlalchemy import exc
-from sqlalchemy import processors
+try:
+    from sqlalchemy import processors
+except ImportError:
+    # Newer versions of sqlalchemy require:
+    from sqlalchemy.engine import processors
 from sqlalchemy import types
 from sqlalchemy import util
 # TODO shouldn't use mysql type
-from sqlalchemy.databases import mysql
+try:
+    from sqlalchemy.databases.mysql import MSTinyInteger
+except ImportError:
+    # Newer versions of sqlalchemy require:
+    from sqlalchemy.dialects.mysql import MSTinyInteger
 from sqlalchemy.engine import default
 from sqlalchemy.sql import compiler
 from sqlalchemy.sql.compiler import SQLCompiler
@@ -121,7 +129,7 @@ def __init__(self, dialect):
 
 _type_map = {
     'boolean': types.Boolean,
-    'tinyint': mysql.MSTinyInteger,
+    'tinyint': MSTinyInteger,
     'smallint': types.SmallInteger,
     'int': types.Integer,
     'bigint': types.BigInteger,

pyhive/sqlalchemy_presto.py

@@ -13,7 +13,11 @@
 from sqlalchemy import types
 from sqlalchemy import util
 # TODO shouldn't use mysql type
-from sqlalchemy.databases import mysql
+try:
+    from sqlalchemy.databases.mysql import MSTinyInteger
+except ImportError:
+    # Newer versions of sqlalchemy require:
+    from sqlalchemy.dialects.mysql import MSTinyInteger
 from sqlalchemy.engine import default
 from sqlalchemy.sql import compiler
 from sqlalchemy.sql.compiler import SQLCompiler
@@ -29,7 +33,7 @@ class PrestoIdentifierPreparer(compiler.IdentifierPreparer):
 
 _type_map = {
     'boolean': types.Boolean,
-    'tinyint': mysql.MSTinyInteger,
+    'tinyint': MSTinyInteger,
     'smallint': types.SmallInteger,
     'integer': types.Integer,
     'bigint': types.BigInteger,

pyhive/sqlalchemy_trino.py

@@ -13,7 +13,11 @@
 from sqlalchemy import types
 from sqlalchemy import util
 # TODO shouldn't use mysql type
-from sqlalchemy.databases import mysql
+try:
+    from sqlalchemy.databases.mysql import MSTinyInteger
+except ImportError:
+    # Newer versions of sqlalchemy require:
+    from sqlalchemy.dialects.mysql import MSTinyInteger
 from sqlalchemy.engine import default
 from sqlalchemy.sql import compiler
 from sqlalchemy.sql.compiler import SQLCompiler
@@ -28,7 +32,7 @@ class TrinoIdentifierPreparer(PrestoIdentifierPreparer):
 
 _type_map = {
     'boolean': types.Boolean,
-    'tinyint': mysql.MSTinyInteger,
+    'tinyint': MSTinyInteger,
     'smallint': types.SmallInteger,
     'integer': types.Integer,
     'bigint': types.BigInteger,