Update sqlalchemy_hive.py

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 import mysql
+except ImportError:
+    # Newer versions require import from dialects not databases
+    from sqlalchemy.dialects import mysql
 from sqlalchemy.engine import default
 from sqlalchemy.sql import compiler
 from sqlalchemy.sql.compiler import SQLCompiler
@@ -121,7 +129,6 @@ def __init__(self, dialect):
 
 _type_map = {
     'boolean': types.Boolean,
-    'tinyint': mysql.MSTinyInteger,
     'smallint': types.SmallInteger,
     'int': types.Integer,
     'bigint': types.BigInteger,
@@ -140,6 +147,16 @@ def __init__(self, dialect):
     'decimal': HiveDecimal,
 }
 
+try:
+    _type_map.update({
+        'tinyint': mysql.MSTinyInteger
+    })
+except AttributeError:
+    # Support for version 2.x of sqlalchemy
+    _type_map.update({
+        'tinyint': mysql.base.MSTinyInteger
+    })
+
 
 class HiveCompiler(SQLCompiler):
     def visit_concat_op_binary(self, binary, operator, **kw):