Only install PyHive for default HiveServer2Client driver

airbnb · Sep 8, 2018 · 6945b0a · 6945b0a
1 parent 49f7073
commit 6945b0a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/omniduct/_version.py b/omniduct/_version.py
@@ -39,7 +39,6 @@
 
     'hiveserver2': [
         'pyhive[hive]>=0.4',  # Primary client
-        'impyla>=0.14.0',  # Primary client
         'thrift>=0.10.0',  # Thrift dependency which seems not to be installed with upstream deps
     ],
 

diff --git a/omniduct/databases/hiveserver2.py b/omniduct/databases/hiveserver2.py
@@ -97,7 +97,10 @@ def _connect(self):
             self._sqlalchemy_engine = create_engine('hive://{}:{}/{}'.format(self.host, self.port, self.schema))
             self._sqlalchemy_metadata = MetaData(self._sqlalchemy_engine)
         elif self.driver == 'impyla':
-            import impala.dbapi
+            try:
+                import impala.dbapi
+            except ModuleNotFoundError:
+                raise ImportError("Please install impyla or specify driver='pyhive'.")
             self.__hive = impala.dbapi.connect(host=self.host,
                                                port=self.port,
                                                auth_mechanism=self.auth_mechanism,