use hive v3 on init

Author: igorvoltaic

pyiceberg/catalog/hive.py

@@ -34,8 +34,8 @@
 )
 from urllib.parse import urlparse
 
-from hive_metastore.v4.ThriftHiveMetastore import Client
-from hive_metastore.v4.ttypes import (
+from hive_metastore.v3.ThriftHiveMetastore import Client
+from hive_metastore.v3.ttypes import (
     AlreadyExistsException,
     CheckLockRequest,
     EnvironmentContext,
@@ -55,10 +55,10 @@
     StorageDescriptor,
     UnlockRequest,
 )
-from hive_metastore.v4.ttypes import (
+from hive_metastore.v3.ttypes import (
     Database as HiveDatabase,
 )
-from hive_metastore.v4.ttypes import (
+from hive_metastore.v3.ttypes import (
     Table as HiveTable,
 )
 from tenacity import retry, retry_if_exception_type, stop_after_attempt, wait_exponential
@@ -153,11 +153,13 @@
 
 logger = logging.getLogger(__name__)
 
+v3type = importlib.import_module("hive_metastore.v3.ttypes")
+v4type = importlib.import_module("hive_metastore.v4.ttypes")
 
 class _HiveClient:
     """Helper class to nicely open and close the transport."""
 
-    _transport: TTransport
+    _transport: TTransport.TTransportBase
     _ugi: Optional[List[str]]
     _hive_version: HiveVersion = HiveVersion(4, 0, 0)
     _hms_v3: object
@@ -184,8 +186,10 @@ def _get_hive_version(self) -> HiveVersion:
             version = map(int, open_client.getVersion().split("."))
             return HiveVersion(*version)
 
-    def _init_thrift_transport(self) -> TTransport:
+    def _init_thrift_transport(self) -> TTransport.TTransportBase:
         url_parts = urlparse(self._uri)
+        if not url_parts.hostname or not url_parts.port:
+            raise ValueError("hive hostname and port must be set")
         socket = TSocket.TSocket(url_parts.hostname, url_parts.port)
         if not self._kerberos_auth:
             return TTransport.TBufferedTransport(socket)